做芯片热分析的朋友，多少都被网格折腾过——结构稍微复杂点，剖分能耗费半天，换个热源分布又要重来。后来我开始用 PINN（Physics-Informed Neural Networks，物理信息神经网络）处理这类问题，发现它的路数和传统数值方法完全不一样：把控制方程直接写进神经网络的损失函数，让网络在拟合数据的同时必须守物理规律。这篇博文就按一条很清晰的实战路径展开，从最简单的 RC 电路开始，一路做到芯片热分析。文章里的代码基于 PyTorch，也会聊到 MATLAB 的实现思路，适合刚入门 PINN、或者正在为热仿真方案发愁的朋友。

1. PINN 到底在做什么：把物理方程塞进损失函数

1.1 从“纯数据拟合”到“物理约束拟合”

先聊本质。普通神经网络学习的是输入到输出的映射关系，核心依赖是数据，数据不够或者噪声大，网络就会开始“瞎编”。PINN 的做法完全不同，它在损失函数里额外加入物理方程残差项，让网络的输出必须尽可能满足控制方程。

举个具体的例子，一个一阶系统如果满足方程 du/dt + a·u = f(t)，那我们可以把残差写成 r(t) = du/dt + a·u - f(t)。如果神经网络给出的预测值 u(t) 完全满足方程，那么残差 r(t) 在所有采样点上都等于 0。训练时我们不仅要求网络输出与已知数据接近，还要求这个残差尽量小，最终得到的是一个既尊重数据、又严格符合物理规律的解。

把这个思路套到生活里特别容易理解：你把物理规律写进考纲，让神经网络当考生。纯数据拟合是开卷考试但参考答案残缺，网络找不到规律就开始编；PINN 等于把考纲变成硬性要求，答案再漂亮也不能违背考纲。过去十年里，AI 在图像、文本等领域靠海量数据堆出了不少成果，但到了物理世界，数据往往稀疏且昂贵，这时候物理约束的价值就体现出来了。

为什么 PINN 能做到这件事？关键在自动微分。神经网络本身是若干光滑函数复合而成，可以用 autograd 对输入变量求任意阶导数。控制方程里的导数项不靠有限差分近似，而是直接解析求导，精度高，误差不会像传统网格方法那样随步长累积。这个特性是整个 PINN 方法的基石。

1.2 损失函数怎么设计：残差项、初值/边值项、数据项

PINN 的损失函数通常是几项加权相加，最常见的形式是 MSE（均方误差）的线性组合，我拆开细讲。

物理残差项 在计算域内部采样一批点，代入控制方程算出残差，然后求均方值。这是主导项，决定网络有没有真正“学会”物理规律。对于 ODE 问题，通常采样几百个点就够；对于二维 PDE 问题，内点一般要上千甚至更多。

初值与边界条件项 在初始时刻或边界上采样点，让网络输出符合约束。如果不用硬约束，这部分就是软约束，权重建议比残差项给得大。原因是初值和边界条件往往决定了方程解的唯一性，如果这里松弛了，即使残差很小，最终解也可能是错的。

数据项（可选） 如果手上有实测数据，比如芯片上几个热敏电阻测到的温度点，可以把网络输出与实测值的差也加入损失，这是 PINN 最有价值的地方：它能同时利用稀疏实测数据和完整物理方程，补全无传感器位置的温度场。

三个典型场景可以直接对照着选：

没有数据、只有方程和边界条件：这是正向求解问题，损失用残差项加边界项。 有方程、有稀疏数据、边界条件不完全：这是数据增强问题，损失用残差项加数据项加已有约束。 有数据、边界条件已知、但控制方程中的参数（如导热系数）未知：这是反问题，把未知参数也变成网络的可训练变量，一起优化。

新手最容易踩的坑是权重配比失衡。我个人的经验是：初值和边界项的权重通常比残差项高一个数量级，比如残差权重为 1，边界权重取 10 或 100。很多网络不收敛的案例，调一下权重比改进网络结构管用得多。

2. 第一个实战：RC 电路的 PINN 建模与求解

2.1 RC 电路的物理模型与无量纲化

RC 电路是 PINN 入门最经典的载体，因为它是最简单的一阶线性常微分方程，物理过程直观，而且有解析解可以对照验证。

充电过程的控制方程是：

RC · du/dt + u = V_in，u(0) = 0

