通过调整激活值分布,无需修改输入就能提升大模型推理能力
Meaningless Tokens, Meaningful Gains: How Activation Shifts Enhance LLM Reasoning
- 发现无意义令牌可重塑激活分布,增强模型表征能力
- 提出轻量级激活重分配模块,显著提升多类推理任务表现
- 适合关注模型内部机制与推理优化的研究者
受插入长串无意义令牌可稳定提升大语言模型推理性能这一反常现象的启发,本文分析了其内在机制,并据此提出更系统的方法实现类似增益。研究发现,性能提升源于模型MLP层中激活值的重新分布:接近零的激活减少,大数值激活增加。这种分布变化通过抑制弱信号、强化强信号,提升了模型的表征能力。基于此,我们提出激活重分配模块(ARM),一种仅在推理阶段运行的轻量级技术,直接调控激活值而不改变输入序列。ARM自适应识别非线性函数后的近零激活,并将其向外转移,以受控方式重现无意义令牌的有益效果。在多种基准和模型架构上的广泛实验表明,ARM能持续提升大模型在推理任务中的表现,且仅需几行简单代码即可实现。本研究不仅揭示了无意义令牌带来意外增益的机制,还提供了一种简单而有效的激活重分配方法,可进一步提升大模型性能。
原文摘要 · Abstract (English)
Motivated by the puzzling observation that inserting long sequences of meaningless tokens before the query prompt can consistently enhance LLM reasoning performance, this work analyzes the underlying mechanism driving this phenomenon and based on these insights proposes a more principled method that allows for similar performance gains. First, we find that the improvements arise from a redistribution of activations in the LLM's MLP layers, where near zero activations become less frequent while large magnitude activations increase. This redistribution enhances the model's representational capacity by suppressing weak signals and promoting stronger, more informative ones. Building on this insight, we propose the Activation Redistribution Module (ARM), a lightweight inference-time technique that modifies activations directly without altering the input sequence. ARM adaptively identifies near-zero activations after the non-linear function and shifts them outward, implicitly reproducing the beneficial effects of meaningless tokens in a controlled manner. Extensive experiments across diverse benchmarks and model architectures clearly show that ARM consistently improves LLM performance on reasoning tasks while requiring only a few lines of simple code to implement. Our findings deliver both a clear mechanistic explanation for the unexpected benefits of meaningless tokens and a simple yet effective technique that harnesses activation redistribution to further improve LLM performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。