← 返回资源分享
Variational Autoencoders for Collaborative Filtering
论文
论文
发布时间2018-02-16
发表arXiv:1802.05814
作者:Dawen Liang,Rahul G. Krishnan,Matthew D. Hoffman,Tony Jebara
详细介绍
We extend variational autoencoders (VAEs) to collaborative filtering for
implicit feedback. This non-linear probabilistic model enables us to go beyond
the limited modeling capacity of linear factor models which still largely
dominate collaborative filtering research.We introduce a generative model with
multinomial likelihood and use Bayesian inference for parameter estimation.
Despite widespread use in language modeling and economics, the multinomial
likelihood receives less attention in the recommender systems literature. We
introduce a different regularization parameter for the learning objective,
which proves to be crucial for achieving competitive performance. Remarkably,
there is an efficient way to tune the parameter using annealing. The resulting
model and learning algorithm has information-theoretic connections to maximum
entropy discrimination and the information bottleneck principle. Empirically,
we show that the proposed approach significantly outperforms several
state-of-the-art baselines, including two recently-proposed neural network
approaches, on several real-world datasets. We also provide extended
experiments comparing the multinomial likelihood with other commonly used
likelihood functions in the latent factor collaborative filtering literature
and show favorable results. Finally, we identify the pros and cons of employing
a principled Bayesian inference approach and characterize settings where it
provides the most significant improvements.
代码仓库 (18)
dawenl/vae_cf官方
mkfilipiuk/VAE-CFTensorFlow
amoussawi/recoderPyTorch
jaywonchung/BERT4Rec-VAE-PytorchPyTorch
jvbalen/autoencoders_cfPyTorch
onurboyar/RecommenderSystems
hasteck/MRF_NeurIPS_2019
anonymouspap/latte_recsysPyTorch
cydonia999/variational-autoencoders-for-collaborative-filtering-pytorchPyTorch
rushhan/VAE-for-Collaborative-Filtering-PytorchPyTorch
