about
Slim attention: cut your context memory in half without loss of accuracy (arxiv.org)
7 points by DrewWas on Mar 12, 2025 | hide | past | pdf | 3 comments on HN

In plain words: Standard attention stores both keys and values for every past token; this keeps only the keys and rebuilds the values on the fly, so results stay exactly the same. It halves context memory and can speed up generation up to 2x.

Abstract · Slim attention: cut your context memory in half without loss -- K-cache is all you need for MHA

Slim attention shrinks the context memory size by 2x for transformer models with MHA (multi-head attention), which can speed up inference by up to 2x for large context windows. Slim attention is an exact, mathematically identical implementation of the standard attention mechanism and therefore doesn't compromise model accuracy. In other words, slim attention losslessly compresses the context memory by a factor of 2. For encoder-decoder transformers, the context memory size can be reduced even further: For the Whisper models for example, slim attention reduces the context memory by 8x, which can speed up token generation by 5x for batch size 64 for example. And for the T5-11B model for example, the memory can be reduced by 32x because its MHA projection dimension is larger than the embedding dimension. See https://github.com/OpenMachine-ai/transformer-tricks for code and more transformer tricks, and https://www.youtube.com/watch?v=uVtk3B6YO4Y for this paper's YouTube video.

Nils Graef, Andrew Wasielewski
arXiv:2503.05840 · cs.LG · submitted Mar 7, 2025 · updated Jun 3, 2025
abstract · pdf · html · 18 pages, 7 figures

add comment on HN

Slim attention shrinks the context memory size by 2x for transformer models with MHA (multi-head attention), which can speed up inference by up to 2x for large context windows. Slim attention is an exact, mathematically identical implementation of the standard attention mechanism and therefore does not compromise model accuracy. In other words, slim attention losslessly compresses the context memory by a factor of 2. For encoder-decoder transformers, the context memory size can be reduced even further: For the Whisper models for example, slim attention reduces the context memory by 8x, which can speed up token generation by 5x for batch size 64 for example. And for rare cases where the MHA projection dimension is larger than the embedding dimension, the memory can be reduced by a factor of 32 for the T5-11B model for example