English

Conditioned Variational Autoencoder for top-N item recommendation

Machine Learning 2020-05-05 v2 Information Retrieval

Abstract

In this paper, we propose a Conditioned Variational Autoencoder (C-VAE) for constrained top-N item recommendation where the recommended items must satisfy a given condition. The proposed model architecture is similar to a standard VAE in which the condition vector is fed into the encoder. The constrained ranking is learned during training thanks to a new reconstruction loss that takes the input condition into account. We show that our model generalizes the state-of-the-art Mult-VAE collaborative filtering model. Moreover, we provide insights on what C-VAE learns in the latent space, providing a human-friendly interpretation. Experimental results underline the potential of C-VAE in providing accurate recommendations under constraints. Finally, the performed analyses suggest that C-VAE can be used in other recommendation scenarios, such as context-aware recommendation.

Keywords

Cite

@article{arxiv.2004.11141,
  title  = {Conditioned Variational Autoencoder for top-N item recommendation},
  author = {Tommaso Carraro and Mirko Polato and Fabio Aiolli},
  journal= {arXiv preprint arXiv:2004.11141},
  year   = {2020}
}
R2 v1 2026-06-23T15:03:06.173Z