Towards Representation Alignment and Uniformity in Collaborative Filtering

被引:106
作者
Wang, Chenyang [1 ]
Yu, Yuanqing [1 ]
Ma, Weizhi [2 ]
Zhang, Min [1 ]
Chen, Chong [1 ]
Liu, Yiqun [1 ]
Ma, Shaoping [1 ]
机构
[1] Tsinghua Univ, BNRist, DCST, Beijing 100084, Peoples R China
[2] Tsinghua Univ, AIR, Beijing 100084, Peoples R China
来源
PROCEEDINGS OF THE 28TH ACM SIGKDD CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, KDD 2022 | 2022年
关键词
Recommender Systems; Collaborative Filtering; Representation Learning; Alignment and Uniformity;
D O I
10.1145/3534678.3539253
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Collaborative filtering (CF) plays a critical role in the development of recommender systems. Most CF methods utilize an encoder to embed users and items into the same representation space, and the Bayesian personalized ranking (BPR) loss is usually adopted as the objective function to learn informative encoders. Existing studies mainly focus on designing more powerful encoders (e.g., graph neural network) to learn better representations. However, few efforts have been devoted to investigating the desired properties of representations in CF, which is important to understand the rationale of existing CF methods and design newlearning objectives. In this paper, we measure the representation quality in CF from the perspective of alignment and uniformity on the hypersphere. We first theoretically reveal the connection between the BPR loss and these two properties. Then, we empirically analyze the learning dynamics of typical CF methods in terms of quantified alignment and uniformity, which shows that better alignment or uniformity both contribute to higher recommendation performance. Based on the analyses results, a learning objective that directly optimizes these two properties is proposed, named DirectAU. We conduct extensive experiments on three public datasets, and the proposed learning framework with a simple matrix factorization model leads to significant performance improvements compared to state-of-the-art CF methods. Our implementations are publicly available(1).
引用
收藏
页码:1816 / 1825
页数:10
相关论文
共 34 条
[1]  
[Anonymous], 2008, P 14 ACM SIGKDD INT, DOI DOI 10.1145/1401890.1401944
[2]  
[Anonymous], 2020, INT C MACH LEARN
[3]  
Bojanowski P, 2017, 34 INT C MACHINE LEA, V70
[4]   Efficient Neural Matrix Factorization without Sampling for Recommendation [J].
Chen, Chong ;
Min, Zhang ;
Zhang, Yongfeng ;
Liu, Yiqun ;
Ma, Shaoping .
ACM TRANSACTIONS ON INFORMATION SYSTEMS, 2020, 38 (02)
[5]   Attentive Collaborative Filtering: Multimedia Recommendation with Item- and Component-Level Attention [J].
Chen, Jingyuan ;
Zhang, Hanwang ;
He, Xiangnan ;
Nie, Liqiang ;
Liu, Wei ;
Chua, Tat-Seng .
SIGIR'17: PROCEEDINGS OF THE 40TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL, 2017, :335-344
[6]  
ChenyangWang Weizhi Ma, 2022, ACM Transactions on Information Systems (TOIS)
[7]   Deep Neural Networks for YouTube Recommendations [J].
Covington, Paul ;
Adams, Jay ;
Sargin, Emre .
PROCEEDINGS OF THE 10TH ACM CONFERENCE ON RECOMMENDER SYSTEMS (RECSYS'16), 2016, :191-198
[8]  
Gao TY, 2021, 2021 CONFERENCE ON EMPIRICAL METHODS IN NATURAL LANGUAGE PROCESSING (EMNLP 2021), P6894
[9]  
Grill J.-B., 2020, ARXIV200607733
[10]   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