WINA通过权重与激活值联合判断神经元,提升大模型推理效率和准确率。
WINA: Weight Informed Neuron Activation for Accelerating Large Language Model Inference
- 结合隐藏状态大小与权重列范数,实现无需训练的稀疏激活
- 在相同稀疏度下,平均性能比TEAL高2.94%以上
- 适用于多种模型架构,适合追求高效推理的研究者
大语言模型(LLM)日益增长的计算需求使得高效推理与激活策略变得愈发关键。尽管混合专家(MoE)等方法通过选择性激活提升了效率,但需专门训练;而无训练的稀疏激活方法因其即插即用设计,具备更广适用性和资源效率。然而,现有方法仅依赖隐藏状态幅值决定激活,导致近似误差高、推理精度不佳。为此,我们提出WINA(Weight Informed Neuron Activation),一种新颖、简单且无需训练的稀疏激活框架,联合考虑隐藏状态幅值与权重矩阵列方向ℓ₂-范数。理论证明该策略可获得更紧的近似误差上界。实验表明,WINA在相同稀疏度下,跨多种LLM架构与数据集,平均性能优于现有最优方法(如TEAL)达2.94%。这标志着无训练稀疏激活的新性能边界,为高效推理提供了稳健基线。代码已开源:https://github.com/microsoft/wina。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。