English

Discrete Factorization Machines for Fast Feature-based Recommendation

Information Retrieval 2018-09-20 v3 Machine Learning Machine Learning

Abstract

User and item features of side information are crucial for accurate recommendation. However, the large number of feature dimensions, e.g., usually larger than 10^7, results in expensive storage and computational cost. This prohibits fast recommendation especially on mobile applications where the computational resource is very limited. In this paper, we develop a generic feature-based recommendation model, called Discrete Factorization Machine (DFM), for fast and accurate recommendation. DFM binarizes the real-valued model parameters (e.g., float32) of every feature embedding into binary codes (e.g., boolean), and thus supports efficient storage and fast user-item score computation. To avoid the severe quantization loss of the binarization, we propose a convergent updating rule that resolves the challenging discrete optimization of DFM. Through extensive experiments on two real-world datasets, we show that 1) DFM consistently outperforms state-of-the-art binarized recommendation models, and 2) DFM shows very competitive performance compared to its real-valued version (FM), demonstrating the minimized quantization loss. This work is accepted by IJCAI 2018.

Keywords

Cite

@article{arxiv.1805.02232,
  title  = {Discrete Factorization Machines for Fast Feature-based Recommendation},
  author = {Han Liu and Xiangnan He and Fuli Feng and Liqiang Nie and Rui Liu and Hanwang Zhang},
  journal= {arXiv preprint arXiv:1805.02232},
  year   = {2018}
}

Comments

Appeared in IJCAI 2018

R2 v1 2026-06-23T01:46:29.018Z