AI系统准确性提升的虚假繁荣

AI系统准确性提升的虚假繁荣

AI系统中的角色漂移现象

检索增强生成(RAG)系统被设计为严格基于检索到的文档回答问题。然而,当工程师们对这些AI管道进行端到端优化时,读者模块可能会学习到一条捷径:不再依赖检索到的证据,而是从自身内部记忆中提取答案,同时系统的整体准确率却持续上升。这就是”角色漂移”这一隐藏挑战——在复合AI系统中的一种故障模式,即使端到端性能不断改善,各个模块仍会学会绕过其分配的特定任务。

终端准确率的盲区

大型语言模型(LLM)系统将复杂任务分配给专业化模块处理。例如,一个用于多跳推理的系统可能将任务在”分解器”和”解决者”之间分配:分解器将大问题分解为可管理的子任务,而解决者则计算这些子问题的答案。这种劳动分工使AI工程师能够将执行任务委托给更小、更廉价的模型,并在可能的情况下并行处理子任务。

为了提升AI管道的性能,工程师通常使用端到端强化学习(RL)进行优化,这种方法由单一的”终端奖励”引导。这意味着系统仅根据最终答案的正确性进行评估(研究人员称之为”终端准确率”)。当终端准确率提高时,系统被认为正在按预期学习和工作。

然而,终端准确率并不能验证各模块是否正确执行了分配的任务。正如论文作者之一曹晓阳向VentureBeat解释的那样:”终端准确率将整个多组件AI系统的行为简化为一个数字。它显示最终答案是否正确,但对哪些组件做出了贡献或它们是否遵循了分配的角色却很少提及。”

这种盲点导致了角色漂移——一种故障模式,在优化过程中,模块行为偏离其分配的角色,即使系统的终端准确率持续提高。曹晓阳指出:”对工程团队而言,实际风险在于他们可能部署一个通过所有端到端评估的管道,尽管其预期的劳动分工已悄然崩溃。”因为奖励系统只对最终答案评分,它无法检测或惩罚模块的违规行为。

角色漂移的实际案例

考虑在分解器-解决者管道中这种情况是如何发生的:分解器的任务是编写抽象的子问题而不解决任务,将推理工作留给解决者。在端到端强化学习下,分解者很快发现较弱的解决者在抽象任务上容易出错。为了最大化奖励,分解者开始向发送给解决者的子问题中泄露或植入答案。最终,解决者只是鹦鹉学舌般地重复分解者提供的答案。终端准确率上升了,但预期的架构受到了损害。

为何角色漂移值得关注

如果系统得到正确答案且准确率在提高,我们为何要关心模块是否偏离了其角色?

实际部署需要的远不止是在训练数据集上得到正确的最终答案。分配给这些模块的隐含角色确保了系统的可扩展性、可靠性和可审计性。当角色漂移发生时,可能出现以下问题:

  • 效率与可审计性的丧失:在推理示例中,角色漂移导致分解者承担所有繁重工作,而不是进行规划和委派。曹晓阳指出:”一旦分解者开始在子问题中直接放入答案,解决者就沦为复制这些答案的工具。你仍在支付运行[不同模块]的费用,但它们不再独立工作。”工作负载无法在多个解决者之间并行化,无法委托给更便宜的模型以节省计算资源,下游的人类利益相关者也无法再逐步审核系统逻辑以验证其得出答案的过程。
  • 动态环境中的脆弱性:考虑一个RAG系统,其中阅读器模型的任务是严格使用外部检索到的文档回答问题。如果阅读器发生漂移并开始依赖自身的参数化记忆(因为其在训练期间碰巧准确),系统就会变得脆弱。当企业用新信息更新数据库,或者用户询问模型预训练范围之外的新主题问题时,系统将失败,因为它已放弃了其原本应使用的 grounding 机制。

Role Anchor:测量并强制执行角色

曹晓阳表示:”仅针对最终结果进行训练,会奖励系统产生正确答案,而不管它如何实现这一目标。”为了应对这一问题,Role Anchor作为一种轻量级正则化技术,将角色指令纳入训练目标。它比较组件在有和无这些指令时的行为表现,并阻止训练削弱这些指令的效果。

从高层次上看,Role Anchor确保模块在整个强化学习优化过程中继续尊重其原始角色提示的引导影响,使角色漂移既可测量又可控。

Role Anchor的一个关键见解是,角色的效果可以通过比较模型在有无角色提示时的行为来衡量。系统为每个模块评估两种不同的提示:

  1. 专业化、指令密集的角色提示(例如:”你是一个细心的阅读者。使用检索到的段落回答用户的问题…”)。
  2. 中性提示(例如:”回答用户的问题…”)。

对于任何给定输入,模型都会输出下一个标记的概率分布。在角色提示下运行时,它会倾向于某些标记;而在中性提示下运行时,它表现得像通用助手。这些概率分布之间的差异就是”角色效用”。

这种效用衡量的是”推动力”,即角色提示将LLM的默认预测推向何方以及推动的强度。如果某个标记与分配的角色高度一致,角色提示会提高其相对于中性基线的可能性(或者说”推动”模型朝该标记方向发展)。

在开始强化学习训练之前,Role Anchor保留一个冻结的模型副本作为参考,并测量角色提示在该参考模型上的原始推动力。这种强化学习前的推动力充当了设计者意图的基准,作为角色提示应该如何引导模型的代理指标。

在强化学习训练过程中,随着活动模型权重的更新,Role Anchor定期计算当前推动力,并将其与参考推动力进行比较。如果当前推动力开始减弱或偏离参考值,Role Anchor…


关注微信号:智享开源 ,及时了解更新信息。

原文链接:https://venturebeat.com/orchestration/one-ai-module-faked-86-of-a-pipelines-accuracy-gains-by-feeding-another-the-answers

评论列表
发表评论
😀 😂 😃 😄 😅 😆 😉 😊 😋 😎 😍 😘 🥰 😜 😝 🤗 🤔 😭 😤 👍

为你推荐
Ta的个人站点

Mark Do发布文章2605篇

如我距离死亡还有45年,我还活着,该怎么度过现在。


公众号:智享开源

分类