ICML 2024

Navigating Complexity: Toward Lossless Graph Condensation via Expanding Window Matching

Yuchen Zhang, Tianle Zhang, Kai Wang, Ziyao Guo, Yuxuan Liang, Xavier Bresson, Wei Jin, Yang You

Main figure for Navigating Complexity: Toward Lossless Graph Condensation via Expanding Window Matching

Abstract

Graph condensation aims to reduce the size of a large-scale graph dataset by synthesizing a compact counterpart without sacrificing the performance of Graph Neural Networks (GNNs) trained on it, which has shed light on reducing the computational cost for training GNNs. Nevertheless, existing methods often fall short of accurately replicating the original graph for certain datasets, thereby failing to achieve the objective of lossless condensation. To understand this phenomenon, we investigate the potential reasons and reveal that the previous state-of-the-art trajectory matching method provides biased and restricted supervision signals from the original graph when optimizing the condensed one. This significantly limits both the scale and efficacy of the condensed graph. In this paper, we make the first attempt toward lossless graph condensation by bridging the previously neglected supervision signals. Specifically, we employ a curriculum learning strategy to train expert trajectories with more diverse supervision signals from the original graph, and then effectively transfer the information into the condensed graph with expanding window matching. Moreover, we design a loss function to further extract knowledge from the expert trajectories. Theoretical analysis justifies the design of our method and extensive experiments verify its superiority across different datasets. Code is released at https://github.com/NUS-HPC-AI-Lab/GEOM.

Proposed method: GEOM

Benchmark Results

Tested on 1 dataset 2 baseline values reported too low

Tables show presented results in global context (best 3 papers + GCN + MLP).

Classic

#ModelAccuracy ↑Paper
1
GraphSAGE+CNA
0.9418 ±0.0033
2
GCN+CNA
0.9366 ±0.0048
3
TransformerConv
0.9350 ±0.0058
21
GCN
0.9004
215
GEOM
actual #215 vs paper #434 219 places worse
0.8527
best reported
actual #215 vs paper #434 219 places worse
paper 0.8290 ±0.0100
560
SFGC
actual #560 vs paper #867 307 places worse
0.8170 ±0.0050
best reported
actual #560 vs paper #867 307 places worse
paper 0.6430 ±0.0160
best today: Simple Graph Condensation (2024)
paper 0.6430 ±0.0160 Δ at pub 0.1740 Δ today 0.1740
615
GCond
actual #615 vs paper #871 256 places worse
0.8090 ±0.0320
best reported
actual #615 vs paper #871 256 places worse
paper 0.6140 ±0.0290
724
MLP
0.7768
878
DC-Graph
0.5930 ±0.0160
This paper
SOTA performance (not significantly different from the best) Baseline / comparison (this paper) Global context (other papers) Best at pub — reported result worse than the best known when published (best reported in green)