Top-Theta Attention: Sparsifying Transformers by Compensated Thresholding

21d ago · Global · primary source: export.arxiv.org

A new method called Top-Theta Attention can dramatically reduce the memory footprint of transformer models during inference without any retraining, according to a paper posted on arXiv. The technique uses static, per-head thresholds to selectively retain only the most significant attention elements. The approach, detailed in a preprint by Konstantin Berestizshevsky, introduces a training-free sparsification strategy for the attention mechanism central to large language models [1]. Unlike existing top-k attention methods, Top-Theta Attention calibrates a fixed threshold for each attention head. This allows the model to dynamically retain a constant number of important elements per row based on content, rather than a pre-set number, making it robust across different types of data [2]. The paper reports that this method achieves a 3-10x reduction in V-cache usage and requires up to 10x fewer attention elements during inference, while degrading accuracy by no more than 1% [1, 2]. The research addresses a core computational bottleneck in transformer models, which are a type of machine learning model with many parameters trained on vast amounts of text for language generation tasks [8]. The preprint was submitted to arXiv, an open-access repository for scientific e-prints that is not peer-reviewed but serves as a primary dissemination platform in fields like computer science and physics [6]. The repository, which began in 1991, now receives about 24,000 new articles per month [6]. The paper has been revised twice since its initial submission in February 2025, with the latest version posted in June 2026 [1]. The authors also introduce compensation techniques to maintain accuracy even under aggressive sparsification, positioning thresholding as a practical alternative to existing methods [2]. The work is presented as a principled approach to making large transformer models more efficient for real-world deployment, where computational and memory constraints are critical.

infrastructureresearch-paper

Background sources we checked (7)
  • arxiv.org ↗ We present Top-Theta (Top-$θ$) Attention, a training-free method for sparsifying transformer attention during inference. Our key insight is that static, per-head thresholds can be calibrated to retain the desired constant number of significant elements per attention row. This app…
  • info.arxiv.org ↗ arXiv Labs - arXiv info | arXiv e-print repository Skip to content # arXiv Labs Attention arXiv Users: arXiv Labs is pausing new proposals ## What are arXiv Labs? arXiv Labs are a way for the community to contribute new, useful features to arXiv. These integrations are avail…
  • blog.arxiv.org ↗ arXivLabs: a space for community innovation – arXiv blog arXiv has launched a new, formalized framework enabling innovative collaborations with individuals and organizations. “Members of our community want to contribute tools that enhance the arXiv experience, and we val…
  • info.arxiv.org ↗ arXivLabs: Showcase - arXiv info | arXiv e-print repository ... # arXivLabs: Showcase ... arXiv is surrounded by a community of researchers and developers working at the cutting edge of information science and technology. ... While the arXiv team is focused on our core mission—pr…
  • en.wikipedia.org ↗ arXiv (pronounced as "archive"—the X represents the Greek letter chi ⟨χ⟩) is an open-access repository of electronic preprints and postprints (known as e-prints) approved for posting after moderation, but not peer reviewed. It consists of scientific papers in the fields of mathem…
  • en.wikipedia.org ↗ 14 (fourteen) is the natural number following 13 and preceding 15.…
  • en.wikipedia.org ↗ A large language model (LLM) is a type of machine learning model designed for natural language processing tasks such as language generation. LLMs are language models with many parameters, and are trained with self-supervised learning on a vast amount of text.…

Sources

Spot something wrong? Report an issue