Collaborative Item Embedding Model for Implicit Feedback Data

  • ThaiBinh NguyenEmail author
  • Kenro Aihara
  • Atsuhiro Takasu
Conference paper
Part of the Lecture Notes in Computer Science book series (LNCS, volume 10360)


Collaborative filtering is the most popular approach for recommender systems. One way to perform collaborative filtering is matrix factorization, which characterizes user preferences and item attributes using latent vectors. These latent vectors are good at capturing global features of users and items but are not strong in capturing local relationships between users or between items. In this work, we propose a method to extract the relationships between items and embed them into the latent vectors of the factorization model. This combines two worlds: matrix factorization for collaborative filtering and item embedding, a similar concept to word embedding in language processing. Our experiments on three real-world datasets show that our proposed method outperforms competing methods on top-n recommendation tasks.


Recommender system Collaborative filtering Matrix factorization Item embedding 



This work was supported by a JSPS Grant-in-Aid for Scientific Research (B) (15H02789).


  1. 1.
    Koren, Y.: Factorization meets the neighborhood: a multifaceted collaborative filtering model. In: 14th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 426–434. ACM (2008)Google Scholar
  2. 2.
    Koren, Y.: Collaborative filtering with temporal dynamics. Commun. ACM 53(4), 89–97 (2010)CrossRefGoogle Scholar
  3. 3.
    Salakhutdinov, R., Mnih, A.: Probabilistic matrix factorization. In: Advances in Neural Information Processing Systems, vol. 20 (2008)Google Scholar
  4. 4.
    Hu, Y., Koren, Y., Volinsky, C.: Collaborative filtering for implicit feedback datasets. In: Eighth IEEE International Conference on Data Mining 2008, ICDM 2008, pp. 263–272. IEEE (2008)Google Scholar
  5. 5.
    Bell, R.M., Koren, Y.: Lessons from the netflix prize challenge. SIGKDD Explor. 9(2), 75–79 (2007)CrossRefGoogle Scholar
  6. 6.
    Pan, R., Zhou, Y., Cao, B., Liu, N.N., Lukose, R.M., Scholz, M., Yang, Q.: One-class collaborative filtering. In: IEEE International Conference on Data Mining (ICDM 2008), pp. 502–511 (2008)Google Scholar
  7. 7.
    Mikolov, T., Sutskever, I., Chen, K., Corrado, G.S., Dean, J.: Distributed representations of words and phrases and their compositionality. In: NIPS, pp. 3111–3119 (2013)Google Scholar
  8. 8.
    Mikolov, T., Chen, K., Corrado, G., Dean, J.: Efficient estimation of word representations in vector space. arXiv preprint arXiv:1301.3781 (2013)
  9. 9.
    Levy, O., Goldberg, Y.: Neural word embedding as implicit matrix factorization. In: Advances in Neural Information Processing Systems 27, pp. 2177–2185. Curran Associates Inc. (2014)Google Scholar
  10. 10.
    Le, Q.V., Mikolov, T.: Distributed representations of sentences and documents. ICML 14, 1188–1196 (2014)Google Scholar
  11. 11.
    Bengio, Y., Ducharme, R., Vincent, P., Jauvin, C.: A neural probabilistic language model. J. Mach. Learn. Res. 3(Feb), 1137–1155 (2003)zbMATHGoogle Scholar
  12. 12.
    Church, K.W., Hanks, P.: Word association norms, mutual information, and lexicography. Comput. Linguist. 1(16), 22–29 (1990)Google Scholar
  13. 13.
    Liang, D., Altosaar, J., Charlin, L., Blei, D.M.: Factorization meets the item embedding: regularizing matrix factorization with item co-occurrence. In: RecSys, pp. 59–66. ACM (2016)Google Scholar
  14. 14.
    Maxwell Harper, F., Konstan, J.A.: The movielens datasets: history and context. TiiS 5(4), 19 (2016)Google Scholar
  15. 15.
    Chen, D., Sain, S.L., Guo, K.: Data mining for the online retail industry: a case study of rfm model-based customer segmentation using data mining. J. Database Mark. Customer Strategy Manage. 19(3), 197–208 (2012)CrossRefGoogle Scholar
  16. 16.
    Gopalan, P., Hofman, J.M., Blei, D.M.: Scalable recommendation with poisson factorization. CoRR, abs/1311.1704 (2013)Google Scholar
  17. 17.
    Singh, A.P., Gordon, G.J.: Relational learning via collective matrix factorization. In: KDD, pp. 650–658. ACM (2008)Google Scholar
  18. 18.
    Zheng, X., Ding, H., Mamitsuka, H., Zhu, S.: Collaborative matrix factorization with multiple similarities for predicting drug-target interactions. In: KDD, pp. 1025–1033. ACM (2013)Google Scholar

Copyright information

© Springer International Publishing AG 2017

Authors and Affiliations

  • ThaiBinh Nguyen
    • 1
    Email author
  • Kenro Aihara
    • 2
  • Atsuhiro Takasu
    • 2
  1. 1.Department of InformaticsSOKENDAI (The Graduate University for Advanced Studies)HayamaJapan
  2. 2.National Institute of InformaticsChiyodaJapan

Personalised recommendations