Please wait a minute...
Frontiers of Computer Science

ISSN 2095-2228

ISSN 2095-2236(Online)

CN 10-1014/TP

Postal Subscription Code 80-970

2018 Impact Factor: 1.129

Front. Comput. Sci.    2019, Vol. 13 Issue (6) : 1255-1265    https://doi.org/10.1007/s11704-018-8049-1
RESEARCH ARTICLE
Bayesian dual neural networks for recommendation
Jia HE1,2, Fuzhen ZHUANG1,2(), Yanchi LIU3, Qing HE1,2, Fen LIN4
1. Key Lab of Intelligent Information Processing of Chinese Academy of Sciences, Institute of Computing Technology, Chinese Academy of Sciences, Beijing 100190, China
2. University of Chinese Academy of Sciences, Beijing 100049, China
3. Rutgers University, Newark 07102, USA
4. Search Product Center, WeChat Search Application Department, Tencent, Beijing 100080, China
 Download: PDF(472 KB)  
 Export: BibTeX | EndNote | Reference Manager | ProCite | RefWorks
Abstract

Most traditional collaborative filtering (CF) methods only use the user-item rating matrix to make recommendations, which usually suffer from cold-start and sparsity problems. To address these problems, on the one hand, some CF methods are proposed to incorporate auxiliary information such as user/item profiles; on the other hand, deep neural networks, which have powerful ability in learning effective representations, have achieved great success in recommender systems. However, these neural network based recommendation methods rarely consider the uncertainty of weights in the network and only obtain point estimates of the weights. Therefore, they maybe lack of calibrated probabilistic predictions and make overly confident decisions. To this end, we propose a new Bayesian dual neural network framework, named BDNet, to incorporate auxiliary information for recommendation. Specifically, we design two neural networks, one is to learn a common low dimensional space for users and items from the rating matrix, and another one is to project the attributes of users and items into another shared latent space. After that, the outputs of these two neural networks are combined to produce the final prediction. Furthermore, we introduce the uncertainty to all weights which are represented by probability distributions in our neural networks to make calibrated probabilistic predictions. Extensive experiments on real-world data sets are conducted to demonstrate the superiority of our model over various kinds of competitors.

Keywords collaborative filtering      Bayesian neural network      hybrid recommendation algorithm     
Corresponding Author(s): Fuzhen ZHUANG   
Just Accepted Date: 02 July 2018   Online First Date: 17 December 2018    Issue Date: 19 July 2019
 Cite this article:   
Jia HE,Fuzhen ZHUANG,Yanchi LIU, et al. Bayesian dual neural networks for recommendation[J]. Front. Comput. Sci., 2019, 13(6): 1255-1265.
 URL:  
