Skip to main navigation Skip to search Skip to main content

Dual-Supervised Contrastive Learning for Bundle Recommendation

Research output: Contribution to journal › Article › peer-review

Abstract

Self-supervised contrastive learning has well advanced the development of bundle recommendation. However, self-supervised contrastive learning may misclassify some positive samples as negative samples, resulting in suboptimal models. Therefore, the industry has proposed supervised contrastive learning to alleviate this drawback. Inspired by this idea, we seek a more elegant contrastive learning paradigm in the field of recommendation, so we propose a dual-supervised contrastive learning for bundle recommendation (DSCBR), which integrates supervised and self-supervised contrastive learning to exploit the full potential of contrastive learning. Specifically, we first construct self-supervised contrastive learning between the two different views (bundle view and item view), which encourages the alignment of the two separately learned views and boosts the effectiveness of learned representations. Second, we use the interaction information to construct supervised contrastive learning and leverage the bundle-bundle cooccurrence graph for further enhancement. By introducing supervised contrastive learning, our model explicitly models user and bundle proximity in different views, improving the model's robustness and generalization. Finally, we jointly perform self-supervised and supervised contrastive learning across multiple views. Extensive experiments on three public datasets demonstrate the effectiveness of our model.

Original languageEnglish
Pages (from-to)3955-3965
Number of pages11
JournalIEEE Transactions on Computational Social Systems
Volume11
Issue number3
DOIs
Publication statusPublished - Jun 2024

Keywords

  • bundle recommendation
  • self-supervised contrastive learning
  • supervised contrastive learning

Fingerprint

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

Cite this