Gradient boosting factorization machines

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

Research output: Chapter in Book/Report/Conference proceedingConference contribution

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 (US)
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
StatePublished - Oct 6 2014
Externally publishedYes
Event8th ACM Conference on Recommender Systems, RecSys 2014 - Foster City, United States
Duration: Oct 6 2014Oct 10 2014

Publication series

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

Other

Other8th ACM Conference on Recommender Systems, RecSys 2014
Country/TerritoryUnited States
CityFoster City
Period10/6/1410/10/14

Keywords

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

ASJC Scopus subject areas

  • Computer Science Applications
  • Information Systems

Fingerprint

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

Cite this