cs.LGApr 21, 2026

Physics-Guided Dimension Reduction for Simulation-Free Operator Learning of Stiff Differential-Algebraic Systems

Authors: Huy Hoang Le, Haoguang Wang, Christian Moya, Marcos Netto, Guang Lin

Organizations: School of Mechanical Engineering, Purdue University, West Lafayette, IN 47907 USA · Department of Mathematics, Purdue University, West Lafayette, IN 47907 USA · Department of Electrical and Computer Engineering, New Jersey Institute of Technology, Newark, NJ 07102 USA

Abstract

Neural surrogates for stiff differential-algebraic equations (DAEs) face two barriers: soft-constraint methods leave algebraic residuals that stiffness amplifies into errors, and hard-constraint methods require trajectory data from stiff integrators. We introduce an extended Newton implicit layer that enforces algebraic constraints exactly and reduces fast dynamics to their quasi-steady-state values in a single differentiable solve. Embedded in a physics-informed DeepONet, the layer recovers all fast and algebraic states exactly from slow-state predictions, removes the per-window stiffness-amplification pathway, and yields a stiffness-scaled Implicit Function Theorem gradient absent from penalty methods. Cascaded implicit layers extend this to multi-component systems with provable convergence. On a grid-forming inverter (stiffness ratio of about 4712), extended Newton attains 1.42% error versus 39.3% (penalty) and 57.0% (standard Newton); augmented Lagrangian and feedback linearization diverged. Two independently trained models compose without retraining (0.72% to 1.16% error, exact constraint satisfaction). Cross-domain validation on the Robertson stiff DAE (stiffness ratio up to 10510^5) confirms generalization. Conformal prediction provides 90% coverage with automatic out-of-distribution detection.

Explore similar work

CardsList