Diffract: Spectral View of LLM Domain Adaptation
本文研究了大语言模型在领域自适应中的持续预训练,揭示了谱变化主要是由奇异向量的偏移而非奇异值的改变所驱动,并利用这一见解开发了 Diffract——一个通过选择性地回退低重要性注意力头同时保留领域连通性的方式,从而实现高效自适应的工具包。
原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明
想象一下,人工智能的世界就像一座巨大且繁忙的图书馆,巨大的机器人正在那里学习阅读和写作。这些被称为“大语言模型”(LLMs)的机器人,首先通过阅读图书馆里几乎所有的内容——新闻、故事、科学和聊天记录——来学习人类语言是如何运作的。这就像一个孩子在成长过程中,通过倾听周围人的谈话来学习说话。但是,当你希望这个机器人成为数学天才、编程奇才或指令大师时,该怎么办呢?你不能从头开始制造一个新的机器人,那样既费时又昂贵。相反,科学家们使用了一种叫做“持续预训练”(Continual Pre-Training, CPT)的技术。把这想象成让那个通用型机器人参加一个专门的“夏令营”:只需几周时间,进行高强度的数学书籍或代码手册的专项阅读。研究人员一直在追问的一个大问题是:机器人的大脑在参加这个夏令营期间究竟发生了怎样的变化? 它是在重写整个记忆,还是仅仅微调了一些特定的笔记?理解这一点至关重要,因为它能帮助我们构建更聪明、更高效的 AI,使其在学习新技能时不会忘记旧有的技能。
在一项名为《Diffract: LLM 领域适应的光谱视角》(Diffract: Spectral View of LLM Domain Adaptation)的新研究中,研究团队决定使用一种特殊的 X 光——奇异值分解(SVD)来窥探机器人的大脑。想象一下,机器人的大脑是由数百万个微小的齿轮和杠杆(数学矩阵)组成的。当机器人学习新知识时,这些齿轮会发生位移。研究人员想要观察它们是如何位移的。他们发现了一个令人惊讶的现象:这些齿轮的“大小”(奇异值)几乎没有发生变化。这就像机器人保留了原有的那一套工具箱。相反,神奇之处在于工具所指向的“方向”(奇异向量)旋转并重新定向,以面向新的学科。
该团队发现,机器人的大脑并不是一个均匀的整体,而是由许多被称为“注意力头”(attention heads)的小型专家组成的。其中一些专家就像变色龙一样,会根据机器人是在学习数学还是编程而改变自己的颜色,而另一些专家则无论如何都保持不变。这被称为“头异质性”(head heterogeneity)。正因如此,研究人员发现,他们可以将多达 60% 的这些专家“倒带”回原始状态,而不会损害机器人的性能。事实上,通过仔细选择要倒带哪些专家,他们甚至让机器人在数学方面的表现提升了高达 4%。这就像是意识到你 60% 的学习笔记其实都是干扰项,移除它们反而能帮你考出好成绩。
此外,这项研究还揭示了一种他们称之为“领域连通性”(domain connectivity)的现象。如果你将一个接受过数学训练的机器人和一个接受过编程训练的机器人进行大脑融合,其结果是一个在两者方面都表现优秀的机器人,且不会丧失原有的技能。这种融合在机器人进行专门化训练之前接受过长期、彻底的教育(即拥有较大的“Token 预算”)时效果最好。研究人员还发布了一个名为“Diffract”的免费工具包,旨在帮助他人观察这些“大脑齿轮”,这表明模型的学习方式更多是关于重新定向现有的结构,而不是从头开始构建新结构。虽然这些发现针对的是他们测试的模型,但它们为我们思考 AI 如何学习提供了一种全新的方式:学习不仅仅是覆盖过去,更是寻找观察未来的正确角度。
您所在领域的论文太多了?
获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。