跳到主要导航 跳到搜索 跳到主要内容

An Empirical Study Towards Prompt-Tuning for Graph Contrastive Pre-Training in Recommendations

  • Haoran Yang
  • , Xiangyu Zhao*
  • , Yicong Li
  • , Hongxu Chen
  • , Guandong Xu*
  • *此作品的通讯作者
  • University of Technology Sydney
  • City University of Hong Kong
  • University of Queensland

科研成果: 书/报告/会议事项章节会议稿件同行评审

摘要

Graph contrastive learning (GCL) has emerged as an effective technology for various graph learning tasks. It has been successfully applied in real-world recommender systems, where the contrastive loss and downstream recommendation objectives are combined to form the overall objective function. However, this approach deviates from the original GCL paradigm, which pre-trains graph embeddings without involving downstream training objectives. In this paper, we propose a novel framework called CPTPP, which enhances GCL-based recommender systems by leveraging prompt tuning. This framework allows us to fully exploit the advantages of the original GCL protocol. Specifically, we first summarize user profiles in graph recommender systems to automatically generate personalized user prompts. These prompts are then combined with pre-trained user embeddings for prompt tuning in downstream tasks. This helps bridge the gap between pre-training and downstream tasks. Our extensive experiments on three benchmark datasets confirm the effectiveness of CPTPP compared to state-of-the-art baselines. Additionally, a visualization experiment illustrates that user embeddings generated by CPTPP have a more uniform distribution, indicating improved modeling capability for user preferences. The implementation code is available online2 for reproducibility.

源语言英语
主期刊名Advances in Neural Information Processing Systems 36 - 37th Conference on Neural Information Processing Systems, NeurIPS 2023
编辑A. Oh, T. Neumann, A. Globerson, K. Saenko, M. Hardt, S. Levine
出版商Neural information processing systems foundation
ISBN(电子版)9781713899921
出版状态已出版 - 2023
已对外发布
活动37th Conference on Neural Information Processing Systems, NeurIPS 2023 - New Orleans, 美国
期限: 10 12月 202316 12月 2023

出版系列

姓名Advances in Neural Information Processing Systems
36
ISSN(印刷版)1049-5258

会议

会议37th Conference on Neural Information Processing Systems, NeurIPS 2023
国家/地区美国
New Orleans
时期10/12/2316/12/23

指纹

探究 'An Empirical Study Towards Prompt-Tuning for Graph Contrastive Pre-Training in Recommendations' 的科研主题。它们共同构成独一无二的指纹。

引用此