arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1805.02232cs.IRcs.LGstat.ML

Discrete Factorization Machines for Fast Feature-based Recommendation

Shandong University(山东大学);National University of Singapore(新加坡国立大学);University of Electronic Science and Technology of China(电子科技大学);Nanyang Technological University(南洋理工大学)

Han Liu, Xiangnan He, Fuli Feng, Liqiang Nie, Rui Liu, Hanwang Zhang

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