其中 V_in 是输入电压，u(t) 是电容两端的电压，τ = RC 是时间常数，决定了充电的快慢。这个方程的解是 u(t) = V_in·(1 - exp(-t/τ))，一条标准的饱和上升曲线。

为什么拿它入门？因为 PINN 的所有核心机制——采样、残差损失、自动微分、优化、解析验证——在这个例子里全都齐了，但代码量又足够短。就像学游泳先在浅水区扑腾，把动作练顺了再下水道，会从容得多。

但直接拿秒做时间单位有个隐患：如果 R 和 C 的取值不同，τ 量级差异会很大，输入 t 的尺度不统一，网络初始梯度容易失衡，收敛速度被拖慢。解法是做无量纲化，令 x = t/τ，y = u/V_in，方程化为：

dy/dx + y = 1，y(0) = 0

不管真实 RC 取多少，无量纲方程都一样，网络只需要在 x ∈ [0, 5] 这个固定区间上训练即可。这是我做 PINN 一直坚持的习惯：先无量纲化，再进网络，十个问题里有九个能减少很多调参时间。

2.2 用 PyTorch 实现 RC 电路的 PINN

网络结构不用复杂，3 层全连接、每层 50 个神经元、激活函数用 tanh 就足够。tanh 是 PINN 里的常客，因为它光滑可导，梯度范围适中。ReLU 在求二阶导时会出现导数突变甚至为零，做热传导这类含二阶导的问题必然翻车，新手最好不要用它。

完整代码我贴一个自己跑过的版本，这里按无量纲化后的方程处理，R=C=V=1，方程就是 du/dt + u = 1，初值 u(0)=0：

import torch import torch.nn as nn import matplotlib.pyplot as plt torch.manual_seed(2024) class RC_PINN(nn.Module): def __init__(self, hidden=50): super().__init__() self.net = nn.Sequential( nn.Linear(1, hidden), nn.Tanh(), nn.Linear(hidden, hidden), nn.Tanh(), nn.Linear(hidden, hidden), nn.Tanh(), nn.Linear(hidden, 1) ) def forward(self, t): return self.net(t) model = RC_PINN() def rc_loss(model): # 物理残差点：t 在 [0, 5] 上取 200 个点 t_phys = torch.linspace(0, 5, 200).reshape(-1, 1).requires_grad_(True) u = model(t_phys) du_dt = torch.autograd.grad( u, t_phys, grad_outputs=torch.ones_like(u), create_graph=True )[0] residual = du_dt + u - 1.0 pde_loss = torch.mean(residual ** 2) # 初值约束：t = 0 时 u = 0 t0 = torch.zeros(1, requires_grad=True) u0 = model(t0) ic_loss = (u0 - 0.0) ** 2 # 初值权重给大一点 return pde_loss + 10.0 * ic_loss optimizer = torch.optim.Adam(model.parameters(), lr=1e-3) for step in range(8000): optimizer.zero_grad() loss = rc_loss(model) loss.backward() optimizer.step() if step % 1000 == 0: print(f"step {step:5d}, loss {loss.item():.2e}") # 验证：与解析解 1 - exp(-t) 对比 t_test = torch.linspace(0, 5, 500).reshape(-1, 1) u_pred = model(t_test).detach().numpy() u_exact = 1 - torch.exp(-t_test).numpy()

这个代码跑在普通 CPU 上也就十几秒。我实测下来，8000 步 Adam 之后，训练区间内最大绝对误差基本可以压到 1e-4 量级，曲线几乎和解析解重合。

2.3 训练技巧：怎么让网络乖乖收敛

RC 问题看着简单，实际训练中还是有不少坑，我讲几个自己踩过的。

第一个坑是初值项权重不够。初值权重太小，网络会补一个“偷懒解”：残差损失很低，但 u(0) 对不上，曲线整体平移，看起来好像也是方程的解，实际上解不唯一。对策很简单，把初值权重调到 10 甚至 100，或者用硬约束：把网络结构改成 u(t) = t·NN(t)，这样 t=0 时输出恒为 0，初值天然满足，网络只负责学余项。这种“硬约束”思想在后面复杂边界问题里更值得推广。

第二个坑是优化器选择。Adam 在 1e-3 学习率下能快速进入低 loss 区域，但后期容易原地踏步，loss 降到 1e-5 以下很费劲。我的习惯是前 2000 步用 Adam，后面切到 L-BFGS 做精修，或者把 Adam 学习率降到 1e-4、1e-5，收敛速度和最终精度都能提升不少。

