arXiv:2512.17877physics.ao-phcs.LG2025-12被引 2

用神经网络自动学习地形跟随坐标,提升大气模拟精度

Learning vertical coordinates via automatic differentiation of a dynamical core

  • 将垂直坐标设为可学习参数,通过自动微分精确计算几何项
  • 在陡峭地形上误差降低1.4至2倍,消除虚假垂直速度条纹
  • 适合关注气象数值模拟与可微分物理建模的研究者

大气模型中的地形跟随坐标常因网格结构影响解的准确性,尤其在陡峭地形处,扭曲的坐标层会引入虚假水平和垂直运动。传统方法如混合或SLEVE坐标依赖人工调参的解析衰减函数,且参数固定。本文提出在可微分动力核心中将参数化垂直坐标作为可学习组件,构建基于积分变换神经网络的NEUVE地形跟随坐标,保证单调性。关键创新在于利用自动微分精确计算几何度量项,避免有限差分导数带来的截断误差。通过时间积分将模拟误差反馈至参数化,使网格结构同时优化物理与数值性能。多个标准测试表明,所学坐标在非线性统计基准中均方误差降低1.4至2倍,并有效消除陡峭地形上的虚假垂直速度条纹。

原文摘要 · Abstract (English)

Terrain-following coordinates in atmospheric models often imprint their grid structure onto the solution, particularly over steep topography, where distorted coordinate layers can generate spurious horizontal and vertical motion. Standard formulations, such as hybrid or SLEVE coordinates, mitigate these errors by using analytic decay functions controlled by heuristic scale parameters that are typically tuned by hand and fixed a priori. In this work, we propose a framework to define a parametric vertical coordinate system as a learnable component within a differentiable dynamical core. We develop an end-to-end differentiable numerical solver for the two-dimensional non-hydrostatic Euler equations on an Arakawa C-grid, and introduce a NEUral Vertical Enhancement (NEUVE) terrain-following coordinate based on an integral transformed neural network that guarantees monotonicity. A key feature of our approach is the use of automatic differentiation to compute exact geometric metric terms, thereby eliminating truncation errors associated with finite-difference coordinate derivatives. By coupling simulation errors through the time integration to the parameterization, our formulation finds a grid structure optimized for both the underlying physics and numerics. Using several standard tests, we demonstrate that these learned coordinates reduce the mean squared error by a factor of 1.4 to 2 in non-linear statistical benchmarks, and eliminate spurious vertical velocity striations over steep topography.

大气模拟可微分建模神经网络坐标系统

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。