Learning-Efficient Yet Generalizable Collaborative Filtering for Item Recommendation

Yuanhao Pu, Xiaolong Chen, Xu Huang, Jin Chen, Defu Lian*, Enhong Chen

*Corresponding author for this work

Research output: Contribution to journalConference article published in journalpeer-review

Abstract

The weighted squared loss is a common component in several Collaborative Filtering (CF) algorithms for item recommendation, including the representative implicit Alternating Least Squares (iALS). Despite its widespread use, this loss function lacks a clear connection to ranking objectives such as Discounted Cumulative Gain (DCG), posing a fundamental challenge in explaining the exceptional ranking performance observed in these algorithms. In this work, we make a breakthrough by establishing a connection between squared loss and ranking metrics through a Taylor expansion of the DCG-consistent surrogate loss-softmax loss. We also discover a new surrogate squared loss function, namely Ranking-Generalizable Squared (RG2) loss, and conduct thorough theoretical analyses on the DCG-consistency of the proposed loss function. Later, we present an example of utilizing the RG2 loss with Matrix Factorization (MF), coupled with a generalization upper bound and an ALS optimization algorithm that leverages closed-form solutions over all items. Experimental results over three public datasets demonstrate the effectiveness of the RG2 loss, exhibiting ranking performance on par with, or even surpassing, the softmax loss while achieving faster convergence.

Original languageEnglish
Pages (from-to)41183-41203
Number of pages21
JournalProceedings of Machine Learning Research
Volume235
Publication statusPublished - 2024
Event41st International Conference on Machine Learning, ICML 2024 - Vienna, Austria
Duration: 21 Jul 202427 Jul 2024

Bibliographical note

Publisher Copyright:
Copyright 2024 by the author(s)

Fingerprint

Dive into the research topics of 'Learning-Efficient Yet Generalizable Collaborative Filtering for Item Recommendation'. Together they form a unique fingerprint.

Cite this