数据与计算发展前沿2025，Vol.7Issue(5)：113-122,10.DOI:10.11871/jfdc.issn.2096-742X.2025.05.009
基于雷达回波外推的短临降水预报模型TrajCast在国产加速器上的实践
Application of Radar Echo Extrapolation Based Model TrajCast on Domestic Accelerators for Short-Term and Imminent Precipitation Forecasting
摘要
Abstract
[Objective]This paper aims to construct a short-term and imminent precipitation forecasting model based on deep learning and explores the application of domestic accelerators in the field of meteorology,with the goal of improving the accuracy of meteorological disaster prevention and mitigation,and providing more reliable meteorological support for urban development and transportation.[Literature Scope]This paper focuses on the development of short-term precipitation forecasting technology,especially the application of deep learning-based radar echo extrapolation methods in the field of me-teorology.[Context]Traditional numerical weather prediction has uncertainty in precipitation forecasting within the first two hours,and radar echo extrapolation methods based on physical models,such as optical flow and cross-correlation methods,have limited prediction accuracy and stability under complex meteorological conditions.[Methods]In this study,we employ specialized datasets for short-term intense precipitation of an AI application developed by China Meteorological Administration.Based on TrajGRU,a deep learning network architecture,we establish the short-term and imminent precipitation forecasting model TrajCast.Employing data parallelism and mixed-precision training techniques,the model training is implemented on domestic accelerators.[Results]Exper-iment results demonstrate that the model outperforms traditional optical flow methods in precipitation forecasting,achieving a 4.1-fold acceleration ratio in model training on a domestic accelerator.[Limitations]The model has certain requirements for data quality and computing resources in practical applications,and its performance under extreme weather conditions still needs further verification.[Conclusions]This study provides new methods and technical support for short-term precipitation forecasting,promotes the application of domestic accelerators in the field of meteorology.The model has been demonstrated and applied in meteorological departments in Hunan,Zhejiang,and Hubei provinces,providing intelligent algorithm support for high-resolution and high-frequency meteorological intelligent forecasting products.关键词
短临降水预报/深度学习/国产加速器/雷达回波外推/混合精度训练Key words
short-term and imminent precipitation forecast/deep learning/domestic accelerator/radar echo extrapolation/mixed-precision training引用本文复制引用
辛昱杭,王琦祎,孙婧,赵春燕,刘雨佳,梁雪,陈杰..基于雷达回波外推的短临降水预报模型TrajCast在国产加速器上的实践[J].数据与计算发展前沿,2025,7(5):113-122,10.基金项目
光合基金"基于TrajGRU雷达回波外推算法的短临降水预报模型在国产加速器上的实践"(gh-fund202302034726) （gh-fund202302034726）
