Per-Loss Adapters for Gradient Conflict in Physics-Informed Neural Networks
About
Physics-informed neural networks (PINNs) train a single neural approximation by minimizing multiple physics- and data-derived losses, but the gradients of these losses often interfere and can stall optimization. Existing remedies typically treat this pathology either through scalar loss balancing or full-parameter-space gradient surgery, leaving it unclear which intervention is most appropriate. We show that PINN gradient conflict is not a uniform failure mode with one universal remedy. Instead, we identify distinct PINN gradient-conflict regimes, each associated with a different intervention class. Persistent directional conflict may require separate loss-indexed parameter subspaces, magnitude imbalance often favors scalar reweighting, and low or transient conflict may require no extra mitigation. To select between scalar reweighting and a lightweight architectural intervention, we propose a diagnostic-first framework. It profiles a 1000-step unmodified PINN run and, when intervention is warranted, uses one low-rank adapter per loss to create explicit loss-indexed parameter subspaces attached to a shared PINN trunk, providing each loss with a direct gradient pathway. Across more than 60 PDE configurations, including forward, inverse, multi-physics, parameter-varying, and high-dimensional problems up to 50D, persistent directional conflict dominates standard forward $K=3$ benchmarks and a natural $K=4$ thermoelastic system, where adapters combined with reweighting yield significant improvements. In contrast, $K=3$ inverse problems and natural $K=5$ and $K=6$ multi-physics systems are largely magnitude-dominated and often favor reweighting alone, while full-parameter-space gradient surgery can fail on heterogeneous parameter spaces.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| PDE solving | Klein-Gordon equation | Relative L2 Error1.5 | 31 | |
| Forward PDE solving | Helmholtz | Relative Error0.0227 | 26 | |
| Forward PDE solving | Forward benchmarks 5 PDEs | P(A<B)100 | 21 | |
| Forward PDE problem solving | Burgers | Relative L2 Error0.0046 | 19 | |
| Forward PDE solving | Helmholtz 10K-epoch | Relative L2 Error1 | 16 | |
| Forward PDE solving | Allen–Cahn 10K-epoch | Relative L2 Error1.19 | 16 | |
| Forward PDE solving | Conv-Diff 10K-epoch | Relative L2 Error1 | 16 | |
| Forward PDE solving | Klein-Gordon 10K-epoch | Relative L2 Error1.6 | 16 | |
| Forward PDE solving | Burgers 10K-epoch | Relative L2 Error4.6 | 16 | |
| Solving coupled thermoelastic PDEs | Natural coupled thermoelastic K = 4 1.0 (val) | L2 Error4.15 | 10 |