Mixed Supervised Graph Contrastive Learning for Recommendation

2404.15954

YC

0

Reddit

0

Published 4/29/2024 by Weizhi Zhang, Liangwei Yang, Zihe Song, Henry Peng Zou, Ke Xu, Yuanjie Zhu, Philip S. Yu

👨‍🏫

Abstract

Recommender systems (RecSys) play a vital role in online platforms, offering users personalized suggestions amidst vast information. Graph contrastive learning aims to learn from high-order collaborative filtering signals with unsupervised augmentation on the user-item bipartite graph, which predominantly relies on the multi-task learning framework involving both the pair-wise recommendation loss and the contrastive loss. This decoupled design can cause inconsistent optimization direction from different losses, which leads to longer convergence time and even sub-optimal performance. Besides, the self-supervised contrastive loss falls short in alleviating the data sparsity issue in RecSys as it learns to differentiate users/items from different views without providing extra supervised collaborative filtering signals during augmentations. In this paper, we propose Mixed Supervised Graph Contrastive Learning for Recommendation (MixSGCL) to address these concerns. MixSGCL originally integrates the training of recommendation and unsupervised contrastive losses into a supervised contrastive learning loss to align the two tasks within one optimization direction. To cope with the data sparsity issue, instead unsupervised augmentation, we further propose node-wise and edge-wise mixup to mine more direct supervised collaborative filtering signals based on existing user-item interactions. Extensive experiments on three real-world datasets demonstrate that MixSGCL surpasses state-of-the-art methods, achieving top performance on both accuracy and efficiency. It validates the effectiveness of MixSGCL with our coupled design on supervised graph contrastive learning.

Create account to get full access

or

If you already have an account, we'll log you in

Overview

  • Recommender systems play a crucial role in online platforms, helping users navigate vast information with personalized suggestions.
  • Graph contrastive learning aims to learn from high-order collaborative filtering signals using unsupervised augmentation on the user-item bipartite graph.
  • This relies on a multi-task learning framework involving both the pair-wise recommendation loss and the contrastive loss, which can lead to inconsistent optimization directions and longer convergence times.
  • The self-supervised contrastive loss also falls short in addressing the data sparsity issue in recommender systems.

Plain English Explanation

Recommender systems are like personal assistants that help us find things we might like on online platforms. They look at the things we've interacted with before and suggest other things we might enjoy, kind of like a friend recommending a book or movie they think we'd like.

The paper proposes a new way to train these recommender systems using a technique called "graph contrastive learning." This approach tries to find patterns in the connections between users and items, like how certain users tend to like certain types of items. It does this by looking at the user-item network and finding ways to "contrast" or highlight the differences between users and items.

However, the traditional way of doing this can sometimes cause problems. The different training objectives (the recommendation task and the contrastive task) can pull the system in different directions, making it take longer to converge on a good set of recommendations. Additionally, the contrastive approach doesn't do a great job of addressing the common issue of "data sparsity" in recommender systems, where there isn't a lot of information about what each user likes.

Technical Explanation

To address these issues, the paper introduces a new approach called "Mixed Supervised Graph Contrastive Learning for Recommendation" (MixSGCL). MixSGCL integrates the training of the recommendation and contrastive losses into a single "supervised contrastive learning" loss, aligning the two tasks within a single optimization direction.

To tackle the data sparsity problem, MixSGCL proposes using "node-wise" and "edge-wise" mixup techniques, which mix existing user-item interactions to generate new, supervised collaborative filtering signals, rather than relying on unsupervised augmentation.

Experiments on three real-world datasets show that MixSGCL outperforms state-of-the-art methods, achieving top performance in terms of both accuracy and efficiency. This validates the effectiveness of the coupled design of supervised graph contrastive learning in MixSGCL.

Critical Analysis

The paper addresses valid concerns with the traditional approach to graph contrastive learning for recommender systems. The proposed MixSGCL method seems promising in aligning the optimization directions of the recommendation and contrastive tasks, as well as generating more informative supervised signals to address data sparsity.

However, the paper does not provide extensive analysis of the computational complexity or runtime efficiency of MixSGCL compared to other methods. Additionally, the paper could have explored the generalizability of the approach to other types of graph-based recommender systems or multi-modal retrieval and ranking tasks.

Further research could also investigate the impact of different subgraph contrastive learning strategies or heterogeneous graph neural networks on the performance of MixSGCL, as well as its applicability to semi-supervised domain adaptation on graphs.

Conclusion

This paper presents a novel approach called MixSGCL that addresses key challenges in using graph contrastive learning for recommender systems. By aligning the recommendation and contrastive learning objectives and generating more informative supervised signals, MixSGCL demonstrates improved accuracy and efficiency compared to state-of-the-art methods.

The insights from this research could help advance the field of recommender systems, enabling online platforms to provide users with even more personalized and relevant suggestions amidst the vast amount of available information. As the digital world continues to grow, innovations in recommender systems will play an increasingly vital role in helping people navigate and discover content tailored to their interests.



This summary was produced with help from an AI and may contain inaccuracies - check out the links to read the original source documents!

Related Papers

Towards Graph Contrastive Learning: A Survey and Beyond

