atlas

Research map / Graph learning / Graph self-supervised learning

ReGCL: Rethinking Message Passing in Graph Contrastive Learning

AAAI 2024 · Graph learning › Graph self-supervised learning · approach: Graph contrastive learning analysis · 20 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Analyzes gradient conflicts between GNN message passing and InfoNCE in node-level graph contrastive learning, and proposes ReGCL with gradient-guided structure learning and gradient-weighted InfoNCE.

Problem

Node-level graph contrastive learning with GNN encoders suffers a conflict: message passing pulls neighbors together while InfoNCE pushes them apart, so some samples receive opposing gradients.

Approach

Derives gradients of InfoNCE with respect to input features (including message passing) to locate conflicting samples, then rewires the graph via a gradient-guided structure learning module (higher-order structure plus stop-gradient edge re-weighting) and down-weights likely false negatives with a gradient-weighted InfoNCE loss.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use ReGCL instead of CCA-SSGNode classification · Coraaccuracy: 84.2±0.4 -> 84.8±0.1Helped
Use ReGCL instead of CCA-SSGNode classification · Citeseeraccuracy: 73.1±0.3 -> 74.3±0.3Helped
Use ReGCL instead of CCA-SSGNode classification · Pubmedaccuracy: 81.6±0.4 -> 83.9±0.3Helped
Use ReGCL instead of CCA-SSGNode classification · Photoaccuracy: 93.1±0.1 -> 92.6±0.3Hurt
Use ReGCL instead of CCA-SSGNode classification · CSaccuracy: 93.3±0.2 -> 93.7±0.3Helped
Remove gradient-guided structure learning (GGSL) (ablation)ReGCL · node classification benchmarksaccuracyHurt
Replace gradient-weighted InfoNCE with normal InfoNCE (ablation)ReGCL · node classification benchmarksaccuracyHurt
Remove both GGSL and GW-NCE (ablation)ReGCL · node classification benchmarksaccuracyHurt
Remove higher-order structure enhancement from GGSL (ablation)ReGCL · node classification benchmarksaccuracyHurt
Keep original graph edges instead of re-wiring them (ablation)ReGCL · node classification benchmarksaccuracyHurt
Increase the neighbor order beyond the first order (ablation)ReGCL · node classification benchmarksaccuracy: 1.7%Helped
Use an overly large neighbor order k (ablation)ReGCL · node classification benchmarksaccuracyHurt

Surprising findings

Limitations

Experimental setup

Datasets: Cora Citeseer Pubmed Amazon Photo Coauthor CS

Benchmarks: Cora Citeseer Pubmed Amazon Photo Coauthor CS

Metrics: test accuracy

Baselines: GCN GAT DeepWalk GAE DGI GRACE MVGRL BGRL GCA CCA-SSG GRADE

Builds on: GRACE InfoNCE GCN

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy