Sharp Stability Threshold and Certification for Designing Stable Residual Architectures
On the map: State-space sequence models (S4, Mamba) → Core SSM architectures for language
Abstract (arXiv)
Whether a deep residual architecture trains stably is usually determined by training it, which is expensive and answers the question only for the architecture, depth, and floating-point format tested. In practice, stability is secured by heuristics for where to place normalization and which type to use, supported by experiments but lacking a common principle. We show that stability can instead be certified before training, directly from the architectural primitives of a residual block, as an explicit function of depth and floating-point format. The certificate has two elements: (a) a power-law growth bound, |v(x)|≤ c|x|q+b, whose exponent q is computed from the block's primitives by an arithmetic of exponents (forward); and (b) gradient bounds from a Lipschitz condition on the block over reachable states (backward). The certificate determines when the state can reach the largest finite floating-point value M: for q≤1, overflow requires at least order log M layers, whereas for q>1 it can occur within order logqlog M layers; the threshold q=1 is sharp. Every normalization and bounded activation sets q=0, and the arithmetic identifies minimal modifications that bring a block from q>1 to q=1 without normalization. Empirically, in forecasting, only q>1 blocks blow up, as predicted by the forward certificate. In GPT-2 on OpenWebText, q=1 blocks without normalization also diverge: the forward certificate holds, while the backward one fails in the attention block with the largest query-key coefficient. Relaxing normalization from q=0 to q=1 improves out-of-distribution generalization in operator learning and accuracy in time-series forecasting. As deep models become more complex and costly to train, our results provide a way to assess the trade-off between stability and representational flexibility directly from block primitives, before training.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Core SSM architectures for language, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 23 of its 32 references that are in the map. First posted on arXiv 16 Jul 2026.
Builds on
23 of 32 references.
Closest ancestors · 1
Same direction, newest first
Other references · 22
- 2017Attention Is All You Needcited by 59,932
- 2016Layer Normalizationcited by 5,579
- 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,827
- 2016Gaussian Error Linear Units (GELUs)cited by 3,431
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
- 2020Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecastingcited by 2,141
- 2022A Time Series is Worth 64 Words: Long-term Forecasting with Transformerscited by 1,981
- 2021Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecastingcited by 1,654
- 2018Spectral Normalization for Generative Adversarial Networkscited by 1,618
- 2019Root Mean Square Layer Normalizationcited by 1,425
- 2017Sigmoid-Weighted Linear Units for Neural Network Function Approximation in Reinforcement Learningcited by 935
- 2020On Layer Normalization in the Transformer Architecturecited by 906
10 more
- 2021Going deeper with Image Transformerscited by 759
- 2020ReZero is All You Need: Fast Convergence at Large Depthcited by 260
- 2019Fixup Initialization: Residual Learning Without Normalizationcited by 194
- 2024GIFT-Eval: A Benchmark For General Time Series Forecasting Model Evaluationcited by 188
- 2020The Lipschitz Constant of Self-Attentioncited by 171
- 2019Transformers without Tears: Improving the Normalization of Self-Attentioncited by 153
- 2025Transformers without Normalizationcited by 69
- 2025Peri-LN: Revisiting Normalization Layer in the Transformer Architecturecited by 32
- 2026Toto 2.0: Time Series Forecasting Enters the Scaling Eracited by 21
- 2025Bounded Hyperbolic Tangent: A Stable and Efficient Alternative to Pre-Layer Normalization in Large Language Modelscited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Dual-End Consistency Model
- 2026Retimed Bellman Flows: Escaping the Impossible Triangle of Velocity Bootstrapping
- 2025Stability and Generalization for Bellman Residuals
- 2026Geometry-Preserving Neural Architectures on Manifolds with Boundary
- 2024Stable Autonomous Flow Matching
- 2026Optimal Self-Distillation for Rectified Flow via Linear Probing
- 2025Elucidating the Design Space of FP4 training
- 2026Stabilizing Physics-Informed Consistency Models via Structure-Preserving Training