第三个坑在采样区间的设置。RC 电路的无量纲解在 x 大于 5 之后基本趋近 1，残差也趋近 0，在 [0, 10] 或更大的区间训练收益不大，反而稀释了解变化最剧烈的 [0, 2] 区间的采样密度。把采样区间压到物理过程真正发生的范围，比盲目扩大采样范围更有效。

2.4 结果怎么验证：对比解析解

PINN 训练完，第一件事不是看 loss 多小，而是拿解析解逐点对比。RC 问题有现成的 1 - exp(-t)，直接在测试点上算最大绝对误差和平均误差，这是最有说服力的验证方式。

从误差分布上看，t 趋近于 0 的区域误差通常最大，原因是初值约束和残差约束两项的梯度在边界附近互相拉扯。这也是所有 PINN 问题的普遍现象：初始和边界附近的精度往往比内部低，做芯片热分析时同样要注意，靠近边界的高温梯度区通常是误差集中区。

这个例子跑通之后，你就具备了 PINN 的全部核心要素：采样点生成、残差损失构造、自动微分求导、优化器选择、解析解验证。后续不管遇到多么复杂的问题，本质都是这个 RC 例子的扩展，区别只在于方程复杂度、空间维数和采样策略。

3. 从 ODE 到 PDE：芯片热分析的核心问题

3.1 芯片热分析为什么难：多尺度、多热源、边界复杂

芯片热分析不是解一个热传导方程就完事，难在工程约束上。

第一是多尺度。芯片内部晶体管特征尺寸已经在纳米级，但整个封装件尺寸是厘米级，横跨好几个数量级。有限元网格要在这么大的尺度范围里做局部加密，剖分本身就是一种负担。第二是多热源。芯片不同功能模块功耗差异巨大，CPU 核心区和缓存区的热流密度能差好几倍，而且很多模块是间歇性工作，热源分布随时变化。第三是边界条件复杂。芯片外表面有散热器、封装材料和空气接触面，属于对流换热边界；底面和 PCB 之间有导热垫，还有与周围结构的绝热边界，这些边界形状不规则，传统 FVM 在网格生成上很痛苦。

PINN 的价值正是在这种场景下体现的：它不需要生成网格，只需要在计算域内和边界上撒点。边界复杂，就在边界参数化之后多采点样；热源分布复杂，就把热源函数写成空间坐标的函数喂给残差计算。对一份长期做仿真的人来说，这种“去网格化”的自由度太有吸引力了。

3.2 热传导方程与边界条件的数学化

一般三维瞬态热传导方程是：

ρ·c_p·∂T/∂t = ∂/∂x(k·∂T/∂x) + ∂/∂y(k·∂T/∂y) + ∂/∂z(k·∂T/∂z) + Q

其中 T(x,y,z,t) 是温度场，ρ 是密度，c_p 是比热容，k 是导热系数，Q 是单位体积生热率。

芯片热分析里经常做两个简化：一是材料各向同性，导热系数 k 取等效平均；二是在某些截面上做二维分析，因为芯片厚度方向的热流相对简单，可以先看 xy 平面的温度分布，三维问题降成二维泊松方程或带时间项的扩散方程。

稳态二维热传导（忽略时间项）可以写成：

k·(∂²T/∂x² + ∂²T/∂y²) + Q(x,y) = 0

也就是 -∇²T = Q/k。边界条件常见有三类，我把它们在 PINN 里的表达方式一起整理：

PINN 处理这三类边界的方式几乎一样：在边界采样点，把对应的方程残差写进损失函数。从 ODE 到 PDE 的过渡，在代码层面就是这么平滑。

4. 芯片热分析的 PINN 实战

4.1 二维稳态热传导的 PINN 实现思路

用一个简化但足够典型的芯片区域来说明：尺寸 1 cm × 1 cm，中心有一小块热源区域（比如中心 0.3 cm × 0.3 cm 的范围发热，Q = 100 W/cm³），其余区域无热源，四周边界温度固定为室温 25 度。这是一个带内热源的二维 Dirichlet 问题。

网络输入从一维的 t 变成二维的 (x, y)，输出是温度 T，损失函数由 PDE 残差和边界残差组成。核心代码片段如下：

