DHCL-BR: Dual Hypergraph Contrastive Learning for Bundle Recommendation
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.
Authors
- Fuzhi Zhang (ORCID: https://orcid.org/0000-0002-9595-3589)
- Zhendong Niu
- Ru Ma (ORCID: https://orcid.org/0009-0008-8908-5034)
- Peng Zhang (ORCID: https://orcid.org/0009-0004-7146-1805)
Institutions
- Beijing Institute of Technology (CN)
- Beijing Electronic Science and Technology Institute (CN)
- Yanshan University (CN)
Publication Details
- Journal
- The Computer Journal
- Published
- 2024-08-09
- DOI
- https://doi.org/10.1093/comjnl/bxae056
- Citations
- 2
- Primary Topic
- Recommender Systems and Techniques
- Type
- article
- Field-Weighted Citation Impact
- 1.56
Funders
- National Natural Science Foundation of China
- National Key Research and Development Program of China