cs.LGMay 11, 2026

Per-Loss Adapters for Gradient Conflict in Physics-Informed Neural Networks

Authors: Bum Jun KimGnankan Landry Regis N'guessan

Organizations: The University of Tokyo, Japan · Axiom Research Group · Department of Applied Mathematics and Computational Science, NM-AIST, Tanzania · Tanzania · African Institute for Mathematical Sciences (AIMS), Research and Innovation Centre, Rwanda · Rwanda

Abstract

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=3K=3 benchmarks and a natural K=4K=4 thermoelastic system, where adapters combined with reweighting yield significant improvements. In contrast, K=3K=3 inverse problems and natural K=5K=5 and K=6K=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.

Explore similar work

CardsList