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

AdaKernel: Learning Adaptive Kernel Parameters for Spatiotemporal Graph Neural Networks

About

Modeling spatial dependencies is central to spatiotemporal data analysis using Graph Neural Networks (GNNs). Traditional methods rely on distance-based kernels with predefined parameters, which restricts model capacity. Although generic adaptive mechanisms (e.g., Graph Attention Networks) offer flexibility, they often fail to capture the underlying geometric structure, performing worse than distance-based models in data-sparse scenarios. Addressing this, we revisit the kernel parameterization problem and theoretically prove that misspecified kernel parameters introduce unavoidable approximation errors in GNNs. To overcome this, we propose AdaKernel, a simple yet effective approach that learns adaptive kernel parameters within the neural network. Unlike methods that learn graph structures from scratch, AdaKernel adopts a structure-preserving strategy that optimizes the scale of physical interactions rather than discarding them. Extensive experiments on Kriging, Imputation, and Forecasting demonstrate that AdaKernel consistently improves various GNN architectures and outperforms model-agnostic adaptive baselines, validating that accurately learned kernel parameters are superior to both fixed priors and fully latent graph structures.

Zhongyue Zhang, Guangyin Jin, Yuxuan Liang, Suwan Yin, Yuankai Wu• 2026

Related benchmarks

TaskDatasetResultRank
Traffic ForecastingMETR-LA
MAE3.007
329
Traffic ForecastingPeMS08
RMSE23.557
242
Traffic ForecastingPEMS-BAY
MAE1.576
85
Traffic ForecastingPeMS03
MAE14.559
58
Spatiotemporal ImputationMETR-LA Block missing
MAE1.849
4
Spatiotemporal ImputationPEMS-BAY Block missing
MAE1.007
4
Spatiotemporal ImputationPEMS-03 Block missing
MAE10.142
4
Spatiotemporal ImputationPEMS-04 Block missing
MAE17.55
4
Spatiotemporal ImputationPEMS-07
MAE14.185
4
Spatiotemporal ImputationPEMS-08 Block missing
MAE13.43
4
Showing 10 of 20 rows

Other info

Follow for update