Modularity-Free Conflict-Averse Training for Generalized PINNs

Heejo Kong, Beomchul Park, Sung-Jin Kim, Seong-Whan Lee
6/18/2026
cs.AI

Official code: heejokong/ModSync (2 stars, last commit 2026-05-21).

Abstract

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}.

View on arXivView PDF

Code Implementations(1)

heejokong/ModSyncOfficial100%

[ICASSP 2026] Official PyTorch Implementation of "Modularity-Free Conflict-Averse Training for Generalized PINNs"

20Shell, PythonDec 22, 20255 months ago
multi-task-learningphysics-informed-machine-learningphysics-informed-neural-networks

Similar Papers

Cite this paper

@article{kong2026modularityfree,
  title  = {Modularity-Free Conflict-Averse Training for Generalized PINNs},
  author = {Heejo Kong and Beomchul Park and Sung-Jin Kim and Seong-Whan Lee},
  year   = {2026},
  eprint = {2606.20156},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2606.20156v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2606.20156.svg)](https://codeofpaper.com/paper/2606.20156)

Discussion