Organizations: State Key Laboratory of Blockchain and Data Security, Zhejiang University · College of Computer Science and Technology, Zhejiang University · State Key Laboratory of Clean Energy Utilization, Zhejiang University
Accurate air quality forecasting is essential for public health and environmental sustainability, but remains challenging due to the complex pollutant dynamics. Existing deep learning methods often model pollutant dynamics as an instantaneous process, overlooking the intrinsic delays in pollutant propagation. Thus, we propose AirDDE, the first neural delay differential equation framework in this task that integrates delay modeling into a continuous-time pollutant evolution under physical guidance. Specifically, two novel components are introduced: (1) a memory-augmented attention module that retrieves globally and locally historical features, which can adaptively capture delay effects modulated by multifactor data; and (2) a physics-guided delay evolving function, grounded in the diffusion-advection equation, that models diffusion, delayed advection, and source/sink terms, which can capture delay-aware pollutant accumulation patterns with physical plausibility. Extensive experiments on three real-world datasets demonstrate that AirDDE achieves the state-of-the-art forecasting performance with an average MAE reduction of 8.79% over the best baselines. The code is available at https://github.com/w2obin/airdde-aaai.
Figures & tables
Figure 1: The architecture of AirDDE.
Dataset
# Factors
# Locations
Time Range
Granularity
KnowAir
18
184
1.1.2015-12.31.2018
3h
China-AQI
8
209
1.1.2017-4.30.2019
1h
US-PM
8
175
1.1.2020-12.31.2021
1h
Table 1: Dataset statistics.
Dataset
KnowAir
China-AQI
US-PM
Metric
MAE
RMSE
SMAPE
MAE
RMSE
MAPE
MAE
RMSE
MAPE
STGNNs
DCRNN (2018)
24.02 ∗
37.87 ∗
0.53 ∗
24.78
38.05
35.76
7.49
9.72
15.85
STGCN (2018)
23.64 ∗
32.48 ∗
0.52 ∗
23.87
37.29
35.03
6.99
8.81
14.72
ASTGCN (2019)
19.92 ∗
31.39 ∗
0.44 ∗
21.91
36.02
34.28
4.86
7.53
12.93
MTGNN (2020)
18.92
30.34
0.41
21.56
35.80
34.65
4.63
7.03
12.85
PM25GNN (2020)
19.32 ∗
30.12 ∗
0.43 ∗
22.01
36.21
34.12
5.24
7.86
13.43
Table 2: Results of AirDDE and baselines. The best results are bolded, and the second best results are underlined. The results with ∗ are cited from AirDualODE ( Tian et al. 2025 ) , while others are rerun using their official codes under multifactor settings.
Variant
MAA
PDE
AirDDE
w/o MAA
w/o GM
w/o LM
w/o PDE
w/o SST
w/o ATT
1st
15.45
15.13
15.37
15.97
15.70
15.00
14.53
2nd
20.18
18.47
17.80
19.93
18.86
19.82
17.16
3rd
21.65
19.57
18.54
21.47
19.79
21.33
18.01
AVG
19.16
17.80
17.44
19.39
18.18
18.78
16.92
Table 3: MAE results of the ablation study on KnowAir.
Figure 2: Results of the long-term study on China-AQI.
Dataset
Time lag
Number of global memory unit
0
1
2
3
8
16
32
64
China-AQI
18.54
17.03
19.87
21.05
19.66
18.49
17.03
18.78
US-PM
4.32
3.98
3.53
4.06
3.93
3.53
4.07
4.25
Table 4: Results of the hyperparameter study on China-AQI.
Figure 3: Station distribution of China-AQI (left) and US-PM (right).
Method
Conf.
GPU Memory (GB)
Training Time (Min/Epoch)
MAE
STGODE
KDD 2021
14.88
6.37
20.53
STG-NCDE
AAAI 2022
5.89
39.14
21.33
AirFormer
AAAI 2023
8.78
2.95
19.60
PDFormer
AAAI 2023
15.71
5.04
19.07
STDDE
WWW 2024
21.43
24.64
21.04
SGODE
AAAI 2024
12.22
11.06
20.17
Table 5: Results of the efficiency study on China-AQI.
Method
Original
Missing Rate
SNR
10%
30%
50%
80db
60db
40db
AirFormer
19.17
23.09
27.83
38.44
24.68
31.85
45.84
PDFormer
19.06
22.43
29.86
41.23
25.86
36.93
48.42
STDDE
22.85
26.11
32.54
40.78
28.75
35.92
49.34
AirPhyNet
21.31
25.20
31.30
37.79
27.82
33.60
46.42
AirDualODE
18.64
21.10
25.83
34.82
23.46
31.14
38.42
Table 6: MAE Results of the robustness study on KnowAir.
Figure 4: Case of city-wise advection with delay effects.
Figure 5: Case of region-wise advection with delay effects.
Accurate air quality forecasting is crucial for protecting public health and guiding environmental policy, yet it remains challenging due to nonlinear spatiotemporal dynamics, wind-driven transport, and distribution shifts across regions. Physics-based models are interpretable but computationally expensive and often rely on restrictive assumptions, whereas purely data-driven models can be accurate but may lack robustness and calibrated uncertainty. To address these limitations, we propose Neural Dynamic Diffusion-Advection Fields (NeuroDDAF), a physics-informed forecasting framework that unifies neural representation learning with open-system transport modeling. NeuroDDAF integrates (i) a GRU-Graph Attention encoder to capture temporal dynamics and wind-aware spatial interactions, (ii) a Fourier-domain diffusion-advection module with learnable residuals, (iii) a wind-modulated latent Neural ODE to model continuous-time evolution under time-varying connectivity, and (iv) an evidential fusion mechanism that adaptively combines physics-guided and neural forecasts while quantifying uncertainty. Experiments on four urban datasets (Beijing, Shenzhen, Tianjin, and Ancona) across 1-3 day horizons show that NeuroDDAF consistently outperforms strong baselines, including AirPhyNet, achieving up to 9.7% reduction in RMSE and 9.4% reduction in MAE on long-term forecasts. On the Beijing dataset, NeuroDDAF attains an RMSE of 41.63 μg/m3 for 1-day prediction and 48.88 μg/m3 for 3-day prediction, representing the best performance among all compared methods. In addition, NeuroDDAF improves cross-city generalization and yields well-calibrated uncertainty estimates, as confirmed by ensemble variance analysis and case studies under varying wind conditions.
Accurate air quality forecasting is essential for public health and urban environmental management, but remains challenging because pollutant channels differ in periodicity and distribution drift, while their concentration trajectories contain both multi-scale dependencies and rapid changes. Recent methods have improved spatial dependency learning and meteorological covariate modeling. However, pollutant channels are still passed through the same normalization rule and temporal backbone, using a shared latent representation for channel-specific distributions and changes at different rates. To address this limitation, we propose AirFlow, a pollutant-aware dual-stream framework that operates on station multivariate observations without additional graph propagation or predefined signal decomposition. Specifically, AirFlow designs two novel blocks: (1) a statistic-guided normalization routing mechanism that selects a normalization path for each pollutant according to its 24-hour autocorrelation and distribution drift; and (2) a hierarchical dual-stream state model that combines multi-scale state space propagation with learnable response coefficients, where gated bidirectional cross-attention exchanges information and adaptively fuses the resulting representations. Experiments on real-world data from multiple cities show that AirFlow achieves the best performance in 34 of 36 metrics comparisons, with reductions of up to 11.11% root mean square error over the state-of-the-art baseline. AirFlow also requires only 0.0483M parameters and 0.0215G FLOPs, achieving high forecasting accuracy with low computational overhead.
Fan Yang, Nan Chen, Yijie Dong +2
School of Software Technology, Zhejiang University Hangzhou, Zhejiang 310000, China
Accurate long-term air quality forecasting is challenging due to long-range transport, meteorological variability, and nonlinear chemical reactions underlying secondary pollutant formation. Purely data-driven models often suffer from long-horizon drift and lack physical consistency, while existing physics-informed approaches commonly follow open-loop designs that fuse neural and physical components only at the output level. We propose AeroChem, a closed-loop physics-informed state space framework for chemically reactive air quality forecasting. AeroChem adopts a two-branch co-correction design: a UDE-based process branch provides a structured but imperfect physical prior, while a Mamba-based virtual measurement branch captures long-range temporal context and produces probabilistic virtual observations. Neural Kalman Fusion couples the two branches by correcting the latent physical state at each forecasting step and feeding the posterior state back into the simulator, reducing long-horizon drift. We further introduce HanoiAir, a real-world dataset integrating pollutant concentrations, meteorology, and emission inventories. Experiments on HanoiAir and a public benchmark show that AeroChem improves long-horizon stability and sudden-change forecasting.