import torch import torch.nn as nn class HeatPINN(nn.Module): def __init__(self, hidden=80): super().__init__() self.net = nn.Sequential( nn.Linear(2, hidden), nn.Tanh(), nn.Linear(hidden, hidden), nn.Tanh(), nn.Linear(hidden, hidden), nn.Tanh(), nn.Linear(hidden, hidden), nn.Tanh(), nn.Linear(hidden, 1) ) def forward(self, x, y): return self.net(torch.cat([x, y], dim=1)) def heat_loss(model, k, Q_func): # 计算域内部采样 n_in = 2000 x_in = torch.rand(n_in, 1) * 1.0 y_in = torch.rand(n_in, 1) * 1.0 x_in = x_in.requires_grad_(True) y_in = y_in.requires_grad_(True) T = model(x_in, y_in) T_x = torch.autograd.grad(T, x_in, grad_outputs=torch.ones_like(T), create_graph=True)[0] T_y = torch.autograd.grad(T, y_in, grad_outputs=torch.ones_like(T), create_graph=True)[0] T_xx = torch.autograd.grad(T_x, x_in, grad_outputs=torch.ones_like(T_x), create_graph=True)[0] T_yy = torch.autograd.grad(T_y, y_in, grad_outputs=torch.ones_like(T_y), create_graph=True)[0] # 方程：-∇²T = Q/k，残差 = T_xx + T_yy + Q/k residual = T_xx + T_yy + Q_func(x_in, y_in) / k pde_loss = torch.mean(residual ** 2) # 边界采样：四条边 T = 25，这里只写一条边的示例 n_bc = 200 x_bc = torch.rand(n_bc, 1) y_bc = torch.zeros_like(x_bc) T_bc = model(x_bc, y_bc) bc_loss = torch.mean((T_bc - 25.0) ** 2) # 其余三条边同理，代码省略 return pde_loss + 20.0 * bc_loss

这个思路对工程场景非常友好：热源函数 Q(x,y) 不一定要写成简洁的解析形式，如果功耗分布来自仿真数据，完全可以做成插值函数喂给残差计算，PINN 照样能学。

4.2 关键细节：权重点采样、归一化、多任务损失加权

芯片热分析比 RC 电路复杂得多，有几个细节直接决定成败。

第一是采样策略。均匀随机采样简单但效率一般，因为温度场在热源附近变化剧烈，远离热源的角落变化平缓。我的建议是“分区域采样加局部加密”：先在全局撒一批点保证覆盖，再在热源边界和可能的高温梯度区多撒一批点，把网络注意力引导到难点区域。网格加密是静态的，PINN 的采样加密可以动态调整，这算是一个很大的自由度。

第二是归一化。坐标 x、y 如果直接用厘米或米为单位，数值量级要统一，最好归一到 [0,1] 或 [-1,1]。温度也做平移缩放，比如把输出变成 (T - 25) / (T_max - 25)，让输出在 O(1) 量级。我见过太多人忽略这一点，结果训练半天 loss 不降，归一化之后几十步就下来了。

第三是多任务损失加权。PDE 残差、边界残差、数据残差的量级天然不同。观察 loss 曲线时，经常是 PDE 残差降得很快，边界残差却卡着不动，问题往往出在权重上。多个任务同时优化时，权重这个超参数相当敏感，固定权重只是一个基础做法。我实际项目中会监控各项 loss 的量级，动态调整权重，保证没有一个任务被“淹没”。

4.3 从代码到工程：PINN 在芯片热分析中的定位

把话说透，PINN 暂时不可能完全取代有限元/有限体积法，芯片热仿真领域的主流商业软件依然是 FEM/FVM 的天下。但 PINN 在几个特定场景下非常有价值。

第一个是快速重算场景。芯片布局稍有改动，传统方法要重新剖分网格、重新求解，算一次几十分钟到几小时。PINN 训练一次以后，对微调参数的问题可以复用之前的训练结果做迁移学习，几分钟内给出新温度场，这对方案迭代阶段很有用。

第二个是稀疏实测数据融合场景。芯片上只有少数几个热敏位置有实测温度，传统模拟难以直接把这些散点数据融合进去，但 PINN 可以非常自然地加一个数据损失项，把实测和仿真揉在一起。这个能力在模型验证环节很解渴。

