让神经网络学会自我纠错,大幅提升物理方程求解精度。
Error-Conditioned Neural Solvers

- 将残差场作为输入,让网络学习如何修正自身误差
- 在湍流问题上精度提升达10倍,且无需昂贵计算
- 特别适合参数变化大或方程不稳定的场景
神经代理模型可快速近似偏微分方程(PDE)参数到解的映射,但通常仅视求解为统计任务:训练后难以纠正约束违反,也无法外推至训练分布之外。现有混合方法通过梯度下降或高斯-牛顿法优化残差以保证物理正确性,但继承了传统优化器的高计算成本和不稳定性。我们理论与实证表明,在病态系统中,最小化残差未必能提升重建精度,解释了为何这些方法虽残差低却预测不准。为此提出误差条件神经求解器(ENS),其核心思想是将残差场作为每一步迭代的直接输入,使网络能读取自身误差的空间结构,并学习迭代修正策略。在四类PDE中,ENS在绝大多数设置下达到最高精度,湍流柯尔莫戈洛夫流中最高提升10倍,同时避免了混合方法的高昂计算开销。其学习到的修正策略具有强泛化能力,可在零样本参数变化与跨方程迁移中表现优异,尤其在残差最小化最不可靠的病态情形下优势最大。
原文摘要 · Abstract (English)
Neural surrogate models offer fast approximate mappings from PDE parameters to solutions, but they typically treat solving as a purely statistical task: once trained, they struggle to correct their own constraint violations and extrapolate beyond the training distribution. Recent hybrid methods promote physical correctness by targeting the PDE residual via gradient descent or Gauss--Newton steps, but inherit the compute cost and instability of the underlying classical optimizers. We show, theoretically and empirically, that numerically minimizing the PDE residual can be an unreliable proxy for reconstruction accuracy in ill-conditioned systems, explaining why these methods often do not make accurate predictions despite achieving low residuals. We propose error-conditioned Neural Solvers (ENS), built on a different principle: rather than an optimization target, the PDE residual field is passed as a direct input to the network at each iteration, enabling it to read the spatial structure of its own errors and learn an update policy to iteratively correct its predictions. Across four PDE families, ENS attains the highest prediction accuracy in the large majority of settings, with gains reaching $10\times$ on turbulent Kolmogorov flow, while avoiding the expensive compute cost of hybrid methods. ENS's learned correction policy generalizes under distribution shift, including zero-shot parameter changes and cross-equation transfer, where its relative advantage is largest in the ill-conditioned regimes where residual minimization is least reliable. Project website: https://neuralsolver.github.io/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。