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

Uniform Graph Pre-training and Prompting for Transferable Recommendation

  • Qing Yu
  • , Lixin Zou
  • , Xiangyang Luo
  • , Xiangyu Zhao*
  • , Chenliang Li
  • *此作品的通讯作者
  • Wuhan University
  • State Key Lab of Mathematical Engineering and Advanced Computing
  • City University of Hong Kong

科研成果: 期刊稿件文章同行评审

摘要

Recently, the paradigm of pre-training and fine-tuning has achieved impressive performance owing to their ability to transfer general knowledge from pre-trained domain to target domain. Meanwhile, graph neural networks (GNNs) have gained prominence in recommender systems. However, there is a lack of unified pre-training and fine-tuning paradigms in graph-based recommendation systems. Applying pre-training and fine-tuning in graph-based recommendation is challenging due to the unique characteristics of recommendation data, including the non-uniform representation, negative transfer effects, and skewed data distributions. To overcome these challenges, we introduce pre-training and prompting recommendation (ProRec), a novel model that synergizes uniform graph pre-training with prompt-tuning for recommendation systems. Specifically, to address the challenge of inconsistent features across different recommendation datasets, ProRec constructs unified input features at the subgraph level and uses a graph auto-encoder for pre-training, laying the foundation for uniform knowledge transfer from the pre-trained domain to the downstream domain. Additionally, ProRec employs prompt-tuning during the fine-tuning phase, which, in a parameter-efficient manner, enhances the generalization of pre-trained knowledge to downstream tasks thereby reducing negative transfer effects. Furthermore, a cross-layer contrastive learning strategy is adopted to eliminate uneven data distribution, promoting more evenly distributed and informative representations. Finally, extensive benchmark comparisons have demonstrated that ProRec outperforms the latest state-of-the-art methods. The source code necessary for replication is available at https://github.com/Code2Q/ProRec.

源语言英语
文章编号ART101
期刊ACM Transactions on Information Systems
43
4
DOI
出版状态已出版 - 9 7月 2025
已对外发布

指纹

探究 'Uniform Graph Pre-training and Prompting for Transferable Recommendation' 的科研主题。它们共同构成独一无二的指纹。

引用此