An Adaptive Graph Pre-training Framework for Localized Collaborative Filtering

被引:21
作者
Wang, Yiqi [1 ]
Li, Chaozhuo [2 ]
Liu, Zheng [2 ]
Li, Mingzheng [3 ]
Tang, Jiliang [1 ]
Xie, Xing [2 ]
Chen, Lei [4 ]
Yu, Philip S. [5 ]
机构
[1] Michigan State Univ, 428 S Shaw Ln Rm 3115, E Lansing, MI 48824 USA
[2] Microsoft Res Asia, 5 Danling St, Beijing 100080, Peoples R China
[3] Microsoft, Shanghai, Peoples R China
[4] Hong Kong Univ Sci & Technol, Dept Comp Sci & Engn, Kowloon, Hong Kong, Peoples R China
[5] Univ Illinois, 851 S Morgan St,Rm 1138 SEO, Chicago, IL 60607 USA
基金
美国国家科学基金会;
关键词
Graph neural networks; recommendation systems; model pre-training;
D O I
10.1145/3555372
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Graph neural networks (GNNs) have been widely applied in the recommendation tasks and have achieved very appealing performance. However, most GNN-based recommendation methods suffer from the problem of data sparsity in practice. Meanwhile, pre-training techniques have achieved great success in mitigating data sparsity in various domains such as natural language processing (NLP) and computer vision (CV). Thus, graph pre-training has the great potential to alleviate data sparsity in GNN-based recommendations. However, pre-training GNNs for recommendations faces unique challenges. For example, user-item interaction graphs in different recommendation tasks have distinct sets of users and items, and they often present different properties. Therefore, the successful mechanisms commonly used in NLP and CV to transfer knowledge from pre-training tasks to downstream tasks such as sharing learned embeddings or feature extractors are not directly applicable to existing GNN-based recommendations models. To tackle these challenges, we delicately design an adaptive graph pre-training framework for localized collaborative filtering (ADAPT). It does not require transferring user/item embeddings, and is able to capture both the common knowledge across different graphs and the uniqueness for each graph simultaneously. Extensive experimental results have demonstrated the effectiveness and superiority of ADAPT.
引用
收藏
页数:27
相关论文
共 82 条
[1]   Toward the next generation of recommender systems: A survey of the state-of-the-art and possible extensions [J].
Adomavicius, G ;
Tuzhilin, A .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2005, 17 (06) :734-749
[2]  
Agarwal D., 2011, P 17 ACM SIGKDD INT, P609, DOI 10.1145/2020408.2020504
[3]  
Berkovsky S, 2007, LECT NOTES ARTIF INT, V4511, P355
[4]  
Bruna J, 2014, Arxiv, DOI [arXiv:1312.6203, 10.48550/arXiv.1312.6203, DOI 10.48550/ARXIV.1312.6203]
[5]  
Cai D, 2020, AAAI CONF ARTIF INTE, V34, P7464
[6]  
Defferrard M, 2016, ADV NEUR IN, V29
[7]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[8]  
Devlin J, 2019, Arxiv, DOI [arXiv:1810.04805, 10.48550/arXiv.1810.04805]
[9]  
Duvenaudt D, 2015, ADV NEUR IN, V28
[10]   Graph Neural Networks for Social Recommendation [J].
Fan, Wenqi ;
Ma, Yao ;
Li, Qing ;
He, Yuan ;
Zhao, Eric ;
Tang, Jiliang ;
Yin, Dawei .
WEB CONFERENCE 2019: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2019), 2019, :417-426