mlsanityData to 14 Oct 2026

Attention Projection Mixing with Exogenous Anchors

arXiv 2601.08131 ↗First posted 13 Jan 2026cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Cross-layer reuse of early attention projections can improve optimization and data efficiency, but it creates a structural conflict: the first layer must simultaneously act as a stable, reusable anchor for all deeper layers and as an effective computational block. We demonstrate that this tension constrains the performance of internal-anchor designs. We propose ExoFormer, which resolves the conflict by learning exogenous anchor projections outside the sequential layer stack. We introduce a unified normalized mixing framework that mixes queries, keys, values, and gate logits using learnable coefficients (exploring coefficient granularities: elementwise, headwise, and scalar), and we show that normalizing anchor sources is key to stable reuse. ExoFormer variants consistently outperform their internal-anchor counterparts, and the dynamic variant yields 1.5x downstream accuracy points while matching validation loss using 1.5x fewer tokens than Gated Attention. We explain this efficacy via an Offloading Hypothesis: external anchors preserve essential token identity, allowing layers to specialize exclusively in feature transformation. We release code and models to facilitate future research.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 27 of its 31 references that are in the map. First posted on arXiv 13 Jan 2026.

Builds on

27 of 31 references.

References · 27

  1. 2017Attention Is All You Needcited by 60,361
  2. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
  3. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  4. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
  5. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  6. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
  7. 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
  8. 2020GLU Variants Improve Transformercited by 1,804
  9. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
  10. 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 856
  11. 2025Muon is Scalable for LLM Trainingcited by 428
  12. 2021DeepViT: Towards Deeper Vision Transformercited by 289
15 more
  1. 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 288
  2. 2025Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Freecited by 283
  3. 2020Query-Key Normalization for Transformerscited by 280
  4. 2024Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durationscited by 170
  5. 2024You Only Cache Once: Decoder-Decoder Architectures for Language Modelscited by 144
  6. 2025The Polar Express: Optimal Matrix Sign Methods and Their Application to the Muon Algorithmcited by 110
  7. 2024Hyper-Connectionscited by 84
  8. 2022Revisiting Over-smoothing in BERT from the Perspective of Graphcited by 72
  9. 2025Attention Sinks and Compression Valleys in LLMs are Two Sides of the Same Coincited by 52
  10. 2025MUDDFormer: Breaking Residual Bottlenecks in Transformers via Multiway Dynamic Dense Connectionscited by 31
  11. 2024Value Residual Learningcited by 27
  12. 2025Cautious Weight Decaycited by 13
  13. 2016The Z-loss: a shift and scale invariant classification loss belonging to the Spherical Familycited by 12
  14. 2025CR-Net: Scaling Parameter-Efficient Training with Cross-Layer Low-Rank Structurecited by 7
  15. 2025Improving Model Representation and Reducing KV Cache via Skip Connections with First Value Headscited by 4

Built on by

Not cited yet.