PARSER:并行读取,深度推理,面向长上下文 LLM Agent
顺序记忆 Agent 通过逐块读取、维护紧凑记忆状态来处理长文档,这使文档遍历与推理深度强耦合,不仅对证据出现位置高度敏感,推理延迟也随文档长度线性增长。 PARSER 将「读取」与「推理」解耦: - 一组轻量 subagent 各自绑定单个 chunk,并行读完整个文档; - 由 lead agent 通过迭代的 scatter-gather 轮次深入推理——每轮向所有 subagent 广播查询,聚合返回的证据,再基于已找到的内容构造更深一层的追问。 该设计把所有可学习行为集中在 lead agent 上,用 强化学习 优化,subagent 则保持冻结的现成模型。 在上下文长度 7K 至 896K tokens 的 多跳问答 任务上,4B 骨干的 PARSER 平均超越最强顺序记忆基线 5.7 分,在 896K tokens 时领先 12.0 分;扩展到 9B 骨干后,相较 DeepSeek-V4-Pro 高出 6.3 分。控制实验显示,PARSER 对证据位置、顺序与距离的扰动均保持稳健,而顺序方法在这些条件下会出现大幅准确率波动;同时推理延迟最多降低 11x。
论文精读
TL;DR PARSER 让多个子代理并行读取文档块,主代理通过迭代 scatter-gather 进行深度推理,解耦阅读与推理,在长上下文多跳 QA 上超越顺序记忆基线,并将延迟降低多达 11 倍。
问题
问题背景
长上下文 LLM 智能体在处理多跳问答、代码库分析、合同审查等任务时,需要在数万到近百万 token 的文档中定位并关联分散的证据。
现有方法局限
当前主流方案是 顺序记忆智能体(sequential memory agents),例如 MemAgent、ReMemR1 等。它们按序读取文本块,并不断更新一个紧凑的 memory state。这种设计将 文档遍历顺序 与 推理深度 绑定,造成两个具体问题:
- 证据位置敏感:如果关键证据出现在文档前部,后续 memory 压缩可能丢失信息;如果出现在后部,则需要读完大量无关块后才能定位。
- 推理延迟线性增长:每读完一个块都要等待下一步推理,端到端延迟随文档长度线性增加,难以扩展到超长上下文。
为什么这个问题难 / 重要
超长上下文中多跳证据往往跨块分布,顺序方法难以在有限 memory budget 下保留远期依赖。并行读取所有块虽然能降低延迟,但如何在没有完整文档视图的情况下聚合局部证据、并驱动多轮深层推理,是一个 架构与优化 的双重挑战。业界对高效长上下文智能体的需求正在快速增长:RAG 系统、企业级知识库问答、长代码库理解等场景都需要在准确率与延迟之间取得平衡。
行业类比
这与 RAG 流水线 中“先并行召回候选块、再由生成器逐步精读合成答案”的模式类似,但 PARSER 将这一思路用于更通用的推理智能体,而不是仅做单轮检索增强。
核心洞察
- “读”与“推理”完全解耦:并行子代理只负责读取每个 chunk,lead agent 通过多轮 scatter-gather 进行深层推理。与 sequential memory agents(如 MemAgent)不同,后者将文档遍历与推理深度绑定,导致对证据出现位置、顺序和距离敏感,且推理延迟随文档长度线性增长;PARSER 将这两个过程分离,使得推理深度不再受限于遍历方式,从而实现在长上下文多跳 QA 上的位置鲁棒性。
- 可学习行为集中在 lead agent,子代理保持冻结,实现高效模块化强化学习。已有方法通常需要对整个 agent 系统进行端到端训练,计算成本高且难以扩展到长上下文;PARSER 只对 lead agent 进行 RL 优化,子代理使用现成模型,显著降低训练开销的同时,在 4B 骨干上平均超过最强 sequential memory baseline 5.7 分,推理延迟降低 11 倍。这种设计也意味着 lead agent 学到的查询策略可迁移到不同子代理实现,提高了工程灵活性。
方法
输入与切分
长文档按固定 token 数切分为多个 chunk,每个 chunk 分配给一个轻量 子代理(subagent),子代理基于冻结的 off-the-shelf 模型,只负责读取本地 chunk 并响应用户查询,自身不参与学习。
关键模块与流程
并行读取:所有子代理同时处理各自 chunk,互不通信,首次读取后保存 chunk 内线索。
主导代理(lead agent)与 scatter-gather 推理:可学习的 lead agent(如 4B/9B 参数)执行多轮迭代:
- 广播查询:lead agent 根据当前已有证据生成一个聚焦查询,广播给所有子代理;
- 证据聚合:每个子代理从自己的 chunk 中返回相关片段,lead agent 汇总所有返回;
- 深层追问:lead agent 综合聚合结果,生成更深的后续查询,进入下一轮 scatter-gather,直到认为信息足够。
优化:所有可学习行为集中在 lead agent,通过 agentic reinforcement learning 优化,奖励基于多跳问答最终答案的正确性;子代理保持冻结。
输出
lead agent 输出最终答案。
与顺序记忆代理(如 MemAgent)相比,PARSER 将文档读取与深度推理解耦,避免证据位置敏感和延迟线性增长,同时将学习信号集中到单一代理,降低训练复杂度。
实验
实验设计
PARSER 在 multi-hop QA 任务上评估,覆盖两个数据集:HotpotQA (in-distribution) 与 2WikiMultiHopQA (out-of-distribution)。上下文长度从 7K 到 896K tokens 共多个档位。模型规模采用 4B 和 9B 两种 backbone。对比基线包括 sequential memory agents(如 MemAgent、ReMemR1)、full-context answering 与 direct corpus interaction。子代理使用 frozen off-the-shelf models,仅 lead agent 通过 RL 优化。控制实验对证据位置、顺序、距离进行扰动。
关键发现
- 4B backbone 平均准确率较最强 sequential memory baseline 高出 5.7 点;在 896K tokens 上下文时提升达到 12.0 点。
- 扩展到 9B backbone 后,PARSER 超越 DeepSeek-V4-Pro 达 6.3 点。
- 推理延迟较 sequential methods 最多降低 11 倍。
- 在证据位置、顺序、距离的扰动实验中,PARSER 保持稳定,而 sequential 方法出现大幅精度波动。
与基线的深度对比
Sequential memory agents 一边读 chunk 一边更新 compact memory,文档遍历深度与推理步骤强耦合,导致对证据出现位置高度敏感,且延迟随文档长度线性增长。PARSER 将读取与推理解耦:子代理并行扫描全部 chunk,lead agent 通过 scatter–gather 迭代,从全局聚合证据并生成更深层 query。该设计将可学习行为集中到 lead agent,用 RL 优化即可,子代理无需训练,工程落地时更易维护和替换。同时,并行读取带来了底延迟和位置鲁棒性,但需要管理多子代理的通信开销与 chunk 大小对召回的影响。
行业影响
落地场景
PARSER 适用于超长文档的多跳推理任务,如企业知识库问答(跨多个手册/工单定位证据并综合回答)和金融合规审计(跨年报、合同验证数据一致性、识别风险条款)。并行子代理读取全量文档,避免顺序扫描导致的证据遗漏。
商业价值
- 降本: 推理延迟最高降低 11x,且不再随上下文长度线性增长;轻量冻结子代理 + 小参数主代理 RL 训练,推理和训练成本均较低。
- 增效: 在 896K tokens 多跳 QA 上,4B 模型超强顺序基线 12 点,9B 超 DeepSeek-V4-Pro 6.3 点;对证据位置/顺序/距离鲁棒,减少因文档排列变化导致的错误,提升用户信任。
与现有产品/工作流集成
可作为插件集成到 LangChain / LlamaIndex 等编排框架:将文档分块后注册为子代理池,主代理负责 scatter-gather 迭代查询。子代理直接复用 off-the-shelf 模型,主代理用 4B/9B 轻量 RL 微调,可替换现有 RAG 的检索+阅读环节,嵌入企业搜索、法律科技、金融分析等产品后端,无需重构数据管道。
局限
- **计算资源总量未显著降低**:PARSER 将文档切片后交给多个子代理并行读取,虽然推理延迟降低,但**总 token 消耗**和**总 GPU 显存**可能高于单序列顺序处理。每个子代理需要加载完整模型权重,且 scatter-gather 轮次中 lead agent 需要聚合所有子代理返回的证据,通信和聚合开销随轮次和子代理数量线性或超线性增长。论文未提供与顺序基线在**总 FLOPs**或**总成本**上的对比,工程部署时需权衡延迟与资源利用率。
- **子代理冻结限制整体能力上限**:可学习的只有 lead agent,子代理使用 off-the-shelf 模型且不参与梯度更新。这简化了训练,但也意味着子代理的**信息提取能力**受其原始模型固化,无法通过 RL 改进。如果子代理在特定领域或长距离指代上表现不佳,lead agent 无法通过反向传播纠正子代理行为,只能通过改变查询策略间接适应。在极长上下文或高难度证据稀疏场景下,子代理召回率低会直接限制最终答案质量。
- **任务泛化与超长上下文验证不足**:实验集中在 multi-hop QA(HotpotQA、2WikiMultiHopQA),虽然覆盖最长 896K tokens,但未展示在**开放域问答、推理密集型任务、代码理解或 agent 工具调用**上的表现。PARSER 的 scatter-gather 机制依赖 lead agent 提出精准查询,若任务本身不依赖“证据检索”范式(如摘要、翻译、编程),查询驱动的聚合可能引入噪声或丢失全局语义。此外,RL 训练数据构造方法对分布外数据集的泛化性仍需进一步验证。