A joint framework for collaborative filtering and metric learning

Tak-Lam WONG, Wai LAM, Haoran XIE, Fu Lee WANG

Research output: Book Chapters | Papers in Conference ProceedingsBook Chapter

1 Scopus Citations

Abstract

We have developed a framework for jointly conducting collaborative filtering and distance metric learning based on regularized singular value decomposition (RSVD), which discovers the user matrix and item matrix in the low rank space. Our approach is able to solve RSVD and simultaneously learn the parameters of Mahalanobis distance considering the ratings given by similar users and dissimilar users. One characteristic of our approach is that the learned model can be effectively applied to rating prediction and other relevant applications such as trust prediction, resulting in a solution which is coherent and optimal to both tasks. Another characteristic is that social community information and similarity information can be easily considered in our framework. We have conducted extensive experiments on rating prediction using real-world datasets to evaluate our framework. We have also compared our framework with other existing works to illustrate the effectiveness. Experimental results show that our framework achieves a promising prediction performance and outperforms the existing works.
Original languageEnglish
Title of host publicationInformation Retrieval Technology : 12th Asia Information Retrieval Societies Conference, AIRS 2016, Beijing, China, November 30–December 2, 2016, proceedings
EditorsShaoping MA, Ji-Rong WEN, Yiqun LIU, Zhicheng DOU, Min ZHANG, Yi CHANG, Xin ZHAO
PublisherSpringer International Publishing AG
Pages184-196
Number of pages13
ISBN (Electronic)9783319480510
ISBN (Print)9783319480503
DOIs
Publication statusPublished - 2016
Externally publishedYes
Event12th Asia Information Retrieval Societies Conference - Tsinghua University, Beijing, China
Duration: 30 Nov 20162 Dec 2016

Publication series

NameLecture Notes in Computer Science
Volume9994
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference12th Asia Information Retrieval Societies Conference
Abbreviated titleAIRS 2016
CountryChina
CityBeijing
Period30/11/162/12/16

    Fingerprint

Bibliographical note

The work described in this paper is substantially supported by grants from the Education University of Hong Kong (Project Codes: RG 30/2014-2015R and RG 18/2015-2016R).

Keywords

  • Collaborative filtering
  • Metric learning
  • Mahalanobis distance

Cite this

WONG, T-L., LAM, W., XIE, H., & WANG, F. L. (2016). A joint framework for collaborative filtering and metric learning. In S. MA, J-R. WEN, Y. LIU, Z. DOU, M. ZHANG, Y. CHANG, & X. ZHAO (Eds.), Information Retrieval Technology : 12th Asia Information Retrieval Societies Conference, AIRS 2016, Beijing, China, November 30–December 2, 2016, proceedings (pp. 184-196). (Lecture Notes in Computer Science; Vol. 9994). Springer International Publishing AG. https://doi.org/10.1007/978-3-319-48051-0_14