Research output: Chapter in Book/Report/Conference proceeding › Conference contribution › Research › peer-review
Recent research has shown the advantages of using autoencoders based on deep neural networks for collaborative filtering. In particular, the recently proposed Mult-VAE model, which used the multinomial likelihood variational autoencoders, has shown excellent results for top-N recommendations. In this work, we propose the Recommender VAE (RecVAE) model that originates from our research on regularization techniques for variational autoencoders. RecVAE introduces several novel ideas to improve Mult-VAE, including a novel composite prior distribution for the latent codes, a new approach to setting the β hyperparameter for the β-VAE framework, and a new approach to training based on alternating updates. In experimental evaluation, we show that RecVAE significantly outperforms previously proposed autoencoder-based models, including Mult-VAE and RaCT, across classical collaborative filtering datasets, and present a detailed ablation study to assess our new developments. Code and models are available at https://github.com/ilya-shenbin/RecVAE.
| Original language | English |
|---|---|
| Title of host publication | WSDM 2020 - Proceedings of the 13th International Conference on Web Search and Data Mining |
| Publisher | Association for Computing Machinery |
| Pages | 528-536 |
| Number of pages | 9 |
| ISBN (Electronic) | 9781450368223 |
| DOIs | |
| State | Published - 20 Jan 2020 |
| Event | 13th ACM International Conference on Web Search and Data Mining, WSDM 2020 - Houston, United States Duration: 3 Feb 2020 → 7 Feb 2020 |
| Name | WSDM 2020 - Proceedings of the 13th International Conference on Web Search and Data Mining |
|---|
| Conference | 13th ACM International Conference on Web Search and Data Mining, WSDM 2020 |
|---|---|
| Country/Territory | United States |
| City | Houston |
| Period | 3/02/20 → 7/02/20 |
ID: 95169019