ABSTRACT
In this paper, we introduce Factorization Machines (FM) which are a new model class that combines the advantages of Support Vector Machines (SVM) with factorization models.
由上可知,FM模型是一种新的模型,其综合了SVM模型与factorization模型的优点。
factorization models的基本形式为: f(x)=q1(x)q2(x)q3(x)...qt(x) 。
Like SVMs, FMs are a general predictor working with any real valued feature vector. In contrast to SVMs, FMs model all interactions between variables using factorized parameters. Thus they are able to estimate interactions even in problems with huge sparsity (like recommender systems) where SVMs fail.
在特征较为稀疏的情况下,如推荐系统等,SVM模型不再适用
为了解决该问题,FM模型引入factorized parameters,该参数用于对交叉特征
进行学习
We show that the model equation of FMs can be calculated in linear time and thus FMs can be optimized directly. So unlike nonlinear SVMs, a transformation in the dual form is not necessary and the model parameters can be estimated directly without the need of any support vector in the solution. We show the relationship to SVMs and the advantages of FMs for parameter estimation in sparse settings.
FM模型避免了SVM模型在训练时的弊端,其模型复杂度为 O(n)
On the other hand there are many different factorization models like matrix factorization
, parallel factor analysis
or specialized models
like SVD++, PITF or FPMC. The drawback of these models is that they are not applicable for general prediction tasks but work only with special input data. Furthermore their model equations and optimization algorithms are derived individually for each task.
factorization models的两个缺点:
- 对输入的数据有限制,如推荐系统中,输入数据的形式为:
uid, sid, score
;- 模型及其所采用的优化方法需要根据具体的task进行定制;
We show that FMs can mimic these models just by specifying the input data (i.e. the feature vectors). This makes FMs easily applicable even for users without expert knowledge in factorization models.
I. INTRODUCTION
Support Vector Machines are one of the most popular predictors in machine learning and data mining. Nevertheless in settings like collaborative filtering, SVMs play no important role and the best models are either direct applications of standard matrix/ tensor factorization models like PARAFAC [1] or specialized models using factorized parameters [2], [3], [4].
In this paper, we show that the only reason why standard SVM predictors are not successful in these tasks is that they cannot learn reliable parameters (‘hyperplanes’) in complex (non-linear) kernel spaces under very sparse data.
重点关注论文中,如何证明为什么non-linear SVM不适用于稀疏数据集。
On the other hand, the drawback of tensor factorization models and even more for specialized factorization models is that
(1) they are not applicable to standard prediction data (e.g. a real valued feature vector in ℝn .)
(2) that specialized models are usually derived individually for a specific task requiring effort in modeling and design of a learning algorithm.
In this paper, we introduce a new predictor, the Factorization Machine (FM), that is a general predictor like SVMs but is also able to estimate reliable parameters under very high sparsity.
The factorization machine models all nested variable interactions(comparable to a polynomial kernel in SVM), but uses a factorized parametrization
instead of a dense parametrization
like in SVMs.
polynomial kernel:多项式核函数,形式为 Kn(X,X′)=(1+γXTX′)n,其中γ>0
polynomial kernel-SVM模型,其模型参数采用
dense parametrization
,而FM模型参数采用factorized parametrization
We show that the model equation of FMs can be computed in linear time and that it depends only on a linear number of parameters. This allows direct optimization and storage of model parameters without the need of storing any training data (e.g. support vectors) for prediction. In contrast to this, non-linear SVMs are usually optimized in the dual form and computing a prediction (the model equation) depends on parts of the training data (the support vectors).
We also show that FMs subsume many of the most successful approaches for the task of collaborative filtering including biased MF, SVD++ [2], PITF [3] and FPMC [4].
In total, the advantages of our proposed FM are:
1) FMs allow parameter estimation under very sparse data where SVMs fail.
2) FMs have linear complexity, can be optimized in the primal and do not rely on support vectors like SVMs. We show that FMs scale to large datasets like Netflix with 100 millions of training instances.
3) FMs are a general predictor that can work with any real valued feature vector. In contrast to this, other state-of-the-art factorization models work only on very restricted input data. We will show that just by defining the feature vectors of the input data, FMs can mimic state-of-the-art models like biased MF, SVD++, PITF or FPMC.
II. PREDICTION UNDER SPARSITY
The most common prediction task is to estimate a function y:ℝn→T from a real valued feature vector x∈ℝn to a target domain T (e.g.
In this paper, we deal with problems where x is highly sparse, i.e. almost all of the elements
Example 1 Assume we have the transaction data of a movie review system. The system records which user u∈U rates a movie (item) i∈I at a certain time t∈R with a rating r∈{
1,2,3,4,5} . Let the users U and items
Let the observed data S be:
An example for a prediction task using this data, is to estimate a function yˆ that predicts the rating behavior of a user for an item at a certain point in time.
Figure 1 shows one example of how feature vectors can be created from S for this task. Here, first there are
We will use this example data throughout the paper for illustration. However please note that FMs are general predictors like SVMs and thus are applicable to any real valued feature vectors and are not restricted to recommender systems.
III. FACTORIZATION MACHINES (FM)
In this section, we introduce factorization machines. We discuss the model equation in detail and show shortly how to apply FMs to several prediction tasks.
A. Factorization Machine Model
1) Model Equation: The model equation for a factorization machine of degree d=2 is defined as:
where the model parameters that have to be estimated are:
A row vi within V describes the i -th variable with
A 2-way FM (degree d=2 ) captures all single and pairwise interactions between variables:
- w0 is the global bias.
- wi models the strength of the i -th variable.
-
wˆi,j:=⟨vi,vj⟩ models the interaction between the i -th andj -th variable. Instead of using an own model parameter wi,j∈ℝ for each interaction, the FM models the interaction by factorizing it. We will see later on, that this is the key point which allows high quality parameter estimates of higher order interactions ( d≥2 ) under sparsity.
2) Expressiveness: It is well known that for any positive definite matrix W , there exists a matrix V such that W=