上下文分解机的自适应更新策略  

Adaptive Parameters Updating Strategy of Context-aware Factorization Machines

作　　者：姚杏 朱福喜[1] 阳小兰[2] 郑麟[1] 刘世超[1] 

机构地区：[1]武汉大学计算机学院,武汉430072 [2]武昌理工学院信息工程学院,武汉430223

出　　处：《计算机科学》2016年第9期71-76,共6页Computer Science

基　　金：国家自然科学基金(61272277);湖北省自然科学基金(2014CFB356)资助

摘　　要：分解机模型已经被成功应用于上下文推荐系统。在分解机模型的学习算法中,交替最小二乘法是一种固定其他参数只求单一参数最优值的学习算法,其参数数目影响计算复杂度。然而当特征数目很大时,参数数目随着特征数目急剧增加,导致计算复杂度很高;即使有些参数已经达到了最优值,每次迭代仍更新所有的参数。因此,主要改进了交替最小二乘法的参数更新策略,为参数引入自适应误差指标,通过权重和参数绝对误差共同决定该参数更新与否,使得每次迭代时重点更新最近两次迭代取值变化较大的参数。这种仅更新自适应误差大于阈值的参数的策略不但减少了需要更新的参数数目,进而加快了算法收敛的速度和缩短了运行时间,而且参数权重由误差决定,又修正了误差。在Yahoo和Movielens数据集上的实验结果证明:改进的参数更新策略运行效率有明显提高。Context-aware factorization machine has been successfully applied in the context-aware recommendation sys- tem. In the learning algorithm of factorization machines, alternating least-squares is a learning algorithm that fixes other parameters just to find the optimal value of a single parameter, and the number of parameters and the sample size will affect the computational complexity. However, when the number of features is large, the number of parameters increases along with the increase of the number of features, resulting in high computational complexity. Even though some parame- ters have achieved the optimal value, all parameters will be updated in each iteration. This paper mainly improved the para-meters updating strategy of alternating least-squares. Adaptive error index was introduced into the parameter. Up- dating the parameter or not is co-determined by the weights and the absolute error of parameters, so that each iteration focuses on parameters whose last two iterative values change greatly. This strategy only updates parameters whose adaptive errors are greater than the thresholds. It not only reduces the number of parameters that need to be updated, so as to accelerate the algorithm convergence speed and shorten the operation time, but also the weight of parameters is de- termined by the error, to correct the error. The results of experiments on Yahoo and Movielens data sets show that the effect of the improved parameter updating strategy is better.

关 键 词：分解机模型 交替最小二乘法 推荐系统 自适应误差 

分 类 号：TP181[自动化与计算机技术—控制理论与控制工程]