用迭代物理监督训练神经泊松求解器,不依赖标签也能高效求解复杂几何问题。
NPSolver: Neural Poisson Solver with Iterative Physics Supervision

- 通过预条件共轭梯度步数提供稳定训练信号,避免传统残差损失的不稳定性。
- 在二维和三维不规则几何上优于现有物理引导与数据驱动模型,误差更低。
- 适合需要快速、可靠求解泊松方程的科学计算与优化控制场景。
在复杂不规则域上高效求解泊松方程仍是科学计算中的基本挑战,经典迭代求解器常因病态系统导致运行时间过长。尽管神经算子提供了快速替代方案,但通常依赖大规模标注数据,或在使用物理约束残差损失时面临训练不稳定问题。我们提出 extsc{NPSolver},一种无需解标签即可训练的神经泊松求解器。它不依赖完全收敛的数值解或原始PDE残差,而是利用少量预条件共轭梯度(PCG)步骤来精炼自身预测,提供更稳定且尺度合理的训练信号。理论分析表明,这种迭代监督作为良好条件化的误差代理有效,且停止梯度设计对优化稳定性至关重要。为进一步捕捉混合边界条件下边界驱动特征,我们引入边界感知的Transolver( extsc{BA-Transolver})架构,显式分离内部与边界标记。在二维和三维不规则几何上的大量实验表明, extsc{NPSolver} 超越了物理引导与数据驱动的基线模型。此外,下游热控任务验证了该模型在执行高效可靠梯度边界控制方面的潜力。代码与数据将开源于 https://github.com/intell-sci-comput/NPSolver。
原文摘要 · Abstract (English)
Efficiently solving Poisson equations on complex, irregular domains remains a fundamental challenge in scientific computing, as classical iterative solvers often suffer from prohibitive runtime due to ill-conditioned systems. While neural operators offer a fast alternative, they typically rely on large-scale labeled datasets or struggle with unstable training dynamics when using physics-informed residual losses. We propose \textsc{NPSolver}, a neural Poisson solver trained without solution labels via iterative physics supervision. Instead of relying on fully converged numerical solutions or raw PDE residuals, \textsc{NPSolver} utilizes a small number of preconditioned conjugate gradient (PCG) steps to refine its own predictions, providing a more stable and well-scaled training signal. Theoretical analysis confirms that this iterative supervision serves as a well-conditioned error proxy and that a stop-gradient design is essential for optimization stability. To better capture boundary-driven features under mixed boundary conditions, we further introduce the Boundary-Aware Transolver (\textsc{BA-Transolver}) architecture that explicitly separates interior and boundary tokenization. Extensive evaluations on 2D and 3D irregular geometries demonstrate that \textsc{NPSolver} outperforms both physics-informed and data-driven baselines. Furthermore, a downstream thermal control task highlights the model's capability for conducting efficient and reliable gradient-based boundary control. We will release our codes and data at https://github.com/intell-sci-comput/NPSolver.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。