mlsanityData to 14 Oct 2026

Emergent Structure in the Marginal Attention Space of Language Models

arXiv 2610.03109 ↗First posted 2 Oct 2026cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

While representation similarity across independently trained language models is well-documented, how internal mechanics such as attention behave across models remains far less characterized. Inspired by this gap, we examine the structure of post-softmax attention weights by marginalizing over query positions, mapping them into a joint token-head "marginal attention space". Evaluating across 60+ diverse LLMs, we find that different properties emerge when reducing this space along its token and head axes. When reduced token-wise, marginal attention yields a text-intrinsic signal robustly conserved across models. To explain this property, we empirically connect marginal attention to the input-output Jacobian of the network, and prove theoretically that under a smoothness assumption, models with similar next-token distributions are guaranteed to have similar input-output Jacobian statistics. When reduced head-wise, it forms a model-private signature conserved across documents. Practically, this provides a natural way to estimate a per-head budget for key-value (KV) cache eviction, effectively decoupling model-specific budget allocation from text-intrinsic token scoring. On standard eviction benchmarks, a per-head budget precomputed offline on pretraining text, combined with a training-free token score, shows competitive performance with methods that recompute the budget on every document or train it per target. Code available at https://github.com/Flegyas/marginal-attention

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 62 of its 72 references that are in the map. First posted on arXiv 2 Oct 2026.

Builds on

62 of 72 references.

References · 62

  1. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
  2. 2023Improved Baselines with Visual Instruction Tuningcited by 4,418
  3. 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,411
  4. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
  5. 2022BLOOM: A 176B-Parameter Open-Access Multilingual Language Modelcited by 1,984
  6. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
  7. 2019Similarity of Neural Network Representations Revisitedcited by 1,755
  8. 2023LongBench: A Bilingual, Multitask Benchmark for Long Context Understandingcited by 1,417
  9. 2024RULER: What's the Real Context Size of Your Long-Context Language Models?cited by 1,346
  10. 2024DeepSeek-Coder: When the Large Language Model Meets Programming -- The Rise of Code Intelligencecited by 1,181
  11. 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,135
  12. 2024ChatGLM: A Family of Large Language Models from GLM-130B to GLM-4 All Toolscited by 1,121
38 more
  1. 2024SnapKV: LLM Knows What You are Looking for Before Generationcited by 880
  2. 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 856
  3. 2023H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Modelscited by 832
  4. 2023RWKV: Reinventing RNNs for the Transformer Eracited by 827
  5. 2019Are Sixteen Heads Really Better than One?cited by 816
  6. 2020Quantifying Attention Flow in Transformerscited by 775
  7. 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 723
  8. 2024TinyLlama: An Open-Source Small Language Modelcited by 618
  9. 20242 OLMo 2 Furiouscited by 340
  10. 2018Insights on representational similarity in neural networks with canonical correlationcited by 324
  11. 2024OLMoE: Open Mixture-of-Experts Language Modelscited by 301
  12. 2024DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Headscited by 288
  13. 2024InfiniGen: Efficient Generative Inference of Large Language Models with Dynamic KV Cache Managementcited by 245
  14. 2014Understanding image representations by measuring their equivariance and equivalencecited by 243
  15. 2024Ada-KV: Optimizing KV Cache Eviction by Adaptive Budget Allocation for Efficient LLM Inferencecited by 235
  16. 2024Retrieval Head Mechanistically Explains Long-Context Factualitycited by 232
  17. 2021Revisiting Model Stitching to Compare Neural Representationscited by 209
  18. 2023Interpreting CLIP's Image Representation via Text-Based Decompositioncited by 196
  19. 2022Relative representations enable zero-shot latent space communicationcited by 191
  20. 2024Compact Language Models via Pruning and Knowledge Distillationcited by 148
  21. 2024Aria: An Open Multimodal Native Mixture-of-Experts Modelcited by 138
  22. 2024Aya Expanse: Combining Research Breakthroughs for a New Multilingual Frontiercited by 129
  23. 2024Keyformer: KV Cache Reduction through Key Tokens Selection for Efficient Generative Inferencecited by 126
  24. 2019On Identifiability in Transformerscited by 116
  25. 2025NVIDIA Nemotron Nano 2: An Accurate and Efficient Hybrid Mamba-Transformer Reasoning Modelcited by 108
  26. 2024RazorAttention: Efficient KV Cache Compression Through Retrieval Headscited by 100
  27. 2025KVzip: Query-Agnostic KV Cache Compression with Context Reconstructioncited by 95
  28. 2025Sparse Autoencoders Trained on the Same Data Learn Different Featurescited by 93
  29. 2025Expected Attention: KV Cache Compression by Estimating Attention from Future Queries Distributioncited by 74
  30. 2023Latent Space Translation via Semantic Alignmentcited by 52
  31. 2022ASIF: Coupled Data Turns Unimodal Models to Multimodal Without Trainingcited by 47
  32. 2026Revisiting the Platonic Representation Hypothesis: An Aristotelian Viewcited by 44
  33. 2025Compactor: Calibrated Query-Agnostic KV Cache Compression with Approximate Leverage Scorescited by 17
  34. 2025Head Pursuit: Probing Attention Specialization in Multimodal Transformerscited by 17
  35. 2026Latent Space Communication via K-V Cache Alignmentcited by 17
  36. 2026Fast KVzip: Efficient and Accurate LLM Inference with Gated KV Evictioncited by 16
  37. 2026KVzap: Fast, Adaptive, and Faithful KV Cache Pruningcited by 15
  38. 2026Learning to Evict from Key-Value Cachecited by 14

50 most cited of 62 All 62 (JSON)

Built on by

Not cited yet.