Conditional Flow Matching for Probabilistic Downscaling of Maximum 3-day Snowfall in Alaska

📄 arXiv: 2604.25172v1 📥 PDF

作者: Douglas Brinkerhoff, Elizabeth Fischer

分类: physics.comp-ph, cs.LG, physics.ao-ph

发布日期: 2026-04-28

🔗 代码/项目: GITHUB


💡 一句话要点

提出WxFlow,基于条件流匹配实现阿拉斯加最大3日降雪概率降尺度,提升光谱保真度。

🎯 匹配领域: 支柱二:RL算法与架构 (RL & Architecture) 支柱四:生成式动作 (Generative Motion)

关键词: 气候降尺度 条件生成模型 流匹配 概率预测 地形影响

📋 核心要点

  1. 气候模型分辨率不足以捕捉复杂地形降水,高分辨率区域模型计算成本高昂,难以进行不确定性量化。
  2. WxFlow利用条件流匹配,学习粗分辨率气候模型输出和高分辨率地形到精细尺度降水场的映射。
  3. 实验表明,WxFlow显著提升了光谱保真度,降低了连续等级概率得分,并能快速生成具有物理意义的概率集合。

📝 摘要(中文)

复杂地形中的降水受几公里尺度上的地形过程控制,但气候模型通常在50-100公里分辨率下运行,缺乏这种地形细节。使用高分辨率区域模型(如WRF)进行动态降尺度可以解决这些过程,但计算成本(每个场景需要数月的挂钟时间)排除了不确定性量化所需的大量集合。我们提出了WxFlow,这是一个基于流匹配的条件生成模型,它学习将粗分辨率气候模型输出和高分辨率地形映射到精细尺度降水场的校准概率集合。应用于阿拉斯加东南部的最大3日降雪量的4公里WRF模拟,WxFlow在光谱保真度方面实现了87.8%的改进,并且相对于传统的基于递减率校正的双三次降尺度,连续等级概率得分显着降低,同时在笔记本电脑上以秒为单位生成50个成员的集合。集合离散度在空间上是连贯的,并且受地形控制,反映了物理上合理的不确定性结构。所有代码都可以在https://github.com/glide-ism/wrf-flow上找到。

🔬 方法详解

问题定义:论文旨在解决气候模型在复杂地形区域降水预测中分辨率不足的问题。现有动态降尺度方法(如WRF)虽然能提高分辨率,但计算成本过高,难以进行大规模集合模拟,从而限制了不确定性分析。传统插值方法(如双三次插值)虽然计算效率高,但精度不足,无法有效利用高分辨率地形信息。

核心思路:论文的核心思路是利用条件生成模型,学习从低分辨率气候模型输出和高分辨率地形到高分辨率降水场的映射关系。具体而言,采用流匹配(Flow Matching)框架,通过学习连续的向量场,将低分辨率输入逐步变换为高分辨率输出。这种方法既能保证生成样本的多样性,又能有效利用条件信息,提高预测精度。

技术框架:WxFlow的整体框架包含以下几个主要步骤:1) 数据预处理:将低分辨率气候模型输出和高分辨率地形数据作为条件输入,高分辨率降水数据作为目标输出。2) 模型训练:使用条件流匹配方法训练生成模型,学习从条件输入到目标输出的连续变换。3) 样本生成:给定新的低分辨率气候模型输出和高分辨率地形数据,生成多个高分辨率降水场样本,形成概率集合。4) 结果评估:使用光谱保真度和连续等级概率得分等指标评估生成结果的质量。

关键创新:论文的关键创新在于将流匹配方法应用于气候降尺度问题,并结合高分辨率地形信息,实现了高精度、高效率的概率降尺度。与传统的动态降尺度方法相比,WxFlow计算成本更低,能够快速生成大规模集合。与传统的插值方法相比,WxFlow精度更高,能够有效利用地形信息。

关键设计:WxFlow使用了一种基于神经网络的条件向量场估计器,该估计器以低分辨率气候模型输出和高分辨率地形数据作为输入,输出一个向量场,用于将低分辨率输入逐步变换为高分辨率输出。损失函数采用流匹配损失,鼓励模型学习一个连续的、平滑的向量场。模型训练过程中,使用了数据增强技术,以提高模型的泛化能力。具体网络结构和参数设置在论文中有详细描述。

🖼️ 关键图片

fig_0
fig_1
fig_2

📊 实验亮点

实验结果表明,WxFlow在阿拉斯加东南部的最大3日降雪量降尺度任务中,光谱保真度提升了87.8%,连续等级概率得分显著降低。与传统的基于递减率校正的双三次降尺度方法相比,WxFlow在精度和效率方面均有显著优势。此外,WxFlow生成的集合离散度在空间上是连贯的,并且受地形控制,反映了物理上合理的不确定性结构。

🎯 应用场景

该研究成果可应用于气候变化影响评估、水资源管理、农业规划等领域。通过快速生成高分辨率降水概率集合,可以更准确地评估极端降水事件的风险,为相关决策提供科学依据。未来可扩展到其他气候要素的降尺度,并应用于其他复杂地形区域。

📄 摘要(原文)

Precipitation in complex terrain is governed by orographic processes operating at scales of a few kilometers, yet climate models typically run at resolutions of 50--100~km where this topographic detail is absent. Dynamical downscaling with high-resolution regional models such as WRF can resolve these processes, but the computational cost -- months of wall-clock time per scenario -- precludes the large ensembles needed for uncertainty quantification. We present WxFlow, a conditional generative model based on flow matching that learns to map coarse-resolution climate model output and high-resolution topography to calibrated probabilistic ensembles of fine-scale precipitation fields. Applied to 4~km WRF simulations of maximum 3-day snowfall over southeast Alaska, WxFlow achieves 87.8\% improvement in spectral fidelity and dramatically lower Continuous Ranked Probability Scores relative to conventional lapse-rate-corrected bicubic downscaling, while generating 50-member ensembles in seconds on a laptop. Ensemble spread is spatially coherent and governed by topography, reflecting physically plausible uncertainty structure. All code is available at https://github.com/glide-ism/wrf-flow.