Mechanism-Aligned Observability for Training Instability of Large Language Models
cs.CL
Submitted: 2026-06-26
Updated: 2026-10-18
License: http://creativecommons.org/licenses/by/4.0/
The gist: Frontier large language model training consumes massive accelerator fleets over long wall-clock times, making stability failures extremely costly.
Abstract
Frontier large language model training consumes massive accelerator fleets over long wall-clock times, making stability failures extremely costly. However, global symptoms such as loss divergences often emerge only after the underlying fault has already affected the training dynamics, further increasing the sunk compute. In which internal representation can a training fault become persistently and interpretably observable before it appears in global training metrics? We propose a mechanism-aligned principle for training observability: identify the internal state into which a fault is persistently written, derive the structure induced by its mechanism, and construct monitors in representations aligned with the computation of the affected module. For FlashAttention, we show that biased backward errors accumulate into a persistent low-dimensional structure in weight updates, producing spectral concentration. For MoE routers, we characterize routing dynamics through per-token routing behavior rather than parameter similarity, which may change without altering the routing function. Experiments show that the two monitors respond selectively to changes in their corresponding modules, before degradation becomes visible in global training metrics.
Sources
- Attention Residuals
- Conditional Memory via Scalable Lookup: A New Axis of Sparsity for Large Language Models
- A Scalable Measure of Loss Landscape Curvature for Analyzing the Training Dynamics of LLMs
- mHC: Manifold-Constrained Hyper-Connections
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering