cs.LGMar 9, 2026

Rethinking Attention Output Projection: Structured Hadamard Transforms for Efficient Transformers

Authors: Shubham Aggarwal, Lokendra Kumar

Abstract

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.

Figures & tables

Explore similar work

CardsList
  1. Learned Queries and Keys Are All You Need: Replacing the Value Projection with Structured Transforms

    Sep 29, 2026Ene Meco, Emadeldeen Hamdan, A. Enis CetinTransformer AttentionEfficient ViTs

  2. Magnitude Profile Pruning: Calibration-Free Structured Attention Head Removal for Transformer Compression

    Aug 11, 2026Kasun Dewage, Marianna Pensky, Heranga K. Rathnasekara +1LLM CompressionAttention Head Pruning

  3. Revisiting the Shape Convention of Transformer Language Models

    Feb 6, 2026Feng-Ting Liao, Guan-Ting Yi, Tzu-Quan Lin +2Efficient Transformer InferenceTransformer FFNs