Variational Factorization Machines for Preference Elicitation in Large-Scale Recommender Systems - Inria - Institut national de recherche en sciences et technologies du numérique Accéder directement au contenu
Communication Dans Un Congrès Année : 2022

Variational Factorization Machines for Preference Elicitation in Large-Scale Recommender Systems

Résumé

Factorization machines (FMs) are a powerful tool for regression and classification in the context of sparse observations, that has been successfully applied to collaborative filtering, especially when side information over users or items is available. Bayesian formulations of FMs have been proposed to provide confidence intervals over the predictions made by the model, however they usually involve Markov-chain Monte Carlo methods that require many samples to provide accurate predictions, resulting in slow training in the context of largescale data. In this paper, we propose a variational formulation of factorization machines that allows us to derive a simple objective that can be easily optimized using standard mini-batch stochastic gradient descent, making it amenable to large-scale data. Our algorithm learns an approximate posterior distribution over the user and item parameters, which leads to confidence intervals over the predictions. We show, using several datasets, that it has comparable or better performance than existing methods in terms of prediction accuracy, and provide some applications in active learning strategies, e.g., preference elicitation techniques.
Fichier principal
Vignette du fichier
main.pdf (301.53 Ko) Télécharger le fichier
Origine : Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03880215 , version 1 (01-12-2022)

Identifiants

  • HAL Id : hal-03880215 , version 1

Citer

Jill-Jênn Vie, Tomas Rigaux, Hisashi Kashima. Variational Factorization Machines for Preference Elicitation in Large-Scale Recommender Systems. IEEE BigData 2022, Dec 2022, Osaka, Japan. ⟨hal-03880215⟩
20 Consultations
29 Téléchargements

Partager

Gmail Facebook X LinkedIn More