Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

WINA: Weight Informed Neuron Activation for Accelerating Large Language Model Inference

About

The growing computational demands of large language models (LLMs) make efficient inference and activation strategies increasingly critical. While recent approaches, such as Mixture-of-Experts (MoE), leverage selective activation but require specialized training, training-free sparse activation methods offer broader applicability and superior resource efficiency through their plug-and-play design. However, many existing methods rely solely on hidden state magnitudes to determine activation, resulting in high approximation errors and suboptimal inference accuracy. To address these limitations, we propose WINA (Weight Informed Neuron Activation), a novel, simple, and training-free sparse activation framework that jointly considers hidden state magnitudes and the column-wise $\ell_2$-norms of weight matrices. We show that this leads to a sparsification strategy that obtains optimal approximation error bounds with theoretical guarantees tighter than existing techniques. Empirically, WINA also outperforms state-of-the-art methods (e.g., TEAL) by up to $2.94\%$ in average performance at the same sparsity levels, across a diverse set of LLM architectures and datasets. These results position WINA as a new performance frontier for training-free sparse activation in LLM inference, advancing training-free sparse activation methods and setting a robust baseline for efficient inference. The source code is available at https://github.com/microsoft/wina.

Sihan Chen, Dan Zhao, Jongwoo Ko, Colby Banbury, Huiping Zhuang, Luming Liang, Pashmina Cameron, Tianyi Chen• 2025

Related benchmarks

TaskDatasetResultRank
Long-context Language UnderstandingLongBench--
219
General ReasoningMMLU
MMLU Accuracy76.6
126
Long-context UnderstandingLongBench
Overall Average Score31.39
115
CodeHumanEval
HumanEval Accuracy53
50
Commonsense ReasoningCommonsense Reasoning
Accuracy74.25
44
Commonsense ReasoningCommonsense Reasoning Suite (PiQA, Arc-C, WinoGrande, HellaSwag, SciQ, OBQA, BoolQ, Arc-E)
PiQA Accuracy82.1
15
Commonsense ReasoningCommonsense Reasoning Suite (PiQA, Arc-C, WinoGrande, HellaSwag, SciQ, OBQA, BoolQ, Arc-E) (test)
PiQA Accuracy80.41
15
Long-context ReasoningLongBench Llama-2-7B-4K
Code Completion62.54
9
Showing 8 of 8 rows

Other info

Follow for update