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
415
Traffic ForecastingPeMS08
RMSE23.557
266
Traffic ForecastingPEMS-BAY
MAE1.576
161
Traffic ForecastingPeMS03
MAE14.559
68
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