https://academic.hep.com.cn/fcs/EN/10.1007/s11704-018-8049-1
https://academic.hep.com.cn/fcs/EN/Y2019/V13/I6/1255
1 S Rendle, C Freudenthaler, Z Gantner, L Schmidt-Thieme. BPR: Bayesian personalized ranking from implicit feedback. In: Proceedings of the 25th Conference on Uncertainty in Artificial Intelligence. 2009, 452–461
2 R Salakhutdinov, A Mnih. Probabilistic matrix factorization. In: Proceedings of the 20th International Conference on Neural Information Processing Systems. 2007, 1257–1264
3 H J Xue, X Dai, J Zhang, S Huang, J Chen. Deep matrix factorization models for recommender systems. In: Proceedings of the 26th International Joint Conference on Artificial Intelligence. 2017, 3203–3209
https://doi.org/10.24963/ijcai.2017/447
4 Y Zhou, D Wilkinson, R Schreiber, R Pan. Large-scale parallel collaborative filtering for the netflix prize. In: Proceedings of the International Conference on Algorithmic Applications in Management. 2008, 337–348
https://doi.org/10.1007/978-3-540-68880-8_32
5 A P Singh, G J Gordon. Relational learning via collective matrix factorization. In: Proceedings of the 14th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. 2008, 650–658
https://doi.org/10.1145/1401890.1401969
6 H Ma, D Zhou, C Liu, M R Lyu, I King. Recommender systems with social regularization. In: Proceedings of the 4th ACM International Conference on Web Search and Data Mining. 2011, 287–296
https://doi.org/10.1145/1935826.1935877
7 A Singh, G J Gordon. A Bayesian matrix factorization model for relational data. In: Proceedings of the 26th Conference on Uncertainty in Artificial Intelligence. 2010, 556–563
8 Y Gao, X Wang, G Lei, Z Chen. Learning to recommend with collaborative matrix factorization for new users. Journal of Computer Research and Development, 2017, 54(8): 1813–1823
9 G E Hinton, S Osindero, Y W Teh. A fast learning algorithm for deep belief nets. Neural Computation, 2006, 18(7): 1527–1554
https://doi.org/10.1162/neco.2006.18.7.1527
10 G E Hinton, R R Salakhutdinov. Reducing the dimensionality of data with neural networks. Science, 2006, 313(5786): 504–507
https://doi.org/10.1126/science.1127647
11 S Li, J Kawale, Y Fu. Deep collaborative filtering via marginalized denoising auto-encoder. In: Proceedings of the 24th ACM International Conference on Information and Knowledge Management. 2015, 811–820
https://doi.org/10.1145/2806416.2806527
12 H Wang, N Wang, D Y Yeung. Collaborative deep learning for recommender systems. In: Proceedings of the 21st ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. 2015, 1235–1244
https://doi.org/10.1145/2783258.2783273
13 X He, L Liao, H Zhang, L Nie, X Hu, T S Chua. Neural collaborative filtering. In: Proceedings of the 26th International Conference on World Wide Web. 2017, 173–182
https://doi.org/10.1145/3038912.3052569
14 X Wang, Y Wang. Improving content-based and hybrid music recommendation using deep learning. In: Proceedings of the 22nd ACM International Conference on Multimedia. 2014, 627–636
https://doi.org/10.1145/2647868.2654940
15 J M Hernández-Lobato, R Adams. Probabilistic backpropagation for scalable learning of Bayesian neural networks. In: Proceedings of the International Conference on Machine Learning. 2015, 1861–1869
16 C Blundell, J Cornebise, K Kavukcuoglu, D Wierstra. Weight uncertainty in neural networks. In: Proceedings of the the 32nd International Conference on Machine Learning. 2015, 1613–1622
17 J Bobadilla, F Ortega, A Hernando, A Gutiérrez. Recommender systems survey. Knowledge-based Systems, 2013, 46(1): 109–132
https://doi.org/10.1016/j.knosys.2013.03.012
18 K Lang. Newsweeder: learning to filter netnews. In: Proceedings of the International Conference on Machine Learning. 1995, 331–339
https://doi.org/10.1016/B978-1-55860-377-6.50048-7
19 T Chen, W Zhang, Q Lu, K Chen, Z Zheng, Y Yu. SVDFeature: a toolkit for feature-based collaborative filtering. Journal of Machine Learning Research, 2012, 13(9): 3619–3622
20 L Hu, J Cao, G Xu, L Cao, Z Gu, C Zhu. Personalized recommendation via cross-domain triadic factorization. In: Proceedings of the 22nd International Conference on World Wide Web. 2013, 595–606
https://doi.org/10.1145/2488388.2488441
21 D Agarwal, B C Chen. Regression-based latent factor models. In: Proceedings of the 15th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. 2009, 19–28
https://doi.org/10.1145/1557019.1557029
22 W J Li, D Y Yeung, Z Zhang. Generalized latent factor models for social network analysis. In: Proceedings of the 22nd International Joint Conference on Artificial Intelligence. 2011, 1705–1710
23 R Salakhutdinov, A Mnih, G Hinton. Restricted boltzmann machines for collaborative filtering. In: Proceedings of the 24th International Conference on Machine learning. 2007, 791–798
https://doi.org/10.1145/1273496.1273596
24 K Georgiev, P Nakov. A non-IID framework for collaborative filtering with restricted boltzmann machines. In: Proceedings of the International Conference on Machine Learning. 2013, 1148–1156
25 A Graves. Practical variational inference for neural networks. In: Proceedings of the 24th International Conference on Neural Information Processing Systems. 2011, 2348–2356
26 R Salakhutdinov, A Mnih. Bayesian probabilistic matrix factorization using markov chain monte carlo. In: Proceedings of the the 25th International Conference on Machine Learning. 2008, 880–887
https://doi.org/10.1145/1390156.1390267
[1] Min GAO, Bin LING, Linda YANG, Junhao WEN, Qingyu XIONG, Shun LI. From similarity perspective: a robust collaborative filtering approach for service recommendations[J]. Front. Comput. Sci., 2019, 13(2): 231-246.
[2] Zhoujun LI, Haijun ZHANG, Senzhang WANG, Feiran HUANG, Zhenping LI, Jianshe ZHOU. Exploit latent Dirichlet allocation for collaborative filtering[J]. Front. Comput. Sci., 2018, 12(3): 571-581.
[3] Shaha AL-OTAIBI, Mourad YKHLEF. Hybrid immunizing solution for job recommender system[J]. Front. Comput. Sci., 2017, 11(3): 511-527.
[4] Qiang GUO , Jianguo LIU , Binghong WANG , . Improved collaborative filtering algorithm based on heat conduction[J]. Front. Comput. Sci., 2009, 3(3): 417-420.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed