mlsanityData to 7 Oct 2026

Rethinking Attention Output Projection: Structured Hadamard Transforms for Efficient Transformers

arXiv 2603.08343 ↗First posted 9 Mar 2026cs.LG cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

The dense output projection in multi head attention scales quadratically with model dimension, contributing significantly to parameter count, memory footprint, and inference cost. We propose replacing this projection with a fixed, parameter free Walsh Hadamard Transform (WHT) followed by a diagonal affine transformation. This approach eliminates approximately 25 percent of attention parameters per block while maintaining global cross-head interaction through an orthogonal, norm-preserving transformation. Our results demonstrate that WHT augmented models exhibit a steeper validation loss curve relative to training FLOPs compared to dense baselines, suggesting superior compute utilization during training. Crucially, we show that efficiency gains including reduced memory footprint and increased throughput grow monotonically with model size, batch size, and sequence length. We evaluate performance across both prefill and decoding stages, finding that the structured transform consistently outperforms dense projections as complexity increases. Our findings indicate that replacing dense projections with structured transforms allows for more compute-efficient architectures that achieve lower loss than dense models at an equivalent training budget.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 10 of its 14 references that are in the map. First posted on arXiv 9 Mar 2026.

Builds on

10 of 14 references.

References · 10

  1. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  2. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  3. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
  4. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
  5. 2020GLU Variants Improve Transformercited by 1,724
  6. 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,311
  7. 2019Fast Transformer Decoding: One Write-Head is All You Needcited by 700
  8. 2014Fastfood: Approximate Kernel Expansions in Loglinear Timecited by 72
  9. 2020Multi-Head Attention: Collaborate Instead of Concatenatecited by 56
  10. 2024MoH: Multi-Head Attention as Mixture-of-Head Attentioncited by 41

Built on by

Not cited yet.