mlsanityData to 7 Oct 2026

LayerTracer: A Joint Task-Particle and Vulnerable-Layer Analysis framework for Arbitrary Large Language Model Architectures

arXiv 2604.20556 ↗First posted 22 Apr 2026cs.CL cs.AIcited by 0

On the map: State-space sequence models (S4, Mamba) → Core SSM architectures for language

Abstract (arXiv)

Currently, Large Language Models (LLMs) feature a diversified architectural landscape, including traditional Transformer, GateDeltaNet, and Mamba. However, the evolutionary laws of hierarchical representations, task knowledge formation positions, and network robustness bottleneck mechanisms in various LLM architectures remain unclear, posing core challenges for hybrid architecture design and model optimization. This paper proposes LayerTracer, an architecture-agnostic end-to-end analysis framework compatible with any LLM architecture. By extracting hidden states layer-by-layer and mapping them to vocabulary probability distributions, it achieves joint analysis of task particle localization and layer vulnerability quantification. We define the task particle as the key layer where the target token probability first rises significantly, representing the model's task execution starting point, and the vulnerable layer is defined as the layer with the maximum Jensen-Shannon (JS) divergence between output distributions before and after mask perturbation, reflecting its sensitivity to disturbances. Experiments on models of different parameter scales show that task particles mainly appear in the deep layers of the model regardless of parameter size, while larger-parameter models exhibit stronger hierarchical robustness. LayerTracer provides a scientific basis for layer division, module ratio, and gating switching of hybrid architectures, effectively optimizing model performance. It accurately locates task-effective layers and stability bottlenecks, offering universal support for LLM structure design and interpretability research.

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 28 references that are in the map. First posted on arXiv 22 Apr 2026.

Builds on

23 of 28 references.

Closest ancestors · 3

Same direction, newest first

  1. 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Duality31 May
  2. 2024Jamba: A Hybrid Transformer-Mamba Language Model28 Mar
  3. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spaces1 Dec

Other references · 20

  1. 2017Attention Is All You Needcited by 59,932
  2. 2023GPT-4 Technical Reportcited by 18,132
  3. 2024The Llama 3 Herd of Modelscited by 14,192
  4. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  5. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
  6. 2025Qwen3 Technical Reportcited by 8,195
  7. 2020Scaling Laws for Neural Language Modelscited by 6,411
  8. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
  9. 2022Locating and Editing Factual Associations in GPTcited by 2,609
  10. 2023A Survey of Large Language Modelscited by 2,537
  11. 2019Similarity of Neural Network Representations Revisitedcited by 1,695
  12. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
8 more
  1. 2022Mass-Editing Memory in a Transformercited by 976
  2. 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 831
  3. 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 483
  4. 2023Finding Neurons in a Haystack: Case Studies with Sparse Probingcited by 328
  5. 2023Linearity of Relation Decoding in Transformer Language Modelscited by 190
  6. 2023Linear attention is (maybe) all you need (to understand transformer optimization)cited by 97
  7. 2017Distinguishing Antonyms and Synonyms in a Pattern-based Neural Networkcited by 19
  8. 2026Neural Chain-of-Thought Search: Searching the Optimal Reasoning Path to Enhance Large Language Modelscited by 7

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Breach in the Shield: Unveiling the Vulnerabilities of Large Language Models
  2. 2026Compressed Sensing for Capability Localization in Large Language Models
  3. 2025Context-Aware Hierarchical Learning: A Two-Step Paradigm towards Safer LLMs
  4. 2025Demystifying the Roles of LLM Layers in Retrieval, Knowledge, and Reasoning
  5. 2024Does Representation Matter? Exploring Intermediate Layers in Large Language Models
  6. 2025LayerIF: Estimating Layer Quality for Large Language Models using Influence Functions
  7. 2025Internal Chain-of-Thought: Empirical Evidence for Layer-wise Subtask Scheduling in LLMs
  8. 2026Rethinking Layer Relevance in Large Language Models Beyond Cosine Similarity