AUC Optimisation and Collaborative Filtering

Charanpal Dhanjal 1 Romaric Gaudel 2 Stephan Clemencon 1
2 SEQUEL - Sequential Learning
Inria Lille - Nord Europe, CRIStAL - Centre de Recherche en Informatique, Signal et Automatique de Lille (CRIStAL) - UMR 9189
Abstract : In recommendation systems, one is interested in the ranking of the predicted items as opposed to other losses such as the mean squared error. Although a variety of ways to evaluate rankings exist in the literature, here we focus on the Area Under the ROC Curve (AUC) as it widely used and has a strong theoretical underpinning. In practical recommendation, only items at the top of the ranked list are presented to the users. With this in mind, we propose a class of objective functions over matrix factorisations which primarily represent a smooth surrogate for the real AUC, and in a special case we show how to prioritise the top of the list. The objectives are differentiable and optimised through a carefully designed stochastic gradient-descent-based algorithm which scales linearly with the size of the data. In the special case of square loss we show how to improve computational complexity by leveraging previously computed measures. To understand theoretically the underlying matrix factorisation approaches we study both the consistency of the loss functions with respect to AUC, and generalisation using Rademacher theory. The resulting generalisation analysis gives strong motivation for the optimisation under study. Finally, we provide computation results as to the efficacy of the proposed method using synthetic and real data.
Document type :
Preprints, Working Papers, ...
Complete list of metadatas

https://hal.archives-ouvertes.fr/hal-01185836
Contributor : Charanpal Dhanjal <>
Submitted on : Friday, August 21, 2015 - 4:09:08 PM
Last modification on : Friday, June 7, 2019 - 11:09:18 AM
Long-term archiving on : Wednesday, April 26, 2017 - 10:08:35 AM

Files

LocalRankingArxiv.pdf
Files produced by the author(s)

Identifiers

  • HAL Id : hal-01185836, version 1
  • ARXIV : 1508.06091

Citation

Charanpal Dhanjal, Romaric Gaudel, Stephan Clemencon. AUC Optimisation and Collaborative Filtering. 2015. ⟨hal-01185836⟩

Share

Metrics

Record views

355

Files downloads

367