Organizations: the Academy of Mathematics and Systems Science, Chinese Academy of Sciences and also with the University of Chinese Academy of Sciences, Beijing 100190, China · the School of Mathematics and Statistics, Shandong University, Weihai, Shandong 264209, China · School of Artificial Intelligence, Beihang University, Beijing 100191, China · the School of Mathematics, Shandong University, Jinan, Shandong 250100, China
Hypergraph neural networks (HGNNs) have demonstrated strong capabilities in modeling complex higher-order relationships. However, existing HGNNs often suffer from over-smoothing as the number of layers increases and lack effective control over message passing among nodes. Inspired by the theory of Ricci flow in differential geometry, we theoretically establish that introducing discrete Ricci flow into hypergraph structures can effectively regulate node feature evolution and thereby alleviate over-smoothing. Building on this insight, we propose Ricci Flow-guided Hypergraph Neural Diffusion(RFHND), a novel message passing paradigm for hypergraphs guided by discrete Ricci flow. Specifically, RFHND is based on a PDE system that describes the continuous evolution of node features on hypergraphs and adaptively regulates the rate of information diffusion at the geometric level, preventing feature homogenization and producing high-quality node representations. Experimental results show that RFHND significantly outperforms existing methods across multiple benchmark datasets and demonstrates strong robustness, while also effectively mitigating over-smoothing.
Figures & tables
Fig. 1 : Ricci flow-guided evolution. (a) Evolution of the manifold geometry under Ricci flow. (b) Evolution of hyperedge weight guided by Ricci flow. (c) Evolution of node feature distributions driven by the updated hyperedge weights.
Metric
Cora
Citeseer
Pubmed
Cora-CA
DBLP-CA
Zoo
NTU2012
ModelNet40
Walmart
Senate
House
∣V∣
2708
3312
19177
2708
41302
101
2012
12311
88860
282
1290
∣E∣
1579
1079
7963
1072
22363
43
2012
12311
69906
315
340
#features
1433
3703
500
1433
1425
16
100
100
100
100
100
#classes
7
6
3
7
6
7
67
40
11
2
2
TABLE I : Dataset Statistics Summary
Models
Cora
Citeseer
Pubmed
Cora-CA
DBLP-CA
Rank ↓
HNHN
76.36 ± 1.92
72.64 ± 1.57
86.90 ± 0.30
77.19 ± 1.49
86.78 ± 0.29
12
HGNN
79.39 ± 1.36
72.45 ± 1.16
86.44 ± 0.44
82.64 ± 1.65
91.03 ± 0.20
8
HCHA
79.14 ± 1.02
72.42 ± 1.42
86.41 ± 0.36
82.55 ± 0.97
90.92 ± 0.22
11
HyperGCN
78.45 ± 1.26
71.28 ± 0.82
82.84 ± 8.67
79.48 ± 2.08
89.38 ± 0.25
13
UniGCNII
78.81 ± 1.05
73.05 ± 2.21
88.25 ± 0.40
83.60 ± 1.14
91.69 ± 0.19
6
AllSetTransformer
78.59 ± 1.47
73.08 ± 1.20
88.72 ± 0.37
83.63 ± 1.47
91.53 ± 0.23
5
TABLE II : Performance comparison on academic hypergraph datasets (Mean accuracy (%) ± standard deviation), with the best results in bold, second-best and third-best are marked with an underline.
Models
Zoo
NTU2012
ModelNet40
Walmart
Senate
House
Rank ↓
HNHN
93.59 ± 5.88
89.11 ± 1.44
97.84 ± 1.25
81.35 ± 0.61
50.93 ± 6.33
67.80 ± 2.59
6
HGNN
92.50 ± 4.58
87.72 ± 1.35
95.44 ± 0.33
80.33 ± 0.42
48.59 ± 4.52
61.39 ± 2.96
11
HCHA
93.65 ± 6.15
87.48 ± 1.87
94.48 ± 0.28
80.33 ± 0.80
48.62 ± 4.41
61.36 ± 2.53
10
HyperGCN
N/A
56.36 ± 4.86
75.89 ± 5.26
81.05 ± 0.59
42.45 ± 3.67
48.32 ± 2.93
13
UniGCNII
93.65 ± 4.37
89.30 ± 1.33
98.07 ± 0.23
81.12 ± 0.67
49.30 ± 4.25
61.70 ± 3.37
7
AllSetTransformer
97.50 ± 3.59
88.69 ± 1.24
98.20 ± 0.20
81.38 ± 0.58
51.83 ± 5.22
69.33 ± 2.20
4
TABLE III : Performance comparison on real-world hypergraph datasets (Mean accuracy (%) ± standard deviation), with the best results in bold, second-best and third-best are marked with an underline.
Model
α=1
α=2
α=3
α=4
α=5
α=6
α=7
HGNN
92.58 ± 0.57
87.40 ± 0.89
82.32 ± 0.84
77.18 ± 1.08
70.06 ± 1.35
69.98 ± 0.85
68.41 ± 1.39
HyperGCN
83.40 ± 1.67
78.13 ± 1.57
77.70 ± 1.01
74.90 ± 1.31
50.15 ± 1.94
52.11 ± 0.75
49.40 ± 1.63
UniGCN
90.42 ± 1.04
82.12 ± 0.67
80.04 ± 0.98
76.86 ± 1.10
71.50 ± 1.50
52.30 ± 1.18
49.20 ± 0.74
EDGNN
93.88 ± 0.67
89.89 ± 0.59
83.52 ± 0.85
76.65 ± 1.21
72.18 ± 1.36
52.98 ± 0.78
49.65 ± 1.16
RFHND
95.63 ± 0.62
90.17 ± 0.67
84.21 ± 0.81
77.48 ± 0.98
74.26 ± 1.57
73.51 ± 1.07
73.46 ± 1.35
TABLE IV : Model Performance On Synthetic Hypergraphs with Controlled Heterophily α .
Fig. 2 : (a) shows the accuracy(%) with various depths on Cora. (b) shows the accuracy(%) with various depths on Cora-CA. (c) shows the accuracy(%) with various depths on Citeseer. (d) shows the dirichlet energy with various depths on Cora. (e) shows the dirichlet energy with various depths on Cora-CA. (f) shows the dirichlet energy with various depths on Citeseer.
Variants
Zoo
NTU2012
ModelNet40
w/o COS
97.16 ± 2.90
90.04 ± 1.09
96.78 ± 0.27
w/o HyperNet
96.89 ± 2.87
91.52 ± 1.12
97.42 ± 0.23
w/o C and H
96.43 ± 2.90
89.79 ± 1.30
96.86 ± 0.27
RFHND (Full Model)
97.70 ± 2.14
93.36 ± 1.18
98.56 ± 0.16
TABLE V : Ablation Study.
Fig. 3 : (a) shows the performance on Citeseer under feature gaussian noise. (b) shows the performance on Citeseer under feature uniform noise. (c) shows the performance on Citeseer under feature mask noise. (d) shows the performance on Cora-CA under structure noise.
Fig. 4 : (a) shows RFHND accuracy on NTU2012 and ModelNet40 with various hidden dimensions. (b) shows RFHND accuracy on NTU2012 and ModelNet40 with various τ .
Fig. 5 : (a)(b)(c) show the visualization of Pubmed node features at t=0 , t=1 and t=2 . (d)(e)(f) show the visualization of DBLP node features at t=0 , t=1 and t=2 .
School of Mathematics and Statistics, Shandong University, Weihai, Shandong 264209, China · Academy of Mathematics and Systems Science, Chinese Academy of Sciences and also with the University of Chinese Academy of Sciences, Beijing 100190, China
School of Mathematics and Statistics, Shandong University · Academy of Mathematics and Systems Science, Chinese Academy of Sciences · School of Mathematical Sciences, University of Chinese Academy of Sciences +2
Beijing University of Posts and Telecommunications, Beijing, China · North China Electric Power University, Beijing, China · University of Kent, Canterbury, United Kingdom +2