Apple 与约翰霍普金斯提出 SELFCOMPACT:让语言模型智能体自行压缩上下文
Self-Compacting Language Model Agents
Apple 与约翰霍普金斯大学研究者提出 SELFCOMPACT,让模型自己决定何时以及如何压缩上下文,由模型调用的压缩工具和一份轻量 rubric 两部分组成,无需微调或外部监督。
论文提出让模型自行决定何时压缩上下文的脚手架,并给出六个基准、七个模型上的成本与效果对比。
作者Tianjian Li†, Jingyu Zhang†, William Jurayj†, Xi Wang†, Chuanyang Jin†, Mehrdad Farajtabar, Eric Nalisnick†, Daniel Khashabi†
由思维链和工具调用组成的长智能体轨迹会积累过时内容,这些内容会锚定后续生成,并最终超出上下文窗口。现有脚手架通过在 token 阈值处触发固定间隔压缩来缓解这一问题。此类触发机制完全不考虑轨迹结构,存在在推导或搜索中途丢弃部分结果的风险。我们提出 SELFCOMPACT,一种允许模型自身决定何时以及如何压缩的脚手架。具体而言,它将两个推理时元素配对:(i) 模型调用以总结累积上下文的压缩工具,以及 (ii) 一个轻量级评分标准,规定何时触发(子任务已解决,或轨迹正在收敛)以及何时抑制(推导中途,或陷入困境时)。两者缺一不可。仅靠工具在开放权重模型中的使用并不均衡,常常在无益的时刻被调用,或者根本不被调用;仅靠评分标准则无法行动。二者结合,无需任何微调或外部监督即可引出有效的自适应压缩。我们展示了在六个基准(竞赛数学和智能体搜索)和七个模型上的实证结果。我们的结果表明,SELFCOMPACT 以一小部分 token 成本达到或超过固定间隔摘要,在数学上比无摘要基线提升高达 18.1 分,在智能体搜索上提升 5–9 分,同时每题成本降低 30–70%。我们的结果揭示了一个元认知差距:尽管未经提示的模型无法可靠判断自身上下文何时腐化,但一个轻量级评分标准填补了这一差距,将何时压缩重新定义为脚手架无需训练即可提供的能力。
- † 约翰斯·霍普金斯大学
相关阅读与更新。
在多样化的交互环境中联合训练单个 LLM 智能体作为通往通用智能体的途径,已引起越来越多的关注。现有的课程和数据选择策略通常在环境层面分配训练,或优先考虑基于局部奖励的信号,而没有明确考虑跨环境的当前 rollout 之间的关系以进行提示组选择。与此同时,随着环境被学习……
为开放域问答设计有效的奖励信号具有挑战性,因为高质量的回答必须同时满足答案质量的多个方面,而这些方面难以用整体标量目标来捕捉。我们引入一个基于评分标准的奖励框架,该框架生成以检索证据为基础、分解为多个质量维度的查询特定评分标准,在……期间提供细粒度监督。
来源:Apple Machine Learning Research · machinelearning.apple.com