Articles | Volume 19, issue 20
https://doi.org/10.5194/gmd-19-10263-2026
https://doi.org/10.5194/gmd-19-10263-2026
Development and technical paper
 | 
28 Oct 2026
Development and technical paper |  | 28 Oct 2026

GPU-accelerated three-dimensional variational data assimilation with automatic differentiation for WRF-Chem

Hancheng Ye, Zengliang Zang, Wei You, Yiwen Hu, Ning Liu, and Yi Li

Related authors

Impacts of direct CALIPSO Level 1B attenuated-backscatter assimilation on three-dimensional aerosol analyses and forecasts over northern China
Wei You, Zengliang Zang, Daichun Wang, Wei Zhong, Shuang Mei, Yiwen Hu, Jianping Guo, and Dan Niu
EGUsphere, https://doi.org/10.5194/egusphere-2026-5573,https://doi.org/10.5194/egusphere-2026-5573, 2026
This preprint is open for discussion and under review for Atmospheric Chemistry and Physics (ACP).
Short summary
A Coupled Data Assimilation Method for Visibility and Relative Humidity CAOP v1.0 module within the WRFDA v4.3 framework: Improving both PM2.5 and Humidity Forecasts
Lina Gao, Junli Jin, Zhiquan Liu, Wei You, Xiang Xie, and Peng Yan
EGUsphere, https://doi.org/10.5194/egusphere-2026-5270,https://doi.org/10.5194/egusphere-2026-5270, 2026
This preprint is open for discussion and under review for Geoscientific Model Development (GMD).
Short summary
Improvement of the computational efficiency in SVD-3DEnVar data assimilation scheme and its preliminary application to the TRAMS 3.0 model
Kun Liu, Daosheng Xu, Fei Zheng, Juanxiong He, Chun Li, Jeremy Cheuk-Hin Leung, Mingyang Zhang, Dingchi Zhao, Quanjun He, Yuewei Zhang, Yi Li, and Banglin Zhang
Geosci. Model Dev., 19, 2279–2297, https://doi.org/10.5194/gmd-19-2279-2026,https://doi.org/10.5194/gmd-19-2279-2026, 2026
Short summary
TemDeep: a self-supervised framework for temporal downscaling of atmospheric fields at arbitrary time resolutions
Liwen Wang, Qian Li, Qi Lv, Xuan Peng, and Wei You
Geosci. Model Dev., 18, 2427–2442, https://doi.org/10.5194/gmd-18-2427-2025,https://doi.org/10.5194/gmd-18-2427-2025, 2025
Short summary
Impacts of meteorology and emission reductions on haze pollution during the lockdown in the North China Plain
Lang Liu, Xin Long, Yi Li, Zengliang Zang, Fengwen Wang, Yan Han, Zhier Bao, Yang Chen, Tian Feng, and Jinxin Yang
Atmos. Chem. Phys., 25, 1569–1585, https://doi.org/10.5194/acp-25-1569-2025,https://doi.org/10.5194/acp-25-1569-2025, 2025
Short summary

Cited articles

Austin, J.: Toward the four dimensional assimilation of stratospheric chemical constituents, J. Geophys. Res.-Atmos., 97, 2569–2588, https://doi.org/10.1029/91JD02603, 1992. 
Beauchamp, M., Karagali, I., Gacitúa, G., Høyer, J. L., Ballarotta, M., and Fablet, R.: Multiscale neural assimilation scheme for high-resolution sea surface temperature reconstruction from satellite observations, Sci. Rep., 15, 39928, https://doi.org/10.1038/s41598-025-23682-9, 2025. 
Boudier, P., Fillion, A., Gratton, S., Gürol, S., and Zhang, S.: Data Assimilation Networks, J. Adv. Model Earth Sy., 15, e2022MS003353, https://doi.org/10.1029/2022MS003353, 2023. 
Carlton, A. G., Bhave, P. V., Napelenok, S. L., Edney, E. O., Sarwar, G., Pinder, R. W., Pouliot, G. A., and Houyoux, M.: Model Representation of Secondary Organic Aerosol in CMAQv4.7, Environ. Sci. Technol., 44, 8553–8560, https://doi.org/10.1021/es100636q, 2010. 
Carter, W. P. L.: Documentation of the SAPRC-99 chemical mechanism for VOC reactivity assessment, Final Report to California Air Resources Board Contract No. 92-329, and South Coast Air Quality Management District Contract No. 95-308, University of California, Riverside, https://ww2.arb.ca.gov/sites/default/files/classic/research/apr/past/92-392_1.pdf (last access: 20 October 2026), 2000. 
Download
Short summary
Our study pioneers a PyTorch-based atmospheric pollutant assimilation system that leverages automatic differentiation and Graphics Processing Unit (GPU) acceleration to achieve order-of-magnitude speedups, while maintaining analysis accuracy consistent with the traditional Fortran-based system.
Share
Altmetrics
Final-revised paper
Preprint