An Intelligent Field-Aware Factorization Machine Model

Cairong Yan,Qinglong Zhang,Xue Zhao,Yongfeng Huang
DOI: https://doi.org/10.1007/978-3-319-55753-3_20
2017-01-01
Abstract:The widely-used field-aware factorization machines model (FFM) takes the interactions of all the text features into consideration which will lead to a large number of invalid calculations. An intelligent field-aware factorization machine model (iFFM) is proposed in this paper. In the model, the key attributes are promoted and the factor selection operations are embedded into the computation process intelligently by using the auto feature engineering technology. Meanwhile, Markov Chain Monte Carlo (MCMC) and stochastic gradient descent (SGD) methods are applied to optimize the loss function and improve the recommendation accuracy. In order to get better diversity, a new model iFFM-2 is put forward, which is the linear weighted combination of iFFM and a model built based on the heat-spreading algorithm. The experimental results show that iFFM can obtain higher accuracy and computation efficiency compared with FFMs, iFFM-2 inherits the accuracy of iFFM, and it can provide better diversity.
What problem does this paper attempt to address?