LLM2D
WeatherODE:一种用于天气预报的夹层物理驱动神经ODE,用于缓解时间离散化挑战
Mitigating Time Discretization Challenges with WeatherODE: A Sandwich Physics-Driven Neural ODE for Weather Forecasting
作者: Peiyuan Liu, Tian Zhou, Liang Sun, Rong Jin
发布日期: 10/10/2024
arXiv ID: oai:arXiv.org:2410.06560v1

摘要

在天气预报领域,传统模型往往受到离散化误差和时间依赖源差异的影响,限制了其预测性能。本文提出了一种名为 WeatherODE 的新型单阶段、物理驱动的常微分方程 (ODE) 模型,旨在提高天气预报精度。通过利用波动方程理论并整合时间依赖源模型,WeatherODE 有效地解决了与时间离散化误差和动态大气过程相关的挑战。此外,我们设计了一种 CNN-ViT-CNN 三明治结构,促进了针对不同但相互关联的任务的有效学习动态,这些任务在平流方程估计中具有不同的优化偏差。通过严格的实验,WeatherODE 在全球和区域天气预报任务中都表现出优越的性能,在均方根误差 (RMSE) 方面分别比最新的最先进方法高出 40.0% 和 31.8%。源代码可在 \url{https://github.com/DAMO-DI-ML/WeatherODE} 获得。