English

Gradient Scaling Effects in Adaptive Spectral PINNs for Stiff Nonlinear ODEs

Machine Learning 2026-05-07 v1

Abstract

Physics-Informed Neural Networks (PINNs) often struggle to train reliably on stiff and oscillatory dynamical systems due to poor optimization conditioning. While prior work has emphasized representational remedies such as spectral parameterizations, the optimization implications of initial-condition (IC) embeddings in adaptive spectral PINNs have not been well characterized. In this work, we show that the choice of IC gating function induces explicit time-dependent gradient scaling, which interacts with spectral representations during training. Using a nonlinear stiff spring-pendulum ODE as a controlled benchmark, we compare exponential and linear IC gates in combination with fixed and adaptive Fourier spectral trunks. We observe stiffness-dependent changes in relative dominance for adaptive PINNs: at moderate stiffness (k=20k=20), exponential gating often yields lower error but exhibits heterogeneous behavior across random seeds, whereas at higher stiffness (k=60k=60), linear gating becomes preferable, with additional reversals observed at larger kk. These trends hold for both relative L2L^2 error and maximum pointwise error and are confirmed by paired Wilcoxon signed-rank tests with Holm correction. Overall, our results demonstrate that IC embeddings are not a neutral design choice in PINNs: the induced gradient scaling materially shapes optimization conditioning in stiff regimes, with distinct sensitivity patterns in baseline and adaptive spectral models.

Cite

@article{arxiv.2605.04502,
  title  = {Gradient Scaling Effects in Adaptive Spectral PINNs for Stiff Nonlinear ODEs},
  author = {Isabela M. Yepes and Pavlos Protopapas},
  journal= {arXiv preprint arXiv:2605.04502},
  year   = {2026}
}

Comments

8 pages, 4 figures, 1 table. This work appeared at the ICLR 2026 AI&PDE Workshop on OpenReview

R2 v1 2026-07-01T12:52:09.996Z