基于Transformer的双分支模型,提升中短期降水预报精度与概率可靠性。
CSU-PCAST: A Dual-Branch Transformer Framework for medium-range ensemble Precipitation Forecasting
- 采用双分支Transformer架构,融合气象变量与地理信息进行联合预测。
- 在2023年全年评估中,短时预报的CSI更高、RMSE更低,湿偏差显著降低。
- 适合需要高精度降水概率预报的气象预警与防灾决策场景。
准确的中短期降水预报对水文气象风险管理至关重要,但仍是数值天气预报(NWP)和数据驱动模型的挑战。本文提出CSU-PCAST,一种基于深度学习的全球降水集合预报框架。模型使用ERA5大气与地表变量(0.25°分辨率)及NASA IMERG数据集的降水标签进行训练,输入57个预报变量与静态地理场,预测大气状态与6小时累积降水。模型采用Swin Transformer主干网络,结合随机噪声条件、时间嵌入与双分支解码器,分别处理降水与非降水变量。推理阶段以操作型GFS分析场初始化,采用自回归策略生成30个集合成员,预报时效达15天。2023年全年评估显示,相比GEFS,CSU-PCAST在短时预报中具有更高的临界成功指数(CSI)与更低均方根误差(RMSE),减轻轻度降水的湿偏差和强降水的干偏差。概率验证表明,其CRPS更低,多个阈值下贝里技能得分更高,集合可靠性改善,尽管两者仍存在集合离散度不足问题。对‘桑巴’极端降水事件的案例研究进一步显示,模型在空间结构和超阈值概率指引上表现更优。结果表明,CSU-PCAST在短至中长期集合降水预报中具备潜力,但仍面临极端降水预测与集合校准的挑战。
原文摘要 · Abstract (English)
Accurate medium-range precipitation forecasting is essential for hydrometeorological risk management but remains challenging for both numerical weather prediction (NWP) systems and data-driven models. We present CSU-PCAST, a deep learning-based ensemble forecasting framework for global precipitation prediction. The model is trained using ERA5 atmospheric and surface variables at 0.25° resolution with precipitation labels from NASA's IMERG dataset. CSU-PCAST uses 57 prognostic variables and static geographical fields to predict both atmospheric states and 6-h accumulated precipitation. The framework employs a Swin Transformer backbone with stochastic noise conditioning, temporal embeddings, and a dual-branch decoder for precipitation and non-precipitation variables. During inference, CSU-PCAST is initialized from operational GFS analyses and generates 30 ensemble members out to 15 days using an autoregressive strategy. Evaluation against GEFS over the full year of 2023 shows improved precipitation forecast skill at short lead times, including higher Critical Success Index (CSI) and lower RMSE during the first several forecast days. CSU-PCAST also reduces GEFS wet bias for light precipitation and dry bias at heavier precipitation thresholds. Probabilistic verification demonstrates lower CRPS, higher Brier Skill Scores at several thresholds, and improved ensemble reliability relative to GEFS, although both systems remain underdispersive. A case study of the Sanba extreme precipitation event further shows improved spatial structure and exceedance-probability guidance. These results demonstrate the potential of CSU-PCAST for short-to-medium-range ensemble precipitation forecasting while highlighting remaining challenges in extreme precipitation prediction and ensemble calibration.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。