简介这是一份基于PINN物理信息神经网络的微分方程求解Python实践资源面向深度学习与科学计算交叉领域的初学者和研究者帮助读者理解如何用神经网络在缺乏解析解时求解常微分方程和偏微分方程尤其适合复杂几何域和非线性问题。压缩包共26个文件以17个ipynb为主要内容覆盖Euler Beam、扩散方程、圆盘拉普拉斯方程、泊松方程多种边界条件、带源项的一维ODE、Lorenz系统等典型算例另含3个py工具脚本、1个md说明、1个结果图及4个备份文件整体仅891KB轻量易运行。目前已有218人学习下载。通过逐行运行的代码笔记和配套脚本读者能完整看到PINN实现流程——数据准备、网络设计、损失函数构造数据项物理约束项、优化训练与结果分析这些代码既适合教学演示也可作为进一步研究DeepXDE等工具的入门跳板。1. 基于PINN的微分方程求解方法Python能补上什么数据稀疏时也能算出解手头只有边界几个监测点计算域却是个不规则几何用有限差分铺网格会让人很别扭换用PINN之后我通常只需要写一个网络、一个自动微分残差项就能把整个解平面算出来。基于PINN的微分方程求解方法Python做的事是把微分方程本身塞进神经网络的损失函数让网络在拟合数据的同时必须满足物理规律。它适合正向求解也适合反演未知参数尤其适合数据稀疏、计算域不规则的场景。这篇文章会按一套最小可跑通、能直接改到自己方程上的流程讲先搭连续时间PINN再处理长时间域最后把调参和边界条件的坑都列出来。2. 连续时间PINN的最小可跑通架构损失函数与三层网络2.1 先选连续时间法还是时间推进法PINN落地前要先选求解策略。“连续时间法”把整个时空坐标作为网络输入一次训练得到全域解结构最简单时间推进法则把时间轴切段一段段滚着解适合长时间域。我入门时强烈建议从连续时间法开始因为变量少损失函数和残差的关系看得清楚等第一篇跑通了再考虑时间推进。下面这张对比表可以帮你快速做判断对比项连续时间法时间推进法网络输入全部时空坐标每段窗口内的坐标适用场景短时域、缓变过程长时域、快变化过程额外参数基本没有窗口数、重叠率典型问题长时间后精度下滑窗口拼接处容易漂移对大多数第一次接触PINN的人不要一开始就上强对流、强非线性方程。先拿一个简单常微分方程把流程走通成本最低。2.2 网络、自动微分与三项损失以一个最简单的ODE为例du/dt -2u, u(0) 1解析解是 e^{-2t}。PINN要做的是让网络输出的 u(t) 既满足初值又满足方程残差为零。损失函数一般拆成三块L λ_r * L_residual λ_0 * L_initial λ_b * L_boundary其中 L_residual 是方程残差的均方误差L_initial 是初值误差L_boundary 是边界误差。没有边界条件时L_boundary可以去掉。核心在于方程残差不是手工离散出来的而是通过自动微分直接算。网络输出对输入 t 求导代入原方程就能得到残差。这种做法的好处是不需要网格拓扑不需要有限差分模板坐标维度增加也只是加输入列而已。2.3 一个可以直接跑的Python训练循环代码与参数下面这段用 PyTorch 写的训练循环是PINN里最常见也最不容易出问题的结构。我通常拿它当模板改方程和初始条件就能用到新问题上。import torch import torch.nn as nn # 网络输入 t输出 u三层全连接 tanh 激活 net nn.Sequential( nn.Linear(1, 20), nn.Tanh(), nn.Linear(20, 20), nn.Tanh(), nn.Linear(20, 1) ) # 域内采样尽量贴满 (0, 1)避开 t0 留给初值点 t_in torch.linspace(0.001, 0.999, 256).reshape(-1, 1) t0 torch.zeros(1, 1) u0 torch.ones(1, 1) def pde_residual(t): t t.clone().requires_grad_(True) u net(t) du_dt torch.autograd.grad(u, t, torch.ones_like(u), create_graphTrue)[0] return du_dt 2.0 * u # 残差du/dt 2u 0 opt torch.optim.Adam(net.parameters(), lr1e-2) for step in range(2000): opt.zero_grad() res pde_residual(t_in) loss_res torch.mean(res**2) loss_ic torch.mean((net(t0) - u0)**2) loss loss_res 10.0 * loss_ic loss.backward() opt.step() if step % 200 0: print(step, round(loss_res.item(), 6), round(loss_ic.item(), 6))这段代码里的三个选择值得说明。激活函数我用的是 Tanh 而不是 ReLU因为 PINN 要求残差里出现导数Tanh 光滑且二阶导连续ReLU 的导数在零点不连续收敛时容易出毛刺。网络宽度选 20 个神经元而不是越大越好因为当前输入只有一个 t这个规模足够表达光滑解过宽反而让训练变慢。域内采样用固定的 linspace 而不是随机采样是为了保证边界附近也有采样覆盖随机采样更适合后面做残差驱动的自适应采样。跑完之后可以用解析解做一个快速校验with torch.no_grad(): t_test torch.linspace(0, 1, 100).reshape(-1, 1) err torch.abs(net(t_test) - torch.exp(-2 * t_test)).max().item() print(max abs err:, err)如果 max abs err 在 1e-3 以下说明网络真的学到了方程解而不是把初值硬背下来。这一步是后续所有调试的基础。2.4 从ODE改到PDE输入变成多个坐标很多读者最终要解的是偏微分方程。把上面的ODE改成PDE并不复杂只需要把网络输入从单一 t 变成 (x, t) 的组合残差也用自动微分同时求一阶和二阶导。扩散方程 ∂u/∂t D·∂²u/∂x² 的残差写法如下x x.clone().requires_grad_(True) t t.clone().requires_grad_(True) u net(torch.cat([x, t], dim1)) u_t torch.autograd.grad(u, t, torch.ones_like(u), create_graphTrue)[0] u_x torch.autograd.grad(u, x, torch.ones_like(u), create_graphTrue)[0] u_xx torch.autograd.grad(u_x, x, torch.ones_like(u_x), create_graphTrue)[0] residual u_t - D * u_xx注意二阶导是对一阶导数再求一次 autogradcreate_graphTrue 必须保留否则梯度无法继续回传。这样做的好处是无论方程里出现几阶导你都不用手工推导差分模板代码结构和ODE版本几乎一致。3. 时间推进解法长时间域的窗口滚动与残差接线3.1 全局连续时间为什么会在长时域翻车连续时间法不是万能的。当计算时间范围从 [0, 1] 拉到 [0, 10]或者方程本身带有快速振荡项时我见过最多的情况是训练总损失已经降到很低但把解画出来中间一段明显偏离参考解前后段反而正常。原因在于 PINN 本质上是全局回归网络为了在整段时间里平均拟合残差会把局部高频信息平滑掉。这是连续时间法的结构性问题不是调大网络就能解决的。所以当时间尺度长、方程本身包含多个时间尺度时我一般直接切成时间推进。常见做法是把 [0, T] 分成若干窗口每个窗口内用连续时间PINN求局部解窗口之间通过“上一窗口末端的预测值”接续。代价是多了窗口数和重叠率两个超参数换来的是长时间域上更可用的精度。3.2 窗口滚动把长时域切成多段下面这段是窗口滚动的流程示意重点在于理解 u_carry 的传递方式T_END 1.0 n_win 8 tau T_END / n_win u_carry 1.0 # 初始条件 for k in range(n_win): t_a k * tau t_b t_a tau # 构造该窗口数据集 # 1) 初值点t_a 处强制 u u_carry # 2) 内点在 (t_a, t_b) 内采样 # 3) 训练一个局部网络损失里包含残差 初值 u_carry local_net(t_b).item() # 作为下一窗口的初值窗口数 n_win 一般取 5 到 20。窗口太少和全局连续时间法区别不大窗口太多每一段都有拼接误差误差会逐段累积。我通常先取 n_win8看结果再决定加密还是放宽。3.3 重叠区是怎么接线的上一窗口预测做软约束只靠一个末端点 hundle窗口边界处很容易出现折角或跳变。更可靠的做法是把上一窗口尾部一小段预测值作为当前窗口的“软约束”让两个窗口在重叠区域内尽量一致。代码上就是多一个 overlap_loss# prev_u_at_t_overlap 是上一窗口在重叠区间的预测值 overlap_loss torch.mean((local_net(t_overlap) - prev_u_at_t_overlap.detach())**2) loss loss_res 5.0 * loss_ic 1.0 * overlap_loss这里必须做 .detach()否则梯度会跨窗口回传到上一个网络两个窗口的参数互相拉扯训练容易乱。重叠区长度一般取 tau 的 10% 到 20%既足够约束连续性又不会让窗口之间过度耦合。3.4 什么时候坚持连续时间什么时候必须切时间推进我的判断标准很简单如果参考解在时间方向上形态变化平缓连续时间法足够如果解在局部出现陡峭变化而你又希望只用一个网络解决大概率会卡在残差不降的阶段。这时候与其提升网络容量不如先切时间推进很多时候精度问题会直接消失。4. 调参避坑的四个常见故障损失不平衡、零解、采样与归一化4.1 初边界损失被残差淹没现象训练日志里 loss_res 一直在降但 loss_ic 几乎不动最终画出来的解整体平移了一段距离。原因残差采样点有几百个而初值点往往只有一两个。均方误差对样本数量敏感残差项在总损失里天然占大头网络优先满足残差把初值丢在一边。解决给初边界损失加权并且把权重设成可观察的常数。我常用的组合是loss 10.0 * loss_ic 20.0 * loss_bc 1.0 * loss_res不要一上来就设 100 倍那样会走向另一个极端边界学得很好但内部残差完全摆烂。从 5 到 20 倍之间去试观察各个分量是否同时下降。提示loss_ic 和 loss_bc 在每次迭代里都要单独打印。只看总损失你很难判断是哪一项在主导。4.2 零解收敛PINN明明loss很低解的均值却全是0现象总损失降到 1e-5 量级看起来训练非常顺利把输出画出来却是一条平线 u0。原因当方程齐次、初始条件恰好为 0 时u0 本身就是一个严格满足方程残差的解。网络初始化时输出接近零优化器很容易滑进这个平凡解损失函数还不会报警。解决在训练循环里周期性打印网络输出的均值和最大值而不仅仅是损失值。代码只需要几行with torch.no_grad(): u_mean net(t_in).mean().item()如果 u_mean 接近 0而参考解明显不是零就要立刻怀疑收敛到了零解。对策是先单独用初边界损失训练几十步让网络先“记住”边界值再逐步放开残差项。这是PINN里最常见的隐性翻车不把物理量拉出来看光看loss永远发现不了。4.3 均匀采样漏掉局部结构残差驱动重采样怎么做现象边界层、激波、陡峭斜坡这些区域误差特别大但均匀采样点在这些位置并不密集。原因PINN是样本点上的回归均匀网格在特征剧烈区天然样本不足网络会把局部突变当作噪声平滑掉。解决每训练 N 步用当前残差的绝对值作为概率分布重新采一批增强点加入训练集。with torch.no_grad(): res_abs torch.abs(pde_residual(t_in).detach()).reshape(-1) prob res_abs / res_abs.sum() idx torch.multinomial(prob, 64, replacementTrue) extra_t t_in[idx] 0.01 * torch.randn_like(t_in[idx])这里加了一个小扰动避免增强点完全落在原采样点上。增强点并入训练集后总量要控制否则训练集无限膨胀每次迭代的计算成本会逐渐失控。4.4 坐标归一化量纲差一到两个数量级自动微分就开始怪现象物理坐标里有秒、有千米、有毫帕量级差距超过两个数量级损失权重怎么调都收敛不顺。原因神经网络对输入尺度敏感方程残差里的导数项系数也依赖坐标量纲。不归一化时梯度里各项的量级混杂优化器很难找到统一的学习率。解决先把物理坐标映射到 [-1, 1] 或 [0, 1]然后记得在方程残差里保留链式法则的转换系数。以时间归一化为例t_norm t / (t_max - t_min) # 原方程 du/dt_phys f自动微分解出的是 du/dt_norm # 需要通过链式法则换算 # du/dt_phys du/dt_norm * (dt_norm/dt_phys) du/dt_norm / (t_max - t_min) residual (1.0 / (t_max - t_min)) * du_dt_norm - f(t)很多入门代码只归一化坐标却忘了在残差方程里补这个系数导致换汤不换药。同样的逻辑也适用于空间坐标扩散方程的 ∂²u/∂x² 要乘两次 1/(x_max - x_min)容易漏我一般会把换元后的系数单独写在注释里。5. 边界条件与初值的处理细节软约束、硬约束与数值验证5.1 软约束和硬约束什么时候硬编码边界软约束就是前面讲的把边界条件加进损失函数通用性强实现简单但需要调权重。硬约束则是把边界条件直接写进网络输出结构让边界值“必须满足”。比如初值 u(0)1可以构造u_pred 1.0 t * net_out这样无论网络输出什么t0 时 u_pred 恒等于 1。对于两端固定为零的 Dirichlet 边界可以用 x(1-x) 作为乘子phi x * (1 - x) u_pred phi * net_outphi 在边界处为 0所以边界条件被强制满足。硬约束能省掉边界损失权重这个超参数还能绕开零解收敛的问题。代价是构造 phi 需要知道边界的几何信息规则矩形域很简单复杂几何域就要用符号距离函数近似实现成本明显上升。我的建议是入门阶段先用软约束等边界老出问题再上硬约束。5.2 边界优先调度先拟合初边值再引入残差很多收敛问题不是数学上无解而是训练顺序不对。一个常见的操作是先让网络只学初边界残差项按迭代步数逐步放大。代码可以直接在损失里加一个退火系数anneal min(1.0, step / 500.0) loss 1.0 * loss_ic 5.0 * loss_bc anneal * loss_res前 500 步里网络主要把边界值“背”下来残差项从 0 逐渐放大到满权重。这种做法能明显减少零解收敛和边界漂移两类问题。你也可以改成用损失阈值触发比如当 loss_ic 降到 1e-3 以下再把残差放大但用 step 数控制更简单省得训练过程被某个边界损失卡住。5.3 验证要落到解上误差计算与损失分量历史PINN 训练日志里最骗人的就是总损失。无论如何要强制自己对参考解做误差计算。有解析解就用解析解没有就用高精度数值解做参照。误差计算代码非常简单with torch.no_grad(): u_pred net(t_test) err torch.abs(u_pred - u_exact) rel_err torch.linalg.norm(err) / torch.linalg.norm(u_exact) print(rel err:, rel_err.item())同时我要求自己每 200 步打印一次 loss_res、loss_ic、loss_bc 三个分量。只有分量曲线才能告诉你到底是边界没学会还是内部残差下不去。总损失下降但解不对各分量此消彼长的例子我见过太多次这就是PINN最典型的黑匣子陷阱。6. 进阶技巧把PINN扩展到反问题与参数辨识正向问题跑通之后最容易见到额外收益的是参数辨识方程形式已知但扩散系数 D、源项强度等参数未知。PINN 处理这种反问题几乎不用改网络结构只需把待定参数从常数变成可训练变量。D nn.Parameter(torch.tensor(1.0, dtypetorch.float32)) def pde_residual(x, t): u net(torch.cat([x, t], dim1)) u_t torch.autograd.grad(u, t, torch.ones_like(u), create_graphTrue)[0] u_x torch.autograd.grad(u, x, torch.ones_like(u), create_graphTrue)[0] u_xx torch.autograd.grad(u_x, x, torch.ones_like(u_x), create_graphTrue)[0] return u_t - D * u_xx opt torch.optim.Adam(list(net.parameters()) [D], lr1e-3)训练时损失函数在方程残差、初边值之外再加入测量数据点上的误差 loss_data。注意两点第一先用固定 D 把网络预训练到残差平稳再放开 D否则网络会利用自身自由度把 D 的错误吸收掉第二每轮打印 D 的当前估计值观察它是稳定在某个区间还是在“散步”。我最早做参数辨识时只盯着总损失数据项降了D 却跑到了负值网络用自己的自由度把物理参数的错误抵消了。后来把 D 的估计轨迹单独画出来才发现问题所在。这类黑匣子调试多打印物理量本身少只盯一条总损失曲线一次能省半天时间。希望帮到你。本文还有配套的精品资源点击获取