Self-supervised category-enhanced graph neural networks for recommendation

被引:1
作者
Yang, Funing [1 ]
Du, Haihui [1 ]
Zhang, Xingliang [2 ]
Yang, Yongjian [1 ]
Wang, Ying [1 ]
机构
[1] Jilin Univ, Coll Comp Sci & Technol, Jilin, Peoples R China
[2] China Mobile Grp Jilin Co Ltd, Changchun, Peoples R China
基金
中国国家自然科学基金;
关键词
Graph neural network; Contrastive learning; Self-supervised learning;
D O I
10.1016/j.knosys.2025.113109
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The emergence of Graph Neural Networks (GNNs) has substantially advanced recommendation systems based on collaborative filtering. Despite their effectiveness, these networks are often susceptible to noise and sparsity problems from datasets. Thus, contrastive learning and knowledge graphs have been introduced as solutions to recommendation systems. However, both of these techniques have been incorporated into recommendation models as auxiliary tasks, neglecting the enhancement of node representations for the overall task. Therefore, we propose a novel contrastive learning method called C Category-Enriched C Contrastive L Learning ( (CECL (CECL) to obtain high-quality user-item interactions. Unlike previous studies, CECL does not introduce categories into the recommendation system as auxiliary supervised signals but uses them as direct supervised signals during model training. This directly supervised signal can enable the model to learn and emphasise category-specific patterns and relationships more efficiently, resulting in more accurate and stable user and item embeddings. By showing optimised category alignment, CECL improves representation quality. Specifically, we first construct initial embeddings for users and items based on the categories of user interactions and those to which items belong. Then, these embeddings are input into the recommendation model. The primary recommendation task is conducted in the supervised model. In the self-supervised model, contrastive pairs are constructed using edge dropout, and the consistency between different views of the same node is maximised to explore stable category preferences. Finally, multi-task training is applied to jointly optimise the recommendation and contrastive learning tasks. Comprehensive experiments conducted on eight real-world datasets validate the effectiveness of the proposed CECL. Compared with existing graph-enhanced methods, CECL achieves an average improvement of 15.17% in Recall@20 and 9.64% in NDCG@20 metrics. Our implementations are available at https://github.com/shark-art/Code.
引用
收藏
页数:11
相关论文
共 38 条
[1]  
Chen Gang, 2023, 2023 3rd International Conference on Neural Networks, Information and Communication Engineering (NNICE), P57, DOI 10.1109/NNICE58320.2023.10105711
[2]   Personalized Food Recommendation as Constrained Question Answering over a Large-scale Food Knowledge Graph [J].
Chen, Yu ;
Subburathinam, Ananya ;
Chen, Ching-Hua ;
Zaki, Mohammed J. .
WSDM '21: PROCEEDINGS OF THE 14TH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING, 2021, :553-561
[3]   Dressing as a Whole: Outfit Compatibility Learning Based on Node-wise Graph Neural Networks [J].
Cui, Zeyu ;
Li, Zekun ;
Wu, Shu ;
Zhang, Xiaoyu ;
Wang, Liang .
WEB CONFERENCE 2019: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2019), 2019, :307-317
[4]   Graph Neural Networks for Recommender System [J].
Gao, Chen ;
Wang, Xiang ;
He, Xiangnan ;
Li, Yong .
WSDM'22: PROCEEDINGS OF THE FIFTEENTH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING, 2022, :1623-1625
[5]   KAGNN: Graph neural network with kernel alignment for heterogeneous graph learning [J].
Han, Mingjing ;
Zhang, Han .
KNOWLEDGE-BASED SYSTEMS, 2024, 289
[6]   LightGCN: Simplifying and Powering Graph Convolution Network for Recommendation [J].
He, Xiangnan ;
Deng, Kuan ;
Wang, Xiang ;
Li, Yan ;
Zhang, Yongdong ;
Wang, Meng .
PROCEEDINGS OF THE 43RD INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR '20), 2020, :639-648
[7]   Neural Collaborative Filtering [J].
He, Xiangnan ;
Liao, Lizi ;
Zhang, Hanwang ;
Nie, Liqiang ;
Hu, Xia ;
Chua, Tat-Seng .
PROCEEDINGS OF THE 26TH INTERNATIONAL CONFERENCE ON WORLD WIDE WEB (WWW'17), 2017, :173-182
[8]  
Huang Junjie, 2023, Database Systems for Advanced Applications: 28th International Conference, DASFAA 2023, Proceedings. Lecture Notes in Computer Science (13944), P373, DOI 10.1007/978-3-031-30672-3_25
[9]   A Survey on Contrastive Self-Supervised Learning [J].
Jaiswal, Ashish ;
Babu, Ashwin Ramesh ;
Zadeh, Mohammad Zaki ;
Banerjee, Debapriya ;
Makedon, Fillia .
TECHNOLOGIES, 2021, 9 (01)
[10]   Hierarchical Fashion Graph Network for Personalized Outfit Recommendation [J].
Li, Xingchen ;
Wang, Xiang ;
He, Xiangnan ;
Chen, Long ;
Xiao, Jun ;
Chua, Tat-Seng .
PROCEEDINGS OF THE 43RD INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR '20), 2020, :159-168