用条件损失训练神经网络,一次性解多个参数化的微分方程。
Loss-Conditional PINNs for Parametric PDE Families
- 将损失权重作为输入,让网络学习不同权重下的解族。
- 一个模型同时逼近多组参数的方程解,性能优于逐个重训的基线。
- 适合需要快速求解参数化方程的科研与工程场景。
物理信息神经网络(PINNs)通过最小化残差、边界、初值和数据损失的加权组合来逼近常微分方程和偏微分方程的解。其性能常受损失权重选择影响:不当权重会导致训练陷入退化解,仅满足部分物理约束。现有方法通常寻找或调整一组最优权重。本文提出新思路:不固定权重,而是在训练中探索整个权重空间。引入LC-PINN,将Dosovitskiy与Djolonga(2020)提出的损失条件训练拓展至PDE残差场景:将损失权重或物理系数作为网络输入,每步优化时从简单先验中采样。这使训练转化为学习由该向量索引的连续解族,无需求解器生成配对数据。LC-PINN介于经典PINNs与算子学习之间——保持完全物理一致性,但可对参数族进行训练摊销。贡献包括:首次将损失条件机制应用于PINNs;统一损失权重与参数系数两种情形(权重用拼接,系数用FiLM);提出固定积分的L-BFGS收尾协议,使参数系数情形可训练。证明了条件最优解的lambda-不变性,并在参数化亥姆霍兹、薛定谔、黏性布格尔斯及布克利-利弗特方程上验证。单个LC-PINN在总成本低于逐实例重训的前提下,达到或超越了基线性能。
原文摘要 · Abstract (English)
Physics-informed neural networks (PINNs) approximate solutions of ODEs and PDEs by minimising a weighted combination of residual, boundary, initial, and data losses. Their performance is often dominated by the choice of loss weights: a poor weighting can drive training to a degenerate solution in which one physical constraint is satisfied while another is ignored. Existing methods select or adapt a single good set of weights. We take a different view: instead of tuning one weight vector, we explore the entire weight space during training. We introduce LC-PINN, which adapts the loss-conditional training of Dosovitskiy and Djolonga (2020) to the PDE-residual setting: the conditioning vector (either the loss weights or a scalar physical coefficient) is treated as a network input and sampled from a simple prior at every optimisation step. This turns PINN training into learning a continuous family of solutions indexed by that vector, with no solver-generated paired data. LC-PINN thus lies between classical PINNs and operator learning: it stays fully physics-informed but amortises training over a parametric family. Our contribution is not the loss-conditional construction itself, but its extension to PINNs, the unification of the loss-weight and parametric-coefficient regimes under one architecture (concatenation for loss weights, FiLM for coefficients), and a fixed-quadrature L-BFGS finishing protocol that makes the parametric-coefficient regime trainable. We give a lambda-invariance result for the conditional optimum and study LC-PINN on parametric Helmholtz, Schrodinger, viscous Burgers, and Buckley-Leverett equations. A single LC-PINN matches or improves retrained per-weight PINN baselines while parameterising the full family in one model, at a total cost that amortises favourably against per-instance retraining.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。