IRLGSIFeb 5, 2023

Adversarial Learning Data Augmentation for Graph Contrastive Learning in Recommendation

arXiv:2302.02317v112 citationsh-index: 45
Originality Incremental advance
AI Analysis

This work addresses data sparsity issues in GNN-based recommendation systems, offering an incremental improvement over existing GCL methods.

The paper tackles the problem of improper data augmentation hindering Graph Contrastive Learning (GCL) in recommendation systems by proposing a learnable data augmentation method guided by the InfoMin principle, resulting in improved performance demonstrated through extensive experiments on four benchmark datasets.

Recently, Graph Neural Networks (GNNs) achieve remarkable success in Recommendation. To reduce the influence of data sparsity, Graph Contrastive Learning (GCL) is adopted in GNN-based CF methods for enhancing performance. Most GCL methods consist of data augmentation and contrastive loss (e.g., InfoNCE). GCL methods construct the contrastive pairs by hand-crafted graph augmentations and maximize the agreement between different views of the same node compared to that of other nodes, which is known as the InfoMax principle. However, improper data augmentation will hinder the performance of GCL. InfoMin principle, that the good set of views shares minimal information and gives guidelines to design better data augmentation. In this paper, we first propose a new data augmentation (i.e., edge-operating including edge-adding and edge-dropping). Then, guided by InfoMin principle, we propose a novel theoretical guiding contrastive learning framework, named Learnable Data Augmentation for Graph Contrastive Learning (LDA-GCL). Our methods include data augmentation learning and graph contrastive learning, which follow the InfoMin and InfoMax principles, respectively. In implementation, our methods optimize the adversarial loss function to learn data augmentation and effective representations of users and items. Extensive experiments on four public benchmark datasets demonstrate the effectiveness of LDA-GCL.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes