Ultra Fast PDE Solving via Physics Guided Few-step Diffusion

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Kong, Cindy Xiangrui, Wang, Yueqi, Zheng, Haoyang, Luo, Weijian, Lin, Guang
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911419500331008
author Kong, Cindy Xiangrui
Wang, Yueqi
Zheng, Haoyang
Luo, Weijian
Lin, Guang
author_facet Kong, Cindy Xiangrui
Wang, Yueqi
Zheng, Haoyang
Luo, Weijian
Lin, Guang
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2602_03627
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Ultra Fast PDE Solving via Physics Guided Few-step Diffusion
Kong, Cindy Xiangrui
Wang, Yueqi
Zheng, Haoyang
Luo, Weijian
Lin, Guang
Machine Learning
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.
title Ultra Fast PDE Solving via Physics Guided Few-step Diffusion
topic Machine Learning
url https://arxiv.org/abs/2602.03627