acceptodds
Under review as a conference paper at ICLR 2027

Gradient–Update Mismatch: Rethinking Conflict-Free Training of Physics-Informed Neural Networks

Abstract

Training Physics-Informed Neural Networks (PINNs) requires jointly optimizing physics residual and initial/boundary condition loss terms, which often induce conflicting gradients. Gradient surgery methods mitigate this issue by constructing directions from loss-specific gradients to reduce conflict before optimizer transformation. However, even when the constructed direction is conflict-free, this property may not be preserved after optimizer transformation. Let denote the direction constructed by gradient surgery, the optimizer proposal, and the conflict-free cone induced by the loss-specific gradients. We show that modern optimizers can transform through mechanisms including historical state, adaptive scaling, preconditioning, and decoupled weight decay, so does not generally imply . We refer to this optimizer-induced discrepancy in conflict-freeness between and as **Gradient-Update Mismatch** (GUM). Accordingly, we propose **Gradient-Update Alignment** (GUA), which performs conflict handling on the update actually applied to the parameters. Specifically, after the optimizer transforms into , GUA projects onto to obtain the aligned update , which is then applied to the parameters. For optimizers with internal state, GUA further aligns the state with targets derived from the applied update. Extensive experiments show that GUM is widespread across diverse optimizers, with conflict rates reaching up to 86.3%. Across all PINN settings, GUA achieves conflict-free applied updates and consistently improves various gradient surgery methods, reducing relative error by up to 98.2%. Data and code are available at [https://anonymous.4open.science/r/GUA-F010](https://anonymous.4open.science/r/GUA-F010).

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.