arXiv:2602.03627cs.LG2026-02被引 1

用物理约束加速微分方程求解,推理速度提升数十倍

Ultra Fast PDE Solving via Physics Guided Few-step Diffusion

  • 通过物理引导蒸馏,将多步扩散模型压缩为几步生成器
  • 在5个基准上推理速度提升数量级,误差降低8倍以上
  • 适合需要快速且物理一致解的科学计算场景

基于扩散模型在求解偏微分方程(PDE)方面展现出优异的精度与泛化能力,但仍面临采样成本高、物理一致性不足的问题,主要源于其多步迭代采样机制和缺乏显式物理约束。为此,我们提出Phys-Instruct,一种新型物理引导蒸馏框架,不仅能(1)通过匹配生成器与先验扩散分布,将预训练的扩散PDE求解器压缩为少步生成器以实现快速采样,还能(2)通过显式注入PDE知识进行蒸馏引导,增强物理一致性。Phys-Instruct建立在坚实的理论基础上,推导出可计算梯度的物理约束训练目标。在五个PDE基准上,Phys-Instruct实现数量级的推理加速,同时相比最先进扩散基线将PDE误差降低超过8倍。此外,生成的无条件学生模型可作为紧凑先验,高效支持多种下游条件任务的物理一致推理。结果表明,Phys-Instruct是一种新颖、有效且高效的深度生成模型驱动的超快PDE求解框架。

原文摘要 · Abstract (English)

Diffusion-based models have demonstrated impressive accuracy and generalization in solving partial differential equations (PDEs). However, they still face significant limitations, such as high sampling costs and insufficient physical consistency, stemming from their many-step iterative sampling mechanism and lack of explicit physics constraints. To address these issues, we propose Phys-Instruct, a novel physics-guided distillation framework which not only (1) compresses a pre-trained diffusion PDE solver into a few-step generator via matching generator and prior diffusion distributions to enable rapid sampling, but also (2) enhances the physics consistency by explicitly injecting PDE knowledge through a PDE distillation guidance. Physic-Instruct is built upon a solid theoretical foundation, leading to a practical physics-constrained training objective that admits tractable gradients. Across five PDE benchmarks, Phys-Instruct achieves orders-of-magnitude faster inference while reducing PDE error by more than 8 times compared to state-of-the-art diffusion baselines. Moreover, the resulting unconditional student model functions as a compact prior, enabling efficient and physically consistent inference for various downstream conditional tasks. Our results indicate that Phys-Instruct is a novel, effective, and efficient framework for ultra-fast PDE solving powered by deep generative models.

PDE求解扩散模型物理引导快速推理

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