扩散模型 + 物理引导推理求解偏微分方程

宇宙微尘
2026-08-21 18:55:10
人工智能
技术教程

这篇解读的对象是 2026 3 月发布在 arXiv 上的论文《Diffusion models with physics-guided inference for solving partial differential equations》。它属于扩散模型 + 物理信息这个交叉方向,但切入角度和主流做法有一个明显不同:它不把物理约束塞进训练过程,而是把物理放到推理阶段去引导采样。下面从动机、方法、理论、实验和复现几个层面,完整梳理这篇工作。



一、研究背景与动机


用神经网络求解偏微分方程(PDE)的研究,近几年大致形成了两条主线。一条是以物理信息神经网络(PINN)为代表的物理嵌入路线,它把 PDE 残差作为软约束写进损失函数,让网络在训练中同时满足数据和方程;另一条是纯数据驱动路线,直接用高保真的数值解或实验数据训练一个回归器,推理时前向算一次就得到解。


PINN 的优势在于无监督、不依赖标签数据,但它有一个被反复提及的短板:网络的参数是绑死在某一组 PDE 系数上的。方程里的参数一变(比如扩散系数、源项强度变了),往往就得重新训练,这在大规模、多场景的参数扫描中代价高昂。而扩散模型这几年在图像、物理场生成上展现出了很强的先验学习能力,被自然地引入到 PDE 求解中,但现有扩散模型在推理阶段往往缺乏显式的物理约束,生成结果未必满足方程。


这篇工作正是冲着这两个痛点来的:能不能让扩散模型像 PINN 一样满足物理,又像纯数据模型一样一次训练、反复复用?作者的答案是物理引导推理”——把训练和物理彻底解耦。



二、核心思想


整篇论文的核心可以用一句话概括:训练阶段,扩散模型只做纯数据驱动的事,完全不碰物理;推理阶段,才把物理以引导项的形式注入反向去噪过程。这样一来,物理约束不参与训练,模型学到的只是一个通用的解分布先验;等真正要解某个具体方程时,再用方程残差把采样轨迹向满足物理的解。换个系数,只要换引导项,模型本体不用动。


三、方法详解


训练阶段:纯数据驱动


作者采用标准的去噪扩散概率模型(DDPM),在高保真 PDE 解的数据集上做训练。U-Net 网络学的是从噪声数据逐步去噪、还原出干净解的映射。这一阶段没有任何方程残差、边界条件之类的物理项,训练目标和普通的生成式扩散模型完全一致。这样做的直接好处,是把学物理学生成这两件事分开了,训练出来的模型是一个与具体 PDE 系数无关的通用求解器雏形。


推理阶段:物理引导的反向去噪


物理真正发挥作用是在反向去噪(从随机噪声逐步生成解)的过程中。作者在每一步采样时引入三项机制:


PDE 残差能量函数。针对当前估计的解,计算它代入 PDE 后留下的残差,把这个残差构造成一个能量函数,作为梯度引导项。直觉上,这个能量越低说明解越满足方程,梯度方向就是把当前估计往更符合物理的位置推。


高斯平滑。残差的梯度在噪声较大的采样早期往往很不稳定、容易出现梯度爆炸,直接把原始梯度灌进采样过程会让轨迹发散。作者用高斯平滑对这个梯度引导做平滑处理,抑制噪声带来的梯度波动,让引导信号更稳定、更平滑。


边界投影算子。数据驱动的生成结果在边界上未必满足给定的边界条件。作者引入投影算子,在采样过程中强制把解投影到满足边界条件的子空间上,严格施加边界约束。


这三项合起来,就构成了一个独立于训练过程的、物理一致的随机迭代。这也是作者把它称为扩散启发的隐式求解器的原因:即使从纯随机噪声出发、过程中还带着随机扰动,这个迭代依然能收敛到 PDE 的解。



四、理论支撑


论文给出了一个随机变分的解释视角,把扩散模型、吉布斯测度(Gibbs measure)与 PDE 解这三者联系了起来。能量函数与高斯平滑、边界投影共同定义了一个目标分布,反向去噪过程可以看作对这个分布的一种随机变分逼近,而其最可能的位置对应的正是满足方程的物理解。这个视角把为什么物理引导推理能 work”从一个工程技巧,上升到了有概率解释支撑的层面。


五、实验与结果


作者在几类经典 PDE 上做了验证,包括泊松方程、扩散方程和 Burgers 方程,并重点测试了变系数场景。结果显示,在方程系数变化的情况下,这套框架无需重新训练即可保持稳健的收敛性、较高的求解精度和良好的泛化能力。尤其是 Burgers 方程这类带非线性的方程,物理引导推理也能稳定逼近其解,说明这套机制不只是对线性、简单方程有效。


需要说明的是,论文原文中的具体数值指标(各类方程的误差对比、收敛曲线等)以原文为准,本文不做臆造;读者若需引用或做对比实验,建议直接查阅原文的表格与图示。


六、贡献与亮点


