武汉大学自然科学学报(英文版)2024,Vol.29Issue(2) :134-144.DOI:10.1051/wujns/2024292134

Fu-Rec:Multi-Task Learning Recommendation Model Fusing Neighbor-Discrimination and Self-Discrimination

ZHENG Sirui HUANG Bo LIU Jin ZENG Guohui YIN Ling LI Zhi SUN Tie
武汉大学自然科学学报(英文版)2024,Vol.29Issue(2) :134-144.DOI:10.1051/wujns/2024292134

Fu-Rec:Multi-Task Learning Recommendation Model Fusing Neighbor-Discrimination and Self-Discrimination

ZHENG Sirui 1HUANG Bo 1LIU Jin 2ZENG Guohui 1YIN Ling 1LI Zhi 3SUN Tie4
扫码查看

作者信息

  • 1. School of Electronic and Electrical Engineering,Shanghai University of Engineering Science,Shanghai 201600,China
  • 2. School of Computer,Wuhan University,Wuhan 430072,Hubei,China
  • 3. School of Computer Science and Engineering,Guangxi Normal University,Guilin 541004,Guangxi,China
  • 4. AIoT Manufacturing Solutions Technology Co.,Ltd.,Hefei 230000,Anhui,China
  • 折叠

Abstract

In recent years,self-supervised learning has achieved great success in areas such as computer vision and natural language pro-cessing because it can mine supervised signals from unlabeled data and reduce the reliance on manual labels.However,the currently gener-ated self-supervised signals are either neighbor discrimination or self-discrimination,and there is no model to integrate neighbor discrimi-nation and self-discrimination.Based on this,this paper proposes Fu-Rec that integrates neighbor-discrimination contrastive learning and self-discrimination contrastive learning,which consists of three modules:(1)neighbor-discrimination contrastive learning,(2)self-discrimination contrastive learning,and(3)recommendation module.The neighbor-discrimination contrastive learning and self-discrimination contrastive learning tasks are used as auxiliary tasks to assist the recommendation task.The Fu-Rec model effectively uti-lizes the respective advantages of neighbor-discrimination and self-discrimination to consider the information of the user's neighbors as well as the user and the item itself for the recommendation,which results in better performance of the recommendation module.Experi-mental results on several public datasets demonstrate the effectiveness of the Fu-Rec proposed in this paper.

Key words

self-supervised learning/recommendation system/contrastive learning/multi-task learning

引用本文复制引用

基金项目

Scientific and Technological Innovation 2030-Major Project of New Generation Artificial Intelligence(2020AAA0109300)

science and Technology Commission of Shanghai Municipality(21DZ2203100)

2023 Anhui Province Key Research and Development Plan Project-Special Project of Science and Technology Cooperation(2023i11020002)

出版年

2024
武汉大学自然科学学报(英文版)
武汉大学

武汉大学自然科学学报(英文版)

CSTPCDCSCD
影响因子:0.066
ISSN:1007-1202
参考文献量33
段落导航相关论文