代码 Agent 的上下文减肥药:SWE-Pruner 是怎么让模型少看多想的

针对代码 Agent 上下文昂贵且易迷失的痛点,SWE-Pruner 提出模仿人类“选择性略读”的策略。该方案通过任务目标引导剪枝,利用 0.6B 小模型动态筛选关键代码片段,而非依赖破坏语法的固定指标。实测显示,其在 SWE-Bench 等任务中 Token 消耗降低 23% 至 54%,成功率反而提升。这一思路为长代码处理提供了新方向,尤其适合需要本地部署或边缘计算的 Agent 落地场景。

发布于2026年5月5日 09:10
编辑零重力瓦力
评论0
阅读2

如今的代码 Agent 动不动就啃几万 Token 的上下文,既贵又慢还容易迷失。SWE-Pruner 认为,不如学学人类程序员“选择性略读”的方式。

怎么做的

论文来自arXiv(2601.16746v2),团队基于一个观察:人类调试代码时不会逐行阅读,而是带着任务目标去“扫”相关部分。比如修 bug 时只看错误处理和调用栈,找性能瓶颈时只看循环和数据库操作。

SWE-Pruner干了三件事

  1. Agent 给自己定一个明确目标,比如“聚焦错误处理”或“关注并发逻辑”,这个目标会变成剪枝的提示。

  2. 训练了一个 0.6 B 的小模型(叫skimmer)来学习如何根据目标选择上下文。

  3. 在四个基准上测试,发现效果不错。在 SWE-Bench Verified 这类 Agent 任务上,Token 减少了 23% 到 54%,成功率反而还提升了。在单轮的长代码问答任务上,最高能压缩 14.84 倍。

为什么值得关注

之前业界做上下文压缩,主要靠困惑度(PPL)这类固定指标,问题是代码有结构,你不能把一个函数切成两半,或者漏掉一个关键的变量声明。PPL导向的压缩会破坏语法和逻辑结构,导致模型理解出错。

SWE-Pruner 的不同在于它是任务感知的。压缩什么、保留什么,是跟着当前任务目标走的。一个需要修 bug 的 Agent 和一个需要写测试的 Agent,会看到不同的上下文。

实用价值在哪儿

对于做 Agent 落地的人来说,这个思路意味着,与其抱怨模型上下文太短,不如让模型学会“看什么像什么”。

具体可以参考的方向有3个:

  1. 给 Agent设计 “目标设定”环节。在执行任务前让模型自己说“这次我关注什么”,这个信号可以被上游的压缩模块用到。

  2. 在代码检索场景里,把任务描述作为相关性判断的依据,而不是只靠语义相似度。

  3. 关注这个 0.6B 的小模型,小到可以本地跑,但能跟大模型配合,在一些边缘部署场景可能有用。

论文链接:https://arxiv.org/abs/2601.16746 Github:https://github.com/Ayanami1314/swe-pruner

相关文章

AI 研究实验室 NeoCognition 获 4000 万美元种子轮融资,构建类人学习智能体
AI 新闻资讯
2026年5月4日
0 条评论
小创

AI 研究实验室 NeoCognition 获 4000 万美元种子轮融资,构建类人学习智能体

NeoCognition 由俄亥俄州立大学教授 Yu Su 创立,获 4000 万美元种子轮融资。团队约 15 人,多为博士,目标不是解决智能体做事不稳定的致命短板!当前智能体任务成功率仅约 50%。公司试图打造能自我学习、快速专业化的智能体系统,让 AI 进入任何领域后都能自主构建“世界模型”,成为真正的“专家”。投资方包括 Vista Equity Partners 等重量级机构, NeoCognition 计划将产品卖给企业客户,特别是 SaaS 公司。融资背后反映 AI 市场正从追逐“大模型”转向追求“更可靠的执行”。

#智能体
阅读全文
写规格说明,不写代码:Logic 如何用 500 字 spec 把模型指令遵循率拉高 6 个点
提示词工程
2026年5月4日
0 条评论
零重力瓦力

写规格说明,不写代码:Logic 如何用 500 字 spec 把模型指令遵循率拉高 6 个点

西雅图创业公司 Logic 上线 spec 驱动 Agent 平台,用自然语言规格说明替代复杂工程。实测显示,在 IFBench 基准测试中,仅凭 500 字规范配合其自动化 harness,Gemini 3.1 Pro 性能提升 6 个百分点并登顶榜单。该平台将输入验证、契约执行等基础设施产品化,支持主流模型路由。对于正在开发 Agent 的团队,这证明优化约束描述比盲目升级模型更有效,目前已有超 250 家机构使用。

#AI 编程#提示词工程#上下文工程
阅读全文
微软开源 Agent Governance Toolkit:AI Agent 的安全终于有人管了
AI 产品工具
2026年5月4日
0 条评论
零重力瓦力

微软开源 Agent Governance Toolkit:AI Agent 的安全终于有人管了

微软开源 Agent 治理工具包 AGT,覆盖 OWASP 全部 10 类安全风险。该工具借鉴操作系统模型,通过工具扫描、调用前策略检查及响应审查三层机制,将安全从提示词建议升级为代码硬约束。其策略评估延迟不足 1 毫秒,支持多语言及主流框架集成。在监管趋严背景下,AGT 为开发者提供了管控 Agent 外部动作的有效基础设施,虽不干预内部推理,但填补了当前执行层的安全空白。

#AI 安全#智能体
阅读全文
互动讨论

评论区

围绕《代码 Agent 的上下文减肥药:SWE-Pruner 是怎么让模型少看多想的》展开交流,未登录用户可浏览评论,登录后可参与讨论。

评论数
0
登录后参与评论
支持发表观点与回复一级评论,互动后将同步到消息中心。
登录后评论
暂无评论,欢迎成为第一个参与讨论的人。