Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Bum Jun Kim, Gnankan Landry Regis N'guessan• 2026

Related benchmarks

TaskDatasetResultRank
PDE solvingKlein-Gordon equation
Relative L2 Error1.5
31
Forward PDE solvingHelmholtz
Relative Error0.0227
26
Forward PDE solvingForward benchmarks 5 PDEs
P(A<B)100
21
Forward PDE problem solvingBurgers
Relative L2 Error0.0046
19
Forward PDE solvingHelmholtz 10K-epoch
Relative L2 Error1
16
Forward PDE solvingAllen–Cahn 10K-epoch
Relative L2 Error1.19
16
Forward PDE solvingConv-Diff 10K-epoch
Relative L2 Error1
16
Forward PDE solvingKlein-Gordon 10K-epoch
Relative L2 Error1.6
16
Forward PDE solvingBurgers 10K-epoch
Relative L2 Error4.6
16
Solving coupled thermoelastic PDEsNatural coupled thermoelastic K = 4 1.0 (val)
L2 Error4.15
10
Showing 10 of 44 rows

Other info

Follow for update