Response Renormalization for Critical Deep Equilibrium Models
Not on the map. Why?
Abstract (arXiv)
Deep Equilibrium Models (DEQs) compute predictions from a hidden representation unchanged by the model update. Training through this equilibrium uses implicit differentiation and requires solving an adjoint system built from the residual Jacobian. If this Jacobian is nearly singular along loss-sensitive directions, small perturbations can be strongly amplified in the adjoint response, producing large, highly sensitive gradients that can make optimization unreliable. We introduce Response Renormalization, a backward-pass framework that lifts selected near-pole denominators while leaving unlifted response channels unchanged. Collective Mode Response Renormalization (CMR) applies this correction in a low-dimensional critical subspace, while Phi-adaptive CMR computes a bounded response mass from a positive susceptibility rule. We derive dense and matrix-free collective formulations, distinguish exact gradients of a modified frozen-anchor residual from backward-response surrogates, and extend the construction to Structured Implicit Layers and Vector Attractors (SILVA). Across 23 multiphysics families spanning partial differential equations, three-dimensional fields, operator maps, complex geometries, and particle systems, CMR and Phi-CMR yield test errors no more than five percent higher than those from models trained with exact implicit differentiation in more than 98% of static and 95% of transient family-seed comparisons. Solver-index experiments show convergence toward the static adjoint, while physical-time rollouts retain predictive fidelity under the evaluated conditions. These results demonstrate that selective response renormalization can control near-critical adjoint amplification without globally damping well-conditioned sensitivity. Therefore, the method can make parameter updates more reliable while preserving the useful gradient information needed for learning.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 23 of its 40 references that are in the map. First posted on arXiv 24 Aug 2026.
Builds on
23 of 40 references.
References · 23
- 2015Automatic differentiation in machine learning: a surveycited by 855
- 2019Deep Equilibrium Modelscited by 615
- 2022PDEBENCH: An Extensive Benchmark for Scientific Machine Learningcited by 413
- 2024Poseidon: Efficient Foundation Models for PDEscited by 200
- 2022Towards Multi-spatiotemporal-scale Generalized PDE Modelingcited by 197
- 2020Multiscale Deep Equilibrium Modelscited by 151
- 2024The Well: a Large-Scale Collection of Diverse Physics Simulations for Machine Learningcited by 145
- 2019Implicit Deep Learningcited by 124
- 2020Monotone operator equilibrium networkscited by 100
- 2021JFB: Jacobian-Free Backpropagation for Implicit Networkscited by 86
- 2021On Training Implicit Modelscited by 75
- 2021Stabilizing Equilibrium Models by Jacobian Regularizationcited by 72
11 more
- 2023CFDBench: A Large-Scale Benchmark for Machine Learning Methods in Fluid Dynamicscited by 46
- 2023LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suitecited by 27
- 2021SHINE: SHaring the INverse Estimate from the forward pass for bi-level optimization and implicit modelscited by 26
- 2023TorchDEQ: A Library for Deep Equilibrium Modelscited by 21
- 2023Deep Equilibrium Based Neural Operators for Steady-State PDEscited by 19
- 2022Deep equilibrium networks are sensitive to initialization statisticscited by 8
- 2023DynaBench: A benchmark dataset for learning dynamical systems from low-resolution datacited by 8
- 2024Deep Equilibrium Algorithmic Reasoningcited by 6
- 2025Reversible Deep Equilibrium Modelscited by 5
- 2016Functional methods for disordered neural networkscited by 4
- 2026SILVA Networks as Structured Implicit Layers and Vector Attractors via Dynamic Interaction Fieldscited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.