Scalable Collaborative Ranking for Personalized Prediction

Ben Dai, Xiaotong Shen, Junhui Wang, Annie Qu

Research output: Contribution to journalArticlepeer-review

Abstract

Personalized prediction presents an important yet challenging task, which predicts user-specific preferences on a large number of items given limited information. It is often modeled as certain recommender systems focusing on ordinal or continuous ratings, as in collaborative filtering and content-based filtering. In this article, we propose a new collaborative ranking system to predict most-preferred items for each user given search queries. Particularly, we propose a ψ-ranker based on ranking functions incorporating information on users, items, and search queries through latent factor models. Moreover, we show that the proposed nonconvex surrogate pairwise ψ-loss performs well under four popular bipartite ranking losses, such as the sum loss, pairwise zero-one loss, discounted cumulative gain, and mean average precision. We develop a parallel computing strategy to optimize the intractable loss of two levels of nonconvex components through difference of convex programming and block successive upper-bound minimization. Theoretically, we establish a probabilistic error bound for the ψ-ranker and show that its ranking error has a sharp rate of convergence in the general framework of bipartite ranking, even when the dimension of the model parameters diverges with the sample size. Consequently, this result also indicates that the ψ-ranker performs better than two major approaches in bipartite ranking: pairwise ranking and scoring. Finally, we demonstrate the utility of the ψ-ranker by comparing it with some strong competitors in the literature through simulated examples as well as Expedia booking data. Supplementary materials for this article are available online.

Original languageEnglish (US)
Pages (from-to)1215-1223
Number of pages9
JournalJournal of the American Statistical Association
Volume116
Issue number535
DOIs
StatePublished - 2021

Bibliographical note

Funding Information:
Research supported in part by NSF grants DMS-1712564, DMS-1721216, DMS-1613190, and Hong Kong RGC grants GRF-11331016, GRF-11303918, and GRF-11300919. The authors thank the editors, the associate editor, and two anonymous referees for helpful comments and suggestions.

Publisher Copyright:
© 2019 American Statistical Association.

Keywords

  • Bipartite ranking
  • Discounted cumulative gain
  • Latent factor models
  • Matrix factorization
  • Mean average precision
  • Recommender systems

Fingerprint

Dive into the research topics of 'Scalable Collaborative Ranking for Personalized Prediction'. Together they form a unique fingerprint.

Cite this