-
题名融合注意力胶囊的深度因子分解机模型
被引量:1
- 1
-
-
作者
顾亦然
姚朱鹏
杨海根
-
机构
南京邮电大学自动化学院、人工智能学院
南京邮电大学智慧校园研究中心
南京邮电大学宽带无线通信技术教育部工程研究中心
-
出处
《通信学报》
EI
CSCD
北大核心
2021年第10期130-139,共10页
-
基金
国防基础科研基金资助项目(No.JCKY2019210B005,No.JCKY2018204B025,No.JCKY2017204B011)
国防重大工程基金资助项目(No.ZQ2019D20401)
装备发展部仿真预研课题(No.41401030301)。
-
文摘
针对深度学习中推荐模型特征组合单一、消解大量有价值特征信息以及过拟合等问题,设计了一种新型的注意力得分机制——注意力胶囊,提出了一种融合注意力胶囊的深度因子分解机模型。基于DeepFM模型,将用户历史点击行为与候选物品进行权重计算,降低了无关特征对模型的影响,充分挖掘了不同历史行为对用户兴趣的差异性影响。训练过程中加入自适应正则化式,在不影响训练速度的前提下,有效地减少了过拟合。在2个公开数据集上进行对比实验,实验结果表明,所提模型相对于其他模型在损失函数和GAUC上均有明显提升。
-
关键词
推荐模型
深度学习
注意力胶囊
因子分解机
自适应正则化
-
Keywords
recommendation model
deep learning
attention capsule
factorization machine
adaptive regularization
-
分类号
TP181
[自动化与计算机技术—控制理论与控制工程]
-