Bayesian Deep Collaborative Matrix Factorization


  • Teng Xiao Sun Yat-sen University
  • Shangsong Liang Sun Yat-sen University
  • Weizhou Shen Sun Yat-sen University
  • Zaiqiao Meng Sun Yat-sen University



In this paper, we propose a Bayesian Deep Collaborative Matrix Factorization (BDCMF) algorithm for collaborative filtering (CF). BDCMF is a novel Bayesian deep generative model that learns user and item latent vectors from users’ social interactions, contents of items as the auxiliary information and user-item rating (feedback) matrix. It alleviates the problem of matrix sparsity by incorporating items’ auxiliary and users’ social information into the model. It can learn more robust and dense latent representations by integrating deep learning into Bayesian probabilistic framework. As being one of deep generative models, it has both non-linearity and Bayesian nature. Additionally, in BDCMF, we derive an efficient EM-style point estimation algorithm for parameter learning. To further improve recommendation performance, we also derive a full Bayesian posterior estimation algorithm for inference. Experiments conducted on two sparse datasets show that BDCMF can significantly outperform the state-of-the-art CF methods.




How to Cite

Xiao, T., Liang, S., Shen, W., & Meng, Z. (2019). Bayesian Deep Collaborative Matrix Factorization. Proceedings of the AAAI Conference on Artificial Intelligence, 33(01), 5474-5481.



AAAI Technical Track: Machine Learning