Factorization machines with libfm

S Rendle - ACM Transactions on Intelligent Systems and …, 2012 - dl.acm.org
ACM Transactions on Intelligent Systems and Technology (TIST), 2012dl.acm.org
Factorization approaches provide high accuracy in several important prediction problems,
for example, recommender systems. However, applying factorization approaches to a new
prediction problem is a nontrivial task and requires a lot of expert knowledge. Typically, a
new model is developed, a learning algorithm is derived, and the approach has to be
implemented. Factorization machines (FM) are a generic approach since they can mimic
most factorization models just by feature engineering. This way, factorization machines …
Factorization approaches provide high accuracy in several important prediction problems, for example, recommender systems. However, applying factorization approaches to a new prediction problem is a nontrivial task and requires a lot of expert knowledge. Typically, a new model is developed, a learning algorithm is derived, and the approach has to be implemented.
Factorization machines (FM) are a generic approach since they can mimic most factorization models just by feature engineering. This way, factorization machines combine the generality of feature engineering with the superiority of factorization models in estimating interactions between categorical variables of large domain. libFM is a software implementation for factorization machines that features stochastic gradient descent (SGD) and alternating least-squares (ALS) optimization, as well as Bayesian inference using Markov Chain Monto Carlo (MCMC). This article summarizes the recent research on factorization machines both in terms of modeling and learning, provides extensions for the ALS and MCMC algorithms, and describes the software tool libFM.
ACM Digital Library