基于粒子群优化的因子分解机算法

作者：邱煜炎; 吴福生

来源：山西师范大学学报(自然科学版), 2020, 34(01): 5-11.

DOI：10.16207/j.cnki.1009-4490.2020.01.002

• 因子分解机
• 离子群优化
• 机器学习
• 推荐系统

因子分解机(Factorization Machine,FM)是利用矩阵分解思路针对推荐系统中数据稀疏情况设计的机器学习算法.传统的FM模型参数是通过梯度下降方法进行优化求解,但针对数据集训练样本较少的情况,梯度下降方法不能保证参数收敛到全局最优.粒子群算法(Particle Swarm Optimization,PSO)是一种快速启发式算法,具有全局搜索的特性.为提高FM模型的表现能力,首先基于PSO算法确定全局最优位置,然后利用梯度下降优化FM参数,本文提出了PSO-FM算法.在数据集Diabetes进行实验对比,结果表明,改进后的基于粒子群的因子分解机算法PSO-FM在模型训练速度和预测准确度上都优于传统的因子分解机FM算法.

• 单位
  蚌埠医学院