arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2505.17697cs.CLcs.LG

激活控制:高效激发语言模型长思维链能力

Activation Control for Efficiently Eliciting Long Chain-of-thought Ability of Language Models

University of California, San Diego(加州大学圣地亚哥分校)

Zekai Zhao, Qi Liu, Kun Zhou, Zihan Liu, Yifei Shao, Zhiting Hu, Biwei Huang

AI总结:

本文发现LLM最后几层的关键激活控制长思维链属性,通过放大激活并插入“wait”标记可免训练激发长CoT,并提出了免训练激活控制与参数高效微调方法,显著提升推理性能。

摘要:

尽管大语言模型(LLMs)展现出卓越的推理性能,但激发其长思维链(CoT)能力通常需要昂贵的强化学习或在高质量蒸馏数据上的监督微调。我们研究了这一能力背后的内部机制,并表明在最后几层中一小部分高影响力的激活在很大程度上主导了长格式推理属性,如输出长度和自我反思。通过简单地放大这些激活并插入“wait”标记,我们可以在无需任何训练的情况下激发长思维链能力,从而显著提高自我反思率和准确性。此外,我们发现激活动态遵循可预测的轨迹,在特殊标记后急剧上升,随后呈指数衰减。基于这些见解,我们提出了一种通用的免训练激活控制技术。它利用少量对比示例来识别关键激活,并在推理时采用简单的解析函数来调节其值以激发长思维链。大量实验证实了我们的方法在高效激发LLMs长思维链推理并提升其性能方面的有效性。此外,我们提出了一种参数高效的微调方法,仅训练一个最后一层激活放大模块和少量LoRA层,在推理基准上以显著更少的参数超越了完整的LoRA微调。我们的代码和数据已公开发布。

英文摘要:

Despite the remarkable reasoning performance, eliciting the long chain-of-thought (CoT) ability in large language models (LLMs) typically requires costly reinforcement learning or supervised fine-tuning on high-quality distilled data. We investigate the internal mechanisms behind this capability and show that a small set of high-impact activations in the last few layers largely governs long-form reasoning attributes, such as output length and self-reflection. By simply amplifying these activations and inserting "wait" tokens, we can invoke the long CoT ability without any training, resulting in significantly increased self-reflection rates and accuracy. Moreover, we find that the activation dynamics follow predictable trajectories, with a sharp rise after special tokens and a subsequent exponential decay. Building on these insights, we introduce a general training-free activation control technique. It leverages a few contrastive examples to identify key activations, and employs simple analytic functions to modulate their values at inference time to elicit long CoTs. Extensive experiments confirm the effectiveness of our method in efficiently eliciting long CoT reasoning in LLMs and improving their performance. Additionally, we propose a parameter-efficient fine-tuning method that trains only a last-layer activation amplification module and a few LoRA layers, outperforming full LoRA fine-tuning on reasoning benchmarks with significantly fewer parameters. Our code and data are publicly released.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