第三个是反问题场景。比如已知芯片背面温度分布，反推芯片内部热源分布，这是经典的反问题。PINN 天生适合：把热源、导热系数等作为额外待优化参数，网络和这些参数一起优化，一次训练既得到温度场，又得到热源反演结果。

所以我的建议是：不要纠结 PINN 能不能全面替代 FEM，而是把它当成工具箱里的一把新扳手，在合适的场景去用它。RC 电路就是磨这把扳手的第一个操作。

5. 常见问题与排查实录

5.1 网络不收敛，Loss 卡在某个值下不去

这是 PINN 新手最常遇到的问题，按概率高低排查即可。

第一步检查采样点数量和范围。如果只取了几十个点，方程约束太弱，网络学不出唯一解，加大到几百上千个点往往立刻见效。第二步检查边界/初值权重。权重太小会导致边界不匹配，整体解被带偏，把边界权重调到 10 以上，大多数 RC 级别的问题都能解决。第三步检查激活函数。ReLU 在 PINN 里容易出问题，换 tanh 是性价比最高的调整。第四步切换优化器。Adam 后期容易原地打转，换 L-BFGS 或降低学习率做精修，往往能把 loss 再压两个数量级。

5.2 边界条件不满足，边界附近误差偏大

如果内部温度场看起来合理，但边界上数值不符合约束，这是软约束权重不够或边界采样太稀疏导致的。边界误差偏大时，先提高边界点密度，再提高权重，通常会有改善。如果还不行，就把边界条件做成硬约束。

硬约束的做法是重构网络输出：令 T(x,y) = T_bc(x,y) + d(x,y)·NN(x,y)，其中 d(x,y) 是到边界的距离函数，在边界上为 0。这样不管 NN 输出什么，T 在边界上都自动等于 T_bc(x,y)。距离函数 d 对矩形区域很好写，对复杂几何不一定有解析形式，但芯片分析里大量区域可以近似成矩形、圆形或多边形组合，这套方案多数时候都用得上。

5.3 训练慢、显存不够怎么办

热分析是二维或三维问题，输入维度比 ODE 高，网络规模稍大，GPU 显存紧张很常见。

我的习惯是控制计算图规模：每个训练步重新采样，而不是一次性生成一万个固定点。这样每个 step 的计算图只包含当前批次的点，显存占用大幅降低。采样点数量在显存和精度之间取平衡，二维问题内部 2000 点、边界 500 点，对大多数情况已经够用。

另一个技巧是分段训练。先用少量点训练出粗糙解，loss 稳定后增加采样点做精细迭代，相当于从粗网格到细网格的自适应策略，训练效率提升非常明显。芯片热分析这种多尺度问题，这种粗到细的迭代方式比一次性全精度训练更稳定。

5.4 认准这套工具链：PyTorch + DeepXDE + MATLAB

分享一下我现在常用的工具组合，供参考。

纯手写 PyTorch 适合学习和定制。RC 电路和简化二维热问题，手写完全够用，改损失函数、改采样方式都方便，遇到问题也能完全控制在自己手里。

如果工程问题更复杂，强烈推荐 DeepXDE 这个库，它专门为 PINN 设计，内置大量 PDE、边界条件和优化器配置，还有自适应采样功能。写热传导问题基本几十行就能跑起来，比自己从零手搓省心得多。

至于 MATLAB，确实有很多朋友问怎么搭 PINN。MATLAB 可以用 Deep Learning Toolbox 里的 dlarray 和 dlgradient 手动实现自动微分和损失计算，思路跟 PyTorch 几乎一样：定义网络、定义输入点、用 dlgradient 求导算残差、更新参数。只是 MATLAB 的自动微分生态相对弱一些，大批量采样时速度也明显不如 PyTorch。如果团队主要是 MATLAB 用户，做算法验证用 MATLAB 没问题，真跑工程案例还是切到 Python 生态更顺畅。

最后讲一点我自己很深的体会：从 RC 电路入手，两天内就能把 PINN 的完整流程摸透，然后切到芯片热分析，剩下的就只是工程细节的填充，而不是原理性的障碍。很多朋友一上来就想啃三维复杂几何、多材料、瞬态问题，结果被一堆问题折磨得想放弃。我的建议是，不管最终目标多复杂，都先拿 RC 电路和一维热传导练手，把损失函数配平、自动微分调通、边界约束做对，再逐步叠加工程复杂性。这条路我验证过很多次，值得再走一遍。