首次解析大模型推理时控制强度的理论机制
Towards Understanding Steering Strength
- 从理论上分析控制强度对中间表示的影响规律
- 发现控制强度与生成概率存在非单调关系
- 适用于需要精细调控语言模型行为的研究者
大语言模型后训练阶段的控制常通过调整中间隐状态的方向实现。尽管方向选择已有诸多方法,但控制幅度的选择仍缺乏理论指导——过小则无效,过大则破坏模型性能。本文首次对控制强度进行系统理论分析,揭示其对下一个词概率、概念出现概率及交叉熵的影响规律,发现控制强度与这些指标间存在非单调关系。我们在11个语言模型上验证了理论预测,涵盖从小型GPT架构到现代大模型的多种规模。
原文摘要 · Abstract (English)
A popular approach to post-training control of large language models (LLMs) is the steering of intermediate latent representations. Namely, identify a well-chosen direction depending on the task at hand and perturbs representations along this direction at inference time. While many propositions exist to pick this direction, considerably less is understood about how to choose the magnitude of the move, whereas its importance is clear: too little and the intended behavior does not emerge, too much and the model's performance degrades beyond repair. In this work, we propose the first theoretical analysis of steering strength. We characterize its effect on next token probability, presence of a concept, and cross-entropy, deriving precise qualitative laws governing these quantities. Our analysis reveals surprising behaviors, including non-monotonic effects of steering strength. We validate our theoretical predictions empirically on eleven language models, ranging from a small GPT architecture to modern models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。