A joint framework for collaborative filtering and metric learning

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

Research output: Contribution to conferencePapers

1 Citation (Scopus)

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
Publication statusPublished - Dec 2016
EventThe 12th Asia Information Retrieval Societies Conference - Tsinghua University, Beijing, China
Duration: 30 Nov 201602 Dec 2016
http://airs2016.ruc.edu.cn/

Conference

ConferenceThe 12th Asia Information Retrieval Societies Conference
Abbreviated titleAIRS 2016
Country/TerritoryChina
CityBeijing
Period30/11/1602/12/16
Internet address

Citation

Wong, T.-L., Lam, W., Xie, H., & Wang, F. L. (2016, December). A joint framework for collaborative filtering and metric learning. Paper presented at The 2016 Asian Information Retrieval Societies Conference (AIRS 2016), Tsinghua University, Beijing, China.

Keywords

  • Collaborative filtering
  • Metric learning
  • Mahalanobis distance

Fingerprint

Dive into the research topics of 'A joint framework for collaborative filtering and metric learning'. Together they form a unique fingerprint.