Graph Contrastive Learning versus Untrained Baselines: The Role of Dataset Size

Smayan Khanna, Doruk Efe Gökmen, Risi Kondor, Vincenzo Vitelli
9/1/2025
cs.LGcond-mat.soft

No official code found. Best implementation: SmayanKhanna/scale_GCL (0 stars, last commit 2025-09-03).

Abstract

Graph Contrastive Learning (GCL) has emerged as a leading paradigm for self-supervised learning on graphs, with strong performance reported on standardized datasets and growing applications ranging from genomics to drug discovery. We ask a basic question: does GCL actually outperform untrained baselines? We find that GCL's advantage depends strongly on dataset size and task difficulty. On standard datasets, untrained Graph Neural Networks (GNNs), simple multilayer perceptrons, and even handcrafted statistics can rival or exceed GCL. On the large molecular dataset ogbg-molhiv, we observe a crossover: GCL lags at small scales but pulls ahead beyond a few thousand graphs, though this gain eventually plateaus. On synthetic datasets, GCL accuracy approximately scales with the logarithm of the number of graphs and its performance gap (compared with untrained GNNs) varies with respect to task complexity. Moving forward, it is crucial to identify the role of dataset size in benchmarks and applications, as well as to design GCL algorithms that avoid performance plateaus.

View on arXivView PDF

Code Implementations(1)

A repository for our paper: Graph Contrastive Learning versus Untrained Baselines: The Role of Dataset Size

00Shell, PythonSep 2, 20251 years ago

Similar Papers

Cite this paper

@article{khanna2025graph,
  title  = {Graph Contrastive Learning versus Untrained Baselines: The Role of Dataset Size},
  author = {Smayan Khanna and Doruk Efe Gökmen and Risi Kondor and Vincenzo Vitelli},
  year   = {2025},
  eprint = {2509.01541},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2509.01541v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2509.01541.svg)](https://codeofpaper.com/paper/2509.01541)

Discussion