arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2308.03470cs.IR
收藏

面向冷启动物品推荐的不确定性感知一致性学习

Uncertainty-aware Consistency Learning for Cold-Start Item Recommendation

Taichi Liu, Chen Gao, Zhenyu Wang, Dong Li, Jianye Hao, Depeng Jin, Yong Li

AI总结:

针对冷启动推荐中冷热物品性能难以同时提升的问题,本文提出仅基于交互数据的不确定性感知一致性学习框架UCC,通过生成低不确定性交互使冷物品分布接近热物品,在基准数据集上平均性能提升27.6%。

摘要:

基于图神经网络(Graph Neural Network, GNN)的模型已成为推荐系统的主流方法。尽管其效果显著,这类模型仍面临冷启动问题,即为交互较少的物品进行推荐。现有用于解决冷启动问题的GNN推荐模型主要侧重于利用用户和物品的辅助特征,而用户-物品交互信息仍未得到充分利用。然而,冷物品与热物品的嵌入分布仍然存在较大差异,因为冷物品的嵌入是从低流行度交互中学习得到的,而热物品的嵌入则来自高流行度交互。因此存在一种“跷跷板”现象,即冷物品和热物品的推荐性能无法同时提升。为此,我们提出了一种仅基于用户-物品交互的不确定性感知一致性学习框架,用于冷启动物品推荐(简称UCC)。在该框架下,我们通过一致性学习训练教师模型(生成器)和学生模型(推荐器),以确保冷物品在额外生成低不确定性交互后,能够具有与热物品相似的分布。因此,所提框架能够同时提升冷物品和热物品的推荐效果,而不会损害其中任何一方。在基准数据集上的大量实验表明,我们提出的方法在热物品和冷物品上均显著优于最先进的方法,平均性能提升达27.6%。

英文摘要:

Graph Neural Network (GNN)-based models have become the mainstream approach for recommender systems. Despite the effectiveness, they are still suffering from the cold-start problem, i.e., recommend for few-interaction items. Existing GNN-based recommendation models to address the cold-start problem mainly focus on utilizing auxiliary features of users and items, leaving the user-item interactions under-utilized. However, embeddings distributions of cold and warm items are still largely different, since cold items' embeddings are learned from lower-popularity interactions, while warm items' embeddings are from higher-popularity interactions. Thus, there is a seesaw phenomenon, where the recommendation performance for the cold and warm items cannot be improved simultaneously. To this end, we proposed a Uncertainty-aware Consistency learning framework for Cold-start item recommendation (shorten as UCC) solely based on user-item interactions. Under this framework, we train the teacher model (generator) and student model (recommender) with consistency learning, to ensure the cold items with additionally generated low-uncertainty interactions can have similar distribution with the warm items. Therefore, the proposed framework improves the recommendation of cold and warm items at the same time, without hurting any one of them. Extensive experiments on benchmark datasets demonstrate that our proposed method significantly outperforms state-of-the-art methods on both warm and cold items, with an average performance improvement of 27.6%.

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

↑