Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Enhanced Graph Neural Networks using K-Hop Gaussian Diffusion

About

Most graph neural network (GNN) cores rely on graph convolutions, typically implemented as message passing between direct (single-hop) neighbors. In many real-world graphs, edges can be noisy or poorly defined, limiting information propagation to local neighborhoods. Existing diffusion kernels, such as Personalized PageRank (PPR) and Heat Kernel, alleviate this issue through global propagation, but still struggle with complex local structures and distant node noise. To address these limitations, we propose a K-Hop Gaussian (KHG) diffusion kernel as a preprocessing module for graph data. KHG introduces multi-hop diffusion with Gaussian weighting for remote nodes, balancing local and global information propagation before applying standard GNNs. Experiments on multiple benchmark datasets demonstrate that KHG significantly outperforms traditional message-passing GNNs, as well as PPR and Heat Kernel diffusion, particularly in noisy or structurally complex graphs.

Xuling Zhang, Peng Wang, Daiyan Li, Aoran Huang, Zeiwei Chen, Yongkui Yang• 2026

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy75.08
1383
Graph ClassificationMUTAG
Accuracy92.98
1229
Node ClassificationChameleon
Accuracy48.86
936
Node ClassificationCiteseer
Accuracy71.3
541
Graph ClassificationENZYMES
Accuracy31.33
419
Node ClassificationCora
Accuracy82.04
43
Showing 6 of 6 rows

Other info

Follow for update