mlsanityData to 7 Oct 2026

Numerical Fragility in Transformers: A Layer-wise Theory for Risk Estimation and Selective Stabilization

arXiv 2510.21770 ↗First posted 17 Oct 2025cs.LG cs.NA math.NAcited by 0

Not on the map. Why?

Abstract (arXiv)

Low-precision execution can induce substantial forward discrepancies in Transformers even for fixed weights and input, yet these discrepancies are usually monitored only at the output and lack a layer-wise theoretical account. We develop a first-order decomposition of output mismatch into layer-local attention, LayerNorm, and residual-transport terms, and derive from it a practical causal risk estimator and a budgeted controller, Bound-Guided Selective Stabilization (BGSS). Controlled sweeps verify the predicted local sign, monotonicity, and transport structure. On GPT-2, the transport-aware combined predictor is positively correlated with FP32-reference mismatch in all 18 runs and improves over a no-transport ablation in 17/18 runs. Reference-patch attribution shows that the same score preserves useful layer ordering information (mean Spearman 0.362). In budget-matched mitigation, BGSS outperforms random same-budget control in onset events (10.67 vs. 11.67), final mismatch (1.243× 10-3 vs. 1.284× 10-3), and worst-case mismatch (3.14× 10-3 vs. 8.49× 10-3), while matching a risk-only same-budget controller on onset suppression and sharply reducing worst-case mismatch (3.14× 10-3 vs. 5.71× 10-3). These results support a theory-to-algorithm account of Transformer numerical fragility in which finite-precision risk can be analyzed, estimated, localized, and selectively stabilized.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 15 of its 34 references that are in the map. First posted on arXiv 17 Oct 2025.

Builds on

15 of 34 references.

References · 15

  1. 2017Attention Is All You Needcited by 59,932
  2. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  3. 2016Layer Normalizationcited by 5,579
  4. 2016Pointer Sentinel Mixture Modelscited by 2,992
  5. 2022Locating and Editing Factual Associations in GPTcited by 2,609
  6. 2022GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformerscited by 1,838
  7. 2019Root Mean Square Layer Normalizationcited by 1,425
  8. 2022SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Modelscited by 1,338
  9. 2023AWQ: Activation-aware Weight Quantization for LLM Compression and Accelerationcited by 1,234
  10. 2017Mixed Precision Trainingcited by 1,135
  11. 2020On Layer Normalization in the Transformer Architecturecited by 906
  12. 2022ZeroQuant: Efficient and Affordable Post-Training Quantization for Large-Scale Transformerscited by 517
3 more
  1. 2019Q8BERT: Quantized 8Bit BERTcited by 296
  2. 2019A Study of BFLOAT16 for Deep Learning Trainingcited by 227
  3. 2023Quantizable Transformers: Removing Outliers by Helping Attention Heads Do Nothingcited by 147

Built on by

Not cited yet.