用流匹配降低计算开销,实现高效稳定的时间依赖PDE求解
Generative Latent Neural PDE Solver using Flow Matching
- 将PDE状态映射到低维潜在空间,统一处理不同网格结构
- 采用粗采样噪声调度的流匹配训练,提升长期稳定性与精度
- 适合需要高鲁棒性和不确定性量化的真实物理模拟场景
自回归下一步预测模型已成为数据驱动神经求解器预报时变偏微分方程(PDE)的主流方法。去噪训练与扩散概率模型密切相关,已被证明可增强神经求解器的时间稳定性,并通过随机推理机制实现集成预测和不确定性量化。原则上,此类训练在训练和推理过程中均需对一系列离散扩散时间步进行采样,不可避免地增加计算开销。此外,大多数扩散模型在规则均匀网格上施加各向同性高斯噪声,限制了其在不规则域中的适应性。本文提出一种基于潜变量的扩散模型用于PDE模拟,将PDE状态嵌入低维潜在空间,显著降低计算成本。该框架使用自动编码器将不同类型网格映射到统一的结构化潜在网格,捕捉复杂几何形状。通过分析常见扩散路径,我们提出在训练和测试中均采用粗采样噪声调度的流匹配方法。数值实验表明,所提模型在准确性和长期稳定性方面优于多个确定性基线,凸显了基于扩散的方法在稳健数据驱动PDE学习中的潜力。
原文摘要 · Abstract (English)
Autoregressive next-step prediction models have become the de-facto standard for building data-driven neural solvers to forecast time-dependent partial differential equations (PDEs). Denoise training that is closely related to diffusion probabilistic model has been shown to enhance the temporal stability of neural solvers, while its stochastic inference mechanism enables ensemble predictions and uncertainty quantification. In principle, such training involves sampling a series of discretized diffusion timesteps during both training and inference, inevitably increasing computational overhead. In addition, most diffusion models apply isotropic Gaussian noise on structured, uniform grids, limiting their adaptability to irregular domains. We propose a latent diffusion model for PDE simulation that embeds the PDE state in a lower-dimensional latent space, which significantly reduces computational costs. Our framework uses an autoencoder to map different types of meshes onto a unified structured latent grid, capturing complex geometries. By analyzing common diffusion paths, we propose to use a coarsely sampled noise schedule from flow matching for both training and testing. Numerical experiments show that the proposed model outperforms several deterministic baselines in both accuracy and long-term stability, highlighting the potential of diffusion-based approaches for robust data-driven PDE learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。