Gradient boosting factorization machines

Chen Cheng*, Fen Xia, Tong Zhang, Irwin King, Michael R. Lyu

*Corresponding author for this work

Research output: Chapter in Book/Conference Proceeding/ReportConference Paper published in a bookpeer-review

62 Citations (Scopus)

Abstract

Recommendation techniques have been well developed in the past decades. Most of them build models only based on user item rating matrix. However, in real world, there is plenty of auxiliary information available in recommendation systems. We can utilize these information as additional features to improve recommendation performance. We refer to recom- mendation with auxiliary information as context-aware rec- ommendation. Context-aware Factorization Machines (FM) is one of the most successful context-aware recommendation models. FM models pairwise interactions between all fea- tures, in such way, a certain feature latent vector is shared to compute the factorized parameters it involved. In prac- tice, there are tens of context features and not all the pair- wise feature interactions are useful. Thus, one important challenge for context-aware recommendation is how to effec- tively select \good" interaction features. In this paper, we focus on solving this problem and propose a greedy interac- tion feature selection algorithm based on gradient boosting. Then we propose a novel Gradient Boosting Factorization Machine (GBFM) model to incorporate feature selection al- gorithm with Factorization Machines into a unified frame- work. The experimental results on both synthetic and real datasets demonstrate the efficiency and effectiveness of our algorithm compared to other state-of-the-art methods.

Original languageEnglish
Title of host publicationRecSys 2014 - Proceedings of the 8th ACM Conference on Recommender Systems
PublisherAssociation for Computing Machinery
Pages265-272
Number of pages8
ISBN (Electronic)9781450326681
DOIs
Publication statusPublished - 6 Oct 2014
Externally publishedYes
Event8th ACM Conference on Recommender Systems, RecSys 2014 - Foster City, United States
Duration: 6 Oct 201410 Oct 2014

Publication series

NameRecSys 2014 - Proceedings of the 8th ACM Conference on Recommender Systems

Conference

Conference8th ACM Conference on Recommender Systems, RecSys 2014
Country/TerritoryUnited States
CityFoster City
Period6/10/1410/10/14

Bibliographical note

Publisher Copyright:
Copyright © 2014 ACM.

Keywords

  • Collaborative filtering
  • Factorization machines
  • Gradient boosting
  • Recommender systems

Fingerprint

Dive into the research topics of 'Gradient boosting factorization machines'. Together they form a unique fingerprint.

Cite this