Lune

ICML2024Top-tier venue

Learning-Efficient Yet Generalizable Collaborative Filtering for Item Recommendation

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

2024Year
8Citations
2Top-tier citations

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 (RG 2 ) loss, and conduct thorough theoretical analyses on the DCGconsistency of the proposed loss function. Later, we present an example of utilizing the RG 2 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 RG 2 loss, exhibiting ranking performance on par with, or even surpassing, the softmax loss while achieving faster convergence. Introduction Collaborative filtering is a typical technique in item recommendations that leverages similarities between user behav-

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext 65b06f07-13ce-4c1e-a4ff-b3613c5e0759

Cited by top-tier papers2

Ask how each one uses it

Builds on6

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines