Collaborative metric learning

Research output: Contribution to journalConference articleResearchpeer-review

  • Cheng Kang Hsieh
  • Longqi Yang
  • Yin Cui
  • Tsung Yi Lin
  • Belongie, Serge
  • Deborah Estrin

Metric learning algorithms produce distance metrics that capture the important relationships among data. In this work, we study the connection between metric learning and collaborative filtering. We propose Collaborative Metric Learning (CML) which learns a joint metric space to encode not only users’ preferences but also the user-user and item-item similarity. The proposed algorithm outperforms state-of-the-art collaborative filtering algorithms on a wide range of recommendation tasks and uncovers the underlying spectrum of users’ fine-grained preferences. CML also achieves significant speedup for Top-K recommendation tasks using off-the-shelf, approximate nearest-neighbor search, with negligible accuracy reduction.

Original languageEnglish
Journal26th International World Wide Web Conference, WWW 2017
Pages (from-to)193-201
Number of pages9
DOIs
Publication statusPublished - 2017
Externally publishedYes
Event26th International World Wide Web Conference, WWW 2017 - Perth, Australia
Duration: 3 Apr 20177 Apr 2017

Conference

Conference26th International World Wide Web Conference, WWW 2017
CountryAustralia
CityPerth
Period03/04/201707/04/2017
SponsorBankwest, Curtin University, Edith Cowan University (ECU), et al., Murdoch University, University of Western Australia

Bibliographical note

Funding Information:
We appreciate the anonymous reviewers for their helpful comments and feedback. This research is partly funded by AOL-Program for Connected Experiences, a Google Focused Research Award and further supported by the small data lab at Cornell Tech which receives funding from UnitedHealth Group, Google, Pfizer, RWJF, NIH and NSF.

Publisher Copyright:
© 2017 International World Wide Web Conference Committee (IW3C2).

ID: 301827073