arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.11467cs.IRcs.LG
收藏

冷启动联合训练舱网络推荐

Joint Training Capsule Network for Cold Start Recommendation

Hangzhou Dianzi University(杭州电子科技大学);University of Illinois at Chicago(伊利诺伊大学芝加哥分校)

Tingting Liang, Congying Xia, Yuyu Yin, Philip S. Yu

摘要:

本文提出了一种新型的神经网络--联合训练胶囊网络(JTCN),用于冷启动推荐任务。我们建议模仿高层次的用户偏好,而不是原始的交互历史的基础上的边信息的新鲜用户。具体而言,一个细心的胶囊层,提出了聚合高层次的用户偏好,从低层次的互动历史,通过一个动态路由协议机制。此外,JTCN以端到端的方式联合训练用于模仿用户偏好的损失和用于推荐的softmax损失。在两个公开数据集上的实验证明了该模型的有效性。在冷启动推荐中,JTCN在Recall@100方面对CiteULike和Amazon的其他最先进的方法分别提高了至少7.07%和16.85%。

英文摘要:

This paper proposes a novel neural network, joint training capsule network (JTCN), for the cold start recommendation task. We propose to mimic the high-level user preference other than the raw interaction history based on the side information for the fresh users. Specifically, an attentive capsule layer is proposed to aggregate high-level user preference from the low-level interaction history via a dynamic routing-by-agreement mechanism. Moreover, JTCN jointly trains the loss for mimicking the user preference and the softmax loss for the recommendation together in an end-to-end manner. Experiments on two publicly available datasets demonstrate the effectiveness of the proposed model. JTCN improves other state-of-the-art methods at least 7.07% for CiteULike and 16.85% for Amazon in terms of Recall@100 in cold start recommendation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