MultiCBR: Multi-view Contrastive Learning for Bundle Recommendation

Yunshan Ma,Yingzhi He,Xiang Wang,Yinwei Wei,Xiaoyu Du,Yuyangzi Fu,Tat-Seng Chua
DOI: https://doi.org/10.1145/3640810
IF: 4.657
2024-01-23
ACM Transactions on Information Systems
Abstract:Bundle recommendation seeks to recommend a bundle of related items to users to improve both user experience and the profits of platform. Existing bundle recommendation models have progressed from capturing only user-bundle interactions to the modeling of multiple relations among users, bundles and items. CrossCBR, in particular, incorporates cross-view contrastive learning into a two-view preference learning framework, significantly improving SOTA performance. It does, however, have two limitations: 1) the two-view formulation does not fully exploit all the heterogeneous relations among users, bundles and items; and 2) the ”early contrast and late fusion” framework is less effective in capturing user preference and difficult to generalize to multiple views. In this paper, we present MultiCBR, a novel Multi -view C ontrastive learning framework for B undle R ecommendation. First, we devise a multi-view representation learning framework capable of capturing all the user-bundle, user-item and bundle-item relations, especially better utilizing the bundle-item affiliations to enhance sparse bundles’ representations. Second, we innovatively adopt an ”early fusion and late contrast” design that first fuses the multi-view representations before performing self-supervised contrastive learning. In comparison to existing approaches, our framework reverses the order of fusion and contrast, introducing the following advantages: 1) our framework is capable of modeling both cross-view and ego-view preferences, allowing us to achieve enhanced user preference modeling; and 2) instead of requiring quadratic number of cross-view contrastive losses, we only require two self-supervised contrastive losses, resulting in minimal extra costs. Experimental results on three public datasets indicate that our method outperforms SOTA methods. The code and dataset can be found in the github repo https://github.com/HappyPointer/MultiCBR.
computer science, information systems
What problem does this paper attempt to address?
### Problems the Paper Attempts to Solve The main goal of this paper is to propose a new multi-view contrastive learning framework—MultiCBR—in the field of bundle recommendation. Specifically, the paper aims to address the following two main issues: 1. **Failure to Fully Utilize Multiple Heterogeneous Relationships**: Existing two-view methods (such as CrossCBR) fail to fully explore the multiple heterogeneous relationships among users, bundles, and items. Particularly for the relationship between bundles and items (BI relationship), using only a unidirectional aggregation approach cannot effectively leverage this information to enhance the representation of sparse bundles. 2. **Design Limitation of "Early Contrast and Late Fusion"**: Existing contrastive learning frameworks (such as CrossCBR) adopt an "early contrast and late fusion" design, which limits their performance in multi-view scenarios. This design not only overlooks cross-view user preferences but also leads to an exponential increase in the number of contrastive loss terms as the number of views increases, resulting in additional computational costs. To address the above issues, MultiCBR proposes a new framework that includes three views: user-bundle interaction view (UB), user-item interaction view (UI), and bundle-item association view (BI). Additionally, through the strategy of "early fusion and late contrast," this model can better capture user preference information with less computational overhead. Experimental results show that MultiCBR significantly outperforms existing state-of-the-art methods on three public datasets.