Wei Ju, Yifan Wang, Yifang Qin, Zhengyang Mao, Zhiping Xiao, Junyu Luo, Junwei Yang, Yiyang Gu, Dongjie Wang, Qingqing Long, Siyu Yi, Xiao Luo, Ming Zhang

YC

0

Reddit

0

In recent years, deep learning on graphs has achieved remarkable success in various domains. However, the reliance on annotated graph data remains a significant bottleneck due to its prohibitive cost and time-intensive nature. To address this challenge, self-supervised learning (SSL) on graphs has gained increasing attention and has made significant progress. SSL enables machine learning models to produce informative representations from unlabeled graph data, reducing the reliance on expensive labeled data. While SSL on graphs has witnessed widespread adoption, one critical component, Graph Contrastive Learning (GCL), has not been thoroughly investigated in the existing literature. Thus, this survey aims to fill this gap by offering a dedicated survey on GCL. We provide a comprehensive overview of the fundamental principles of GCL, including data augmentation strategies, contrastive modes, and contrastive optimization objectives. Furthermore, we explore the extensions of GCL to other aspects of data-efficient graph learning, such as weakly supervised learning, transfer learning, and related scenarios. We also discuss practical applications spanning domains such as drug discovery, genomics analysis, recommender systems, and finally outline the challenges and potential future directions in this field.

Read more

5/21/2024

Dual-perspective Cross Contrastive Learning in Graph Transformers

Dual-perspective Cross Contrastive Learning in Graph Transformers

Zelin Yao, Chuang Liu, Xueqi Ma, Mukun Chen, Jia Wu, Xiantao Cai, Bo Du, Wenbin Hu

YC

0

Reddit

0

Graph contrastive learning (GCL) is a popular method for leaning graph representations by maximizing the consistency of features across augmented views. Traditional GCL methods utilize single-perspective i.e. data or model-perspective) augmentation to generate positive samples, restraining the diversity of positive samples. In addition, these positive samples may be unreliable due to uncontrollable augmentation strategies that potentially alter the semantic information. To address these challenges, this paper proposed a innovative framework termed dual-perspective cross graph contrastive learning (DC-GCL), which incorporates three modifications designed to enhance positive sample diversity and reliability: 1) We propose dual-perspective augmentation strategy that provide the model with more diverse training data, enabling the model effective learning of feature consistency across different views. 2) From the data perspective, we slightly perturb the original graphs using controllable data augmentation, effectively preserving their semantic information. 3) From the model perspective, we enhance the encoder by utilizing more powerful graph transformers instead of graph neural networks. Based on the model's architecture, we propose three pruning-based strategies to slightly perturb the encoder, providing more reliable positive samples. These modifications collectively form the DC-GCL's foundation and provide more diverse and reliable training inputs, offering significant improvements over traditional GCL methods. Extensive experiments on various benchmarks demonstrate that DC-GCL consistently outperforms different baselines on various datasets and tasks.

Read more

6/4/2024

🔎

Community-Invariant Graph Contrastive Learning

Shiyin Tan, Dongyuan Li, Renhe Jiang, Ying Zhang, Manabu Okumura

YC

0

Reddit

0

Graph augmentation has received great attention in recent years for graph contrastive learning (GCL) to learn well-generalized node/graph representations. However, mainstream GCL methods often favor randomly disrupting graphs for augmentation, which shows limited generalization and inevitably leads to the corruption of high-level graph information, i.e., the graph community. Moreover, current knowledge-based graph augmentation methods can only focus on either topology or node features, causing the model to lack robustness against various types of noise. To address these limitations, this research investigated the role of the graph community in graph augmentation and figured out its crucial advantage for learnable graph augmentation. Based on our observations, we propose a community-invariant GCL framework to maintain graph community structure during learnable graph augmentation. By maximizing the spectral changes, this framework unifies the constraints of both topology and feature augmentation, enhancing the model's robustness. Empirical evidence on 21 benchmark datasets demonstrates the exclusive merits of our framework. Code is released on Github (https://github.com/ShiyinTan/CI-GCL.git).

Read more

5/3/2024

Unifying Graph Convolution and Contrastive Learning in Collaborative Filtering

Unifying Graph Convolution and Contrastive Learning in Collaborative Filtering

Yihong Wu, Le Zhang, Fengran Mo, Tianyu Zhu, Weizhi Ma, Jian-Yun Nie

YC

0

Reddit

0

Graph-based models and contrastive learning have emerged as prominent methods in Collaborative Filtering (CF). While many existing models in CF incorporate these methods in their design, there seems to be a limited depth of analysis regarding the foundational principles behind them. This paper bridges graph convolution, a pivotal element of graph-based models, with contrastive learning through a theoretical framework. By examining the learning dynamics and equilibrium of the contrastive loss, we offer a fresh lens to understand contrastive learning via graph theory, emphasizing its capability to capture high-order connectivity. Building on this analysis, we further show that the graph convolutional layers often used in graph-based models are not essential for high-order connectivity modeling and might contribute to the risk of oversmoothing. Stemming from our findings, we introduce Simple Contrastive Collaborative Filtering (SCCF), a simple and effective algorithm based on a naive embedding model and a modified contrastive loss. The efficacy of the algorithm is demonstrated through extensive experiments across four public datasets. The experiment code is available at url{https://github.com/wu1hong/SCCF}. end{abstract}

Read more

6/24/2024