Graph Contrastive Learning with Wasserstein Distance for Recommendation

Jiecheng Sun,Jianmin Li,Ying Ma
DOI: https://doi.org/10.1109/icsp58490.2023.10248760
2023-01-01
Abstract:Contrastive learning has been applied to graph collaborative filtering in recent years as a support task to boost efficiency. The majority of these approaches ignore global differences in favor of local variations among subgraphs. Additionally, different data augmentation strategies need to be developed for various datasets. Further, similarity metrics based on vector inner products might obscure sample-to-sample structural variations. To achieve this, we propose a method that adaptively samples subgraphs first, then uses a Wasserstein-based subgraph similarity measure to capture subgraph overall differences. To be more precise, we sample subgraph nodes from close to far, create subgraph edges using a similarity measure, and then build a contrastive loss based on Wasserstein distance to capture the distinction between pairs. Through extensive experimentation, we show the proposed method to be effective.
What problem does this paper attempt to address?