跳到主要内容
AI News HubLIVE
站内改写2 分钟阅读

超越在线与离线之争:LLM智能体蒸馏应在哪一层面跟随学生模型?

文章摘要

该论文提出一种分层智能体蒸馏框架,将“交互历史由谁产生”和“动作令牌前缀由谁生成”视为两个独立层面的在线程度,并系统研究了它们的配对、混合比例与训练时序。实验表明,让学生模型产生历史、教师模型生成动作前缀通常是最强的单独训练信号;两层都完全跟随学生效果最差;但完全在线数据在混合配置中仍有价值,且两层的配对方式比在线数据总量更重要,先使用教师上下文、后使用完全在线上下文的效果优于静态混合。

来源arXiv AI作者: Ya Gao, Zhengtao Zou, Pekka Marttinen
超越在线与离线之争:LLM智能体蒸馏应在哪一层面跟随学生模型?
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

LLM 智能体(agent)的蒸馏长期被简化为一个二选一的问题:用教师模型生成的示范,还是用学生模型自己产生的交互经验。这两条路线大致对应离线策略(off-policy)与在策略(on-policy)蒸馏,二者都已被证明有效。但 arXiv 论文《Beyond On-Policy vs. Off-Policy: At What Level Should LLM Agent Distillation Follow the Student?》指出,这种二分法本身忽略了一个层次结构。作者主张,“在策略程度”至少应拆成两个层面:一层决定交互历史由哪一方产生,另一层决定当前动作的 token 前缀由哪一方生成。这两层并不必然使用同一个策略,因此可以独立选择、自由搭配。

围绕这一观察,论文提出了一个分层的智能体蒸馏框架,把上述两个层面的在策略性显式形式化,并允许研究者控制它们的配对方式、混合比例以及训练日程。为了在可控条件下研究该问题,作者采用自蒸馏(self-distillation)设定:教师模型是冻结的,并且拥有学生模型无法获得的任务指导信息,从而模拟“教师更强、学生需学习”的典型场景。实验设计为三阶段对照研究,覆盖三个模型和四个交互式环境,规模上足以检验不同配置的稳定性。

研究给出的第一条结论较为反直觉:在单独训练时,最强的信号通常来自“学生诱导的交互历史”搭配“教师生成的动作前缀”。也就是说,让历史由学生自己探索产生,但在生成具体动作时以教师的高质量前缀为引导。相对地,如果两个层面都完全跟随学生,即完全在策略,效果普遍最差。这提示在策略数据并非越多越好,历史与动作层面的来源需要区别对待。

不过,完全在策略的数据并非没有价值。作者发现,当把它作为混合配置的一部分而非单独使用引入时,它能带来收益。更值得注意的是,即便两个层面上的教师/学生比例完全相同,不同的配对方式也会导致显著不同的性能。这说明“有多少在策略数据”并不是唯一关键,“在策略数据出现在哪个层面、如何与教师数据配对”同样重要。

最后,训练日程也影响结果。论文发现,把完全由教师生成的上下文放在训练前期、将完全在策略的上下文推迟到后期,比静态混合更能提升表现。综合这些发现,作者为智能体蒸馏提供了可操作的实践原则:应回答在哪里跟随学生、跟随多少、以及在什么时候跟随,而不是仅在整体上选择在策略或离线策略。

展开要点与分析

文章情报

工程师进阶

要点

  • 论文将智能体蒸馏中的在线程度拆分为两个层级:交互历史由谁诱导,以及当前动作的令牌前缀由谁生成。
  • 在三个模型和四个交互环境中,学生诱导历史搭配教师生成动作前缀通常提供最强的单独训练信号,而完全跟随学生在两层都表现最差。
  • 两层教师/学生比例相同的混合数据可能产生显著不同的性能,说明配对方式比在线数据总量更重要。
  • 先使用完全由教师生成的上下文、再延迟引入完全在线上下文,优于静态混合训练。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。