超图
捆绑
计算机科学
杠杆(统计)
图形
二部图
对偶(语法数字)
理论计算机科学
人工智能
机器学习
数学
离散数学
文学类
艺术
复合材料
材料科学
作者
Peng Zhang,Zhendong Niu,Ru Ma,Fuzhi Zhang
标识
DOI:10.1093/comjnl/bxae056
摘要
Abstract As an extension of conventional top-K item recommendation solution, bundle recommendation has aroused increasingly attention. However, because of the extreme sparsity of user-bundle (UB) interactions, the existing top-K item recommendation methods suffer from poor performance when applied to bundle recommendation. While some graph-based approaches have been proposed for bundle recommendation, these approaches primarily leverage the bipartite graph to model the UB interactions, resulting in suboptimal performance. In this paper, a dual hypergraph contrastive learning model is proposed for bundle recommendation. First, we model the direct and indirect UB interactions as hypergraphs to represent the higher-order UB relations. Second, we utilize the hypergraph convolution networks to learn the user and bundle embeddings from the hypergraphs, and improve the learned embeddings through a bidirectional contrastive learning strategy. Finally, we adopt a joint loss that combines the InfoBPR loss supporting multiple negative samples and the contrastive losses to optimize model parameters for prediction. Experiments on the real-world datasets indicate that our model performs better than the state-of-the-art baseline methods.
科研通智能强力驱动
Strongly Powered by AbleSci AI