这篇工作的贡献可以归纳为几点。其一,它把扩散模型的训练与 PDE 约束彻底解耦,使得换系数不用重训成为可能,这是相对 PINN 的一大实用优势。其二,它提供了一个随机变分的解释,把扩散模型、吉布斯测度和 PDE 解联系起来,给物理引导推理提供了理论支撑。其三,它构成了一个统一的、可替代经典数值求解器和 PINN 的框架,在变系数场景下表现出稳健的收敛和高精度。


七、局限性与讨论


需要客观看待的是,这篇工作的验证主要落在经典、较低维的 PDE 上,向高维、复杂几何或强非线性、多尺度问题上的扩展还有待观察。物理引导推理本质上是在推理时做额外的能量下降,采样步数与引导强度之间的权衡、以及额外的计算开销是否会抵消免重训带来的收益,也都是工程落地时需要掂量的问题。但作为一个把训练物理解耦的清晰范式,它的思路值得后续工作沿着更高效的引导、更广的方程类型继续延伸。


八、复现与代码


作者在 GitHub 上开源了代码,仓库地址为 github.com/Prometheus-cotigo/Pde-guide-Diffusion-Model。实现主要基于 MATLAB,并提供了 1D Burgers 方程的推理示例。对于想快速上手复现或做对比实验的同学,可以重点关注其反向去噪采样中如何接入残差能量梯度、高斯平滑和边界投影这三段代码,这是整套方法的核心,也最容易迁移到自己的方程上。


九、总结


这篇论文的价值不在于提出了多么复杂的网络结构,而在于它给出了一个干净的范式分离:扩散模型负责学生成先验,物理负责在推理时纠偏。这种训练数据驱动、推理物理引导的分工,既规避了 PINN 随参数重训练的痛点,又补上了扩散模型推理阶段物理一致性不足的短板。如果你正在找扩散模型与 PINN 结合方向的切入点,这篇工作是一个很适合作为 baseline 或改进起点的参考。



代码仓库:github.com/Prometheus-cotigo/Pde-guide-Diffusion-Model


231
0
0
0
关于作者
相关文章
  • 图解人工智能——Diffusion Model 扩散模型
    了解详情 
  • 机器学习引导的单原子电催化剂用于CO₂还原的基因逆向设计 ...
    了解详情 
  • mRNA与蛋白质相关性由什么决定?
    了解详情 
  • 扩散模型+大语言模型双轮驱动:AI小分子生成从虚拟设计走向临床 ...
    本文系统梳理了2016年以来AI驱动小分子设计的发展脉络。文章将分子生成AI拆解为四大核心模块:分 ...
    了解详情 
领取成功
本月5个550bit真机配额已发放给您,配额将在2个月后到期,请及时使用哦~
活动中心
联系我们
二维码
返回顶部
返回
活动中心

完成任务,轻松获取真机配额

×
每日必做
新手任务
长期任务
其他任务
快速回复 返回顶部 返回列表
玻色有奖小调研
填写问卷,将免费赠送您1个1000bit真机配额
(单选) 您是从哪个渠道得知我们的?*
您是从哪个社交媒体得知我们的?*
您是通过哪个学校的校园宣讲得知我们的呢?
取消

提交成功

真机配额已发放到您的账户,可前往【云平台】查看

量子AI开发者认证

考核目标

开发者能够成功搭建Kaiwu-PyTorch-Plugin项目基础环境,并成功运行示例代码,根据示例提示,输出指定的值并填写至相应的输入框中。

通过奖励

5个一年效期的1000量子比特真机配额

专属「量子AI开发者」社区认证标识

开发者权益

每月固定权益:5个550量子比特真机配额
前往考核

第一步

按照README提示成功安装Kaiwu-PyTorch-Plugin库环境依赖
前往GitHub

第二步

运行 community-assessment 分支下的 run_rbm.py 代码示例

第三步

理解示例代码,手动打印并填写如下数值:

正相采样的状态

负相采样的状态

正相的能量值

负相的能量值

*

提交答案

开发者权益

每月固定权益:5个550量子比特的真机配额

恭喜您完成考核

您将获得量子AI开发者认证标识及考核奖励

1000 bit*5

配额

Quantum AI Developer Certification

Assessment Objectives

Developers should successfully set up the basic environment for the Kaiwu-PyTorch-Plugin project, run the QBM-VAE sample code, and calculate the correct FID value based on the random seed value provided by the system.

Pass Rewards

10 quotas for 550-qubit real quantum machines with a one-year validity period

Exclusive "Quantum AI Developer" Community Certification Badge

Developer Benefits

Fixed Monthly Benefits: 5 quotas for 550-qubit real quantum machines
Proceed to Assessment

Step 1

Install the environment dependencies for the Kaiwu-PyTorch-Plugin library according to the README instructions
Go to GitHub

Step 2

Replace the Seed Value

Your seed value is

Step 3

Enter the FID Value You Calculated

*

Submit Answer

Developer Benefits

Fixed Monthly Benefits: 5 quotas of 550-qubit real machines

Congratulations on Completing the Assessment

You will receive the Quantum AI Developer Certification Badge and Assessment Rewards

550bit*10

Quotas