Decoding Future Semantics from Language Model Hidden States: Rollout Trade-offs and Causal Steering
本研究表明,线性探针可以从语言模型的隐藏状态中提取未来的语义信息,揭示了一种权衡关系,即这种直接读取的有效性相对于自回归展开的有效性,取决于嵌入空间、生成时界以及提示词固有的可预测性。
原始论文采用 CC BY 4.0 许可(https://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明
现代计算机程序通过预测下一个词来编写文本,这些程序被称为语言模型。它们观察你提供给它们的词,并猜测下一个词,然后利用这个新词来猜测再下一个词,从而逐步构建出一个句子。长期以来,科学家们一直认为这些程序的“未来视野”仅限于其下一次预测所能触及的范围。然而,一项新的研究表明,在这些模型复杂的数学机制内部,存在着一层隐藏的信息层,早在计算机实际打出任何一个新词之前,该层就已经包含了即将要表达的大意。这项研究探讨了这些内部计算机状态是否持有对未来文本的“预览”,这种预览与仅仅阅读提示词相比是否足够清晰,以及了解这种预览是否能让我们改变计算机的写作内容。
研究人员,来自韩国外国语大学的崔载旭(Jaekeol Choi),试图通过使用几种不同版本的流行语言模型来测试这一想法。他们将重点放在计算机开始生成新句子的那一刻。就在这个精确的瞬间,模型已经处理了用户的输入,并持有一个特定的内部表示——一种总结了它目前为止所理解的一切的复杂数字模式。研究团队提出了一个简单的问题:如果你观察这个内部模式,你能否判断出即将生成的句子的主题或情感倾向?为了找出答案,他们训练了一个简单的工具来读取这些内部模式,并将其转化为对未来文本的摘要。他们将这种方法与其他预测未来的方式进行了对比,例如让计算机先实际写出几个词,或者仅根据原始提示词来预测未来而不观察计算机的大脑内部。
研究发现,答案是肯定的。模型的内部模式确实包含了一张关于未来文本含义的线性可访问地图。当研究人员观察模型在生成句子之前的内部状态时,他们能够成功预测该句子的总体语义内容。这一现象在不同规模和架构的模型中均成立,从拥有数亿参数的小型模型到拥有数十亿参数的大型模型。这种读取未来含义的能力在计算机处理网络的深层网络中最为强大。然而,这种预览的清晰度很大程度上取决于研究人员如何定义“含义”。当他们使用一种特定的方式来衡量含义时,内部预览明显优于仅基于提示词进行的猜测。但当他们使用其他衡量含义的方式时,观察计算机大脑内部所获得的优势就变得微乎其微,有时甚至与仅通过观察起始词所能做出的猜测几乎没有区别。
研究人员还将这种“内部预览”方法与更直观的策略进行了比较,即让计算机先实际写出几个词。如果你让计算机生成三个或五个词,你自然会比仅仅尝试从隐藏状态中进行猜测获得更清晰的句子走向。这并不令人意外,因为计算机已经完成了编写这些词的工作。然而,研究揭示了一个有趣的权衡:随着目标句子变得越来越长,以及未来变得越来越遥远,让计算机多写几个词所带来的优势就会逐渐减小。如果目标是理解一个 256 个单词之后的句子的含义,那么内部预览在竞争中会表现得异常出色,足以与让计算机生成一段短前缀的方法相媲美。内部状态提供了一种在无需承担生成所有中间词的计算成本的情况下,获取遥远未来感知的方法,尽管它从未完全超越生成长序列的准确性。
这项研究中最引人入胜的部分涉及测试这种知识是否可以被用来改变计算机的行为。研究人员提取了内部状态中指向特定情感(如积极或消极)的方向,并尝试在计算机开始写作之前向该方向进行引导。他们发现,这种引导确实奏效了。通过根据预测结果调整内部状态,他们可以比进行随机调整更有效地改变生成文本的基调。然而,这种成功在计算机大脑的所有部分并非统一的。在某些模型中,能够最容易读取未来含义的层,也是进行更改的最佳位置。而在其他模型中,这两者完全无法匹配;显示出最清晰未来预览的层,有时反而是干预以改变结果的错误位置。事实上,在某些情况下,试图在具有最强预览能力的层进行修改,实际上会将计算机推向与预期相反的方向。
这些发现表明,虽然语言模型确实在其内部状态中携带了关于自身未来输出的信息,但这些信息并不是一段等待被读取的预写剧本。相反,它是一个复杂的、分层的信号,在网络的某些部分比其他部分更容易被访问。研究明确了,仅仅因为计算机能够“看到”其自身词汇的未来含义,并不意味着它已经完全预先计算出了整个响应。它还强调,能够读取信号与能够控制信号是两回事。预测未来的能力与引导未来的能力是相关但截然不同的能力,它们在模型内部并不总是在同一个地方发生。这项工作为这些智能系统如何组织关于其即将表达内容的的信息提供了一张更清晰的地图,表明它们的内部状态比此前理解的包含更丰富的未来导向数据,但也说明了利用这些数据进行控制比简单的读取过程要复杂得多。
您所在领域的论文太多了?
获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。