Modularity-Free Conflict-Averse Training for Generalized PINNs
About
Physics-informed neural networks (PINNs) have become a powerful framework for solving PDEs by embedding physical laws into differentiable objectives. Despite their advances, training PINNs remains fragile: recent conflict-averse optimization schemes alleviate gradient interference between residual and boundary losses, but we show that their effectiveness deteriorates as model capacity increases. In this paper, we identify a capacity-induced failure mode, where overparameterized networks undergo functional modularity, self-partitioning into task-exclusive modules that suppress cross-objective interaction and hinder convergence toward Pareto-stationary points. To address this issue, we propose a novel framework, Modular-Sparsity Synchronization (ModSync), which integrates structural optimization into conflict-averse training by penalizing task-exclusive connections while preserving interaction-promoting pathways. Extensive experiments across diverse PDE benchmarks demonstrate that ModSync consistently prevents capacity-driven failures, sustains robust cross-objective coupling, and achieves state-of-the-art accuracy. Codes are available at \url{https://github.com/heejokong/ModSync}.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Solving partial differential equations | Helmholtz 2D (test) | Absolute Error0.0384 | 10 | |
| Solving partial differential equations | Helmholtz 3D (test) | Absolute Error19.86 | 10 | |
| Solving partial differential equations | Klein-Gordon 2D (test) | Absolute Error0.07 | 10 | |
| Solving partial differential equations | Burgers' (test) | Absolute Error0.0321 | 10 | |
| Solving partial differential equations | Klein-Gordon 3D (test) | Absolute Error4.274 | 10 |