DHCL-BR: Dual Hypergraph Contrastive Learning for Bundle Recommendation

Peng Zhang, Zhendong Niu*, Ru Ma, Fuzhi Zhang*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

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.

Original languageEnglish
Pages (from-to)2906-2919
Number of pages14
JournalComputer Journal
Volume67
Issue number10
DOIs
Publication statusPublished - 1 Oct 2024

Fingerprint

Dive into the research topics of 'DHCL-BR: Dual Hypergraph Contrastive Learning for Bundle Recommendation'. Together they form a unique fingerprint.

Cite this