What Does Layer-Importance Reveal About Transformers and State-Space Models?
On the map: State-space sequence models (S4, Mamba) → Core SSM architectures for language
Abstract (arXiv)
Transformers and state-space models (SSMs) are the two dominant families of sequence models, and a central open question is how far the analytical knowledge built for transformers transfers to SSMs. We address this through the lens of layer importance which underpins compression, selective fine-tuning, and interpretability across both families. We decompose layer importance into two distinct notions. Necessity captures how much the pretrained model depends on a layer's existing contribution, measured by the loss increase from bypassing it. Plasticity captures where the model absorbs new information during fine-tuning, measured by the magnitude of task-specific weight updates. Our analysis reveals that the two families behave fundamentally differently: in every evaluated residual transformer up to 14B parameters, Necessity and Plasticity anti-align across depth, whereas in the evaluated Mamba-style SSMs they point to overlapping regions. The sign of this alignment also predicts downstream adaptation behavior. In the evaluated transformers, concentrating updates in the most plastic layers increases catastrophic forgetting, while this tier-dependent effect disappears in the evaluated Mamba-style SSMs.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Core SSM architectures for language, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 20 of its 31 references that are in the map. First posted on arXiv 15 Sep 2026.
Builds on
20 of 31 references.
Closest ancestors · 5
Same direction, newest first
- 2025Mamba-Shedder: Post-Transformer Compression for Efficient Selective Structured State Space Models28 Jan
- 2024Falcon Mamba: The First Competitive Attention-free 7B Language Model7 Oct
- 2024MambaLRP: Explaining Selective State Space Sequence Models11 Jun
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Duality31 May
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spaces1 Dec
Other references · 15
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
- 2017Axiomatic Attribution for Deep Networkscited by 3,384
- 2022Locating and Editing Factual Associations in GPTcited by 2,691
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2018Sanity Checks for Saliency Mapscited by 1,156
- 2023RWKV: Reinventing RNNs for the Transformer Eracited by 827
- 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 383
- 2024The Unreasonable Ineffectiveness of the Deeper Layerscited by 226
- 2024AlphaLoRA: Assigning LoRA Experts Based on Layer Training Qualitycited by 20
- 2025Demystifying the Roles of LLM Layers in Retrieval, Knowledge, and Reasoningcited by 15
- 2025PLoP: Precise LoRA Placement for Efficient Finetuning of Large Modelscited by 14
3 more
- 2025Layer Importance for Mathematical Reasoning is Forged in Pre-Training and Invariant after Post-Trainingcited by 9
- 2026ShapLoRA: Allocation of Low-rank Adaption on Large Language Models via Shapley Value Inspired Importance Estimationcited by 3
- 2025TELL-TALE: Task Efficient LLMs with Task Aware Layer Eliminationcited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Global Divergence, Local Convergence: Representation Geometry in SSMs and Transformers
- 2025Memory Determines Learning Direction: A Theory of Gradient-Based Optimization in State Space Models
- 2026Expressivity-Efficiency Tradeoffs for Hybrid Sequence Models
- 2026First-Order Predictable but Pairwise Fragile: Local Task Adaptation in Trained Transformers
- 2026Geometric and Behavioral Stratification in Transformer Residual Streams
- 2025Contextual Representation Flow Across Depth in State-Space and Transformer Architectures
- 2026Priming: Hybrid State Space Models From Pre-trained Transformers
- 2026The Topological Trouble With Transformers