Article Dans Une Revue Transactions on Machine Learning Research Journal Année : 2025

A Fused Gromov-Wasserstein Approach to Subgraph Contrastive Learning

Résumé

Self-supervised learning has become a key method for training deep learning models when labeled data is scarce or unavailable. While graph machine learning holds great promise across various domains, the design of effective pretext tasks for self-supervised graph representation learning remains challenging. Contrastive learning, a popular approach in graph self-supervised learning, leverages positive and negative pairs to compute a contrastive loss function. However, current graph contrastive learning methods often struggle to fully use structural patterns and node similarities. To address these issues, we present a new method called Fused Gromov-Wasserstein Subgraph Contrastive Learning (FOSSIL). Our method integrates node-level and subgraph-level contrastive learning, seamlessly combining a standard node-level contrastive loss with the Fused Gromov-Wasserstein distance. This combination helps our method capture both node features and graph structure together. Importantly, our approach works well with both homophilic and heterophilic graphs and can dynamically create views for generating positive and negative pairs. Through extensive experiments on benchmark graph datasets, we show that FOSSIL outperforms or achieves competitive performance compared to current state-of-the-art methods.

Fichier principal
Vignette du fichier
3177_A_Fused_Gromov_Wasserstei.pdf (908.08 Ko) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)
Licence

Dates et versions

hal-05166788 , version 1 (07-08-2025)

Licence

Identifiants

Citer

Amadou S. Sangare, Nicolas Dunou, Jhony H. Giraldo, Fragkiskos D. Malliaros. A Fused Gromov-Wasserstein Approach to Subgraph Contrastive Learning. Transactions on Machine Learning Research Journal, 2025. ⟨hal-05166788⟩
1433 Consultations
170 Téléchargements

Altmetric

Partager

  • More