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