ICML 2025

Toward Data-centric Directed Graph Learning: An Entropy-driven Approach

Xunkai Li, Zhengyu Wu, Kaichi Yu, Hongchao Qin, Guang Zeng, Ronghua Li, Guoren Wang

Main figure for Toward Data-centric Directed Graph Learning: An Entropy-driven Approach

Abstract

The directed graph (digraph), as a generalization of undirected graphs, exhibits superior representation capability in modeling complex topology systems and has garnered considerable attention in recent years. Despite the notable efforts made by existing DiGraph Neural Networks (DiGNNs) to leverage directed edges, they still fail to comprehensively delve into the abundant data knowledge concealed in the digraphs. This data-level limitation results in model-level sub-optimal predictive performance and underscores the necessity of further exploring the potential correlations between the directed edges (topology) and node profiles (feature and labels) from a data-centric perspective, thereby empowering model-centric neural networks with stronger encoding capabilities. In this paper, we propose Entropy-driven Digraph knowlEdge distillatioN (EDEN), which can serve as a data-centric digraph learning paradigm or a model-agnostic hot-and-plug data-centric Knowledge Distillation (KD) module. The core idea is to achieve data-centric ML, guided by our proposed hierarchical encoding theory for structured data. Specifically, EDEN first utilizes directed structural measurements from a topology perspective to construct a coarse-grained Hierarchical Knowledge Tree (HKT). Subsequently, EDEN quantifies the mutual information of node profiles to refine knowledge flow in the HKT, enabling data-centric KD supervision within model training. As a general framework, EDEN can also naturally extend to undirected scenarios and demonstrate satisfactory performance. In our experiments, EDEN has been widely evaluated on 14 (di)graph datasets (homophily and heterophily) and across 4 downstream tasks. The results demonstrate that EDEN attains SOTA performance and exhibits strong improvement for prevalent (Di)GNNs.

Benchmark Results

Tested on 2 datasets Non-standard metric on 1 dataset

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

Classic

Heterophily Benchmark

Tolokers

Global leaderboard →
Non-default metric. This paper reports Accuracy; the dataset default is ROC-AUC. Values below use Accuracy only.

Sorted by Accuracy. Highlighted rows are from other papers.

#ModelAccuracy ↑Paper
1
NT*
0.8591 ±0.0083
2
Polynormer
0.8591 ±0.0074
3
Dir-NT
0.8502 ±0.0077
7
GCN
0.8364 ±0.0067
11
EDEN
0.8130 ±0.0020
This paper
13
HoloNet
0.7940 ±0.0020
This paper
14
D-HYPR
0.7920 ±0.0020
This paper
15
DiGCN
0.7910 ±0.0030
This paper
16
MagNet
0.7900 ±0.0020
This paper
17
DIMPA
0.7890 ±0.0020
This paper
18
GATv2
0.7880 ±0.0020
This paper
19
DGCN
0.7870 ±0.0030
This paper
23
GCNII
0.7850 ±0.0010
This paper
24
AGT
0.7850 ±0.0020
This paper
28
MLP
0.7816 ±0.0002
SOTA performance (not significantly different from the best) Proposed method (this paper) Baseline / comparison (this paper) Global context (other papers)