小模型的表征可线性迁移,用来引导大模型行为
Linear Representation Transferability Hypothesis: Leveraging Small Models to Steer Large Models
- 用线性映射连接不同规模模型的隐藏状态空间
- 小模型的控制向量在大模型中仍保持语义效果
- 为跨尺度模型对齐提供新思路,适合模型压缩与对齐研究者
人们假设结构相似、在相似数据上训练的神经网络会学习到与任务相关的共享表征。本文在此基础上提出线性表征可迁移性(LRT)假说:在相同数据上训练的不同规模模型,其表征空间可通过仿射变换关联。我们通过学习小模型与大模型间隐藏状态的仿射映射,验证了将小模型的控制向量(指向特定行为的方向)迁移到大模型后,其语义效果仍能保留。实验显示该映射具有强泛化能力,表明小模型学习到的表征可用于引导大模型行为。该结果支持了跨模型尺度的表征对齐机制,为理解大规模模型间的表征一致性提供了新视角。
原文摘要 · Abstract (English)
It has been hypothesized that neural networks with similar architectures trained on similar data learn shared representations relevant to the learning task. We build on this idea by extending the conceptual framework where representations learned across models trained on the same data can be expressed as linear combinations of a \emph{universal} set of basis features. These basis features underlie the learning task itself and remain consistent across models, regardless of scale. From this framework, we propose the \textbf{Linear Representation Transferability (LRT)} Hypothesis -- that there exists an affine transformation between the representation spaces of different models. To test this hypothesis, we learn affine mappings between the hidden states of models of different sizes and evaluate whether steering vectors -- directions in hidden state space associated with specific model behaviors -- retain their semantic effect when transferred from small to large language models using the learned mappings. We find strong empirical evidence that such affine mappings can preserve steering behaviors. These findings suggest that representations learned by small models can be used to guide the behavior of large models, and that the LRT hypothesis may be a promising direction on understanding representation alignment across model scales.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。