递归多智能体系统 Recursive Multi-Agent Systems
用潜空间递归把异构LLM智能体连成环路,以端到端梯度共同优化整个多智能体系统。
前置知识
自回归潜空间生成 (Autoregressive Latent Generation)
与标准解码器把每步隐藏状态 $h_t$ 投影到词表空间再采样下一个 token 不同,潜空间生成直接把 $h_t$ 作为下一步前向传播的输入嵌入,从而在整个推理过程中保留连续表示。其形式化定义是 $h_{t+1} = f_\theta([E_{\leq t}; h_t])$,生成的新隐藏状态也称为模型的「进行中的潜思考 (ongoing latent thought)」。
RecursiveMAS 的核心思想正是把这种「单模型内部」的潜空间递归模式扩展到「多智能体之间」,让异构 agent 通过连续向量而非文本交互,是理解整个框架的前提。
循环语言模型 RLM / LoopLM
把 Transformer 的 $L$ 层堆叠复用 $n$ 次以加深推理深度,形式化为 $H^{(r)} = f_\theta(H^{(r-1)})$,最终一轮的 $H^{(n)}$ 用于预测。这一思路在 LoopLM 等工作中被验证为一种新的推理扩展轴,可以替代单纯的参数或数据扩展。
论文把 RLM 中「同一组层 + 多个前向步」的范式类比到「同一个 agent + 多轮递归」,于是多智能体系统被视作一层类似 RLM 的循环结构,每个 agent 就好比 RLM 中的一个层。
Multi-Agent System (MAS) 协作模式
MAS 把多个具备不同角色/专长的 LLM 组合成协作系统,常见范式包括 Sequential(按流水线顺序分解问题)、Mixture-of-Agents(多专家并行+聚合器综合)、Distillation(专家→学生蒸馏)和 Deliberation(带工具调用的多轮审议)。
RecursiveMAS 声称结构无关 (structure-agnostic),需要在 Sequential、Mixture、Distillation、Deliberation 四种典型架构下都做实验验证,因此读者要先了解这些范式才能看懂 case。
LoRA 与全参数 SFT 微调
LoRA 在冻结原模型参数的前提下,仅训练两个低秩分解矩阵实现参数高效微调(通常 <1% 可训参数),适合大规模 LLM 适配;Full-SFT 则反向传播所有模型参数,更新能力强但成本高、易过拟合。
RecursiveMAS 选择只训练轻量级 RecursiveLink(约 0.31% 可训参数),其训练成本和 GPU 占用都明显低于 LoRA 和 Full-SFT,这是评估其复现性与实用性的关键对比点。
研究动机
现有 LLM 在复杂任务上常因容量有限、生成短视、解空间探索不足而失效,因此学界越来越多地通过多智能体系统 (MAS) 把异构模型协同起来。然而目前的 MAS 仍面临两类根本瓶颈:一是用文本做智能体间通信时,sequential dependency 导致长尾延迟——后一个智能体必须等前一个完整解码完才能开工;二是基于 prompt 的迭代优化(如 TextGrad、prompt tuning)虽能改进交互上下文,但每个智能体自身的能力无法提升;而要直接对每个 agent 做 SFT 又会因 agent 数量多、参数规模大而难以为继。因此作者指出 'updating all model parameters is non-trivial',且 $m|V|d_h$ 的词表解码代价居高不下,使得多轮递归的 MAS 既慢又难训。
本文的目标是论文希望在不更新任何底层 LLM 参数的前提下,把整个 MAS 重铸为一个统一的递归计算体——所有异构 agent 通过一个轻量级连接模块 RecursiveLink 在潜空间内反复迭代传递隐藏状态,从而同时实现:(i) 端到端的系统级共优化(whole-system co-optimization),让所有 agent 通过 shared credit 一起进化;(ii) 推理时的递归缩放 (test-time recursion scaling),可堆叠多轮 recursions 持续提升精度;(iii) 显著优于文本递归 MAS 的运行效率与 token 经济性。
与已有工作不同的是,作者的关键切入角度是跳出「单模型内递归」的 RLM 范式,将 recursion 提升到「系统级」——同时引入 latent-space communication 来取代传统的文本中介。这样既规避了文本递归的延迟与梯度消失(Theorem 4.1),又能保留文本生成的可解释性(仅在最后一轮递归解码输出)。换言之,他们把 MAS 从「一串串的文本交互」重组为「一段段的潜空间循环」,让协作本身成为可缩放的轴向,这一视角在 LoopLM、TextGrad、MoA 等既有工作中都未被探索。
核心方法
直觉上,作者希望把多智能体协作类比为「一个 RLM 的多层堆叠」:每个 agent 负责消费上一节点的连续 latent thought,产出一段新的 continuous latent thought,并通过可学习的 RecursiveLink 把它送入下一个 agent,最后一个 agent 在递归最后一轮才把 latent 输出解码成文本。从技术路线看,整个系统包含两个可学习模块——agent 内部的 inner RecursiveLink $R_{in}(h) = h + W_2 \sigma(W_1 h)$(保持同维、激活为 GELU、带残差),以及跨 agent 的 outer RecursiveLink $R_{out}(h) = W_3 h + W_2 \sigma(W_1 h)$(多一个 $W_3$ 用来对齐不同 agent 的 hidden dim)。递归运行 $n$ 轮后,只有最后一个 agent 的 LM Head 解码出最终答案,其余 $n-1$ 轮的交互全部在潜空间完成,因此被称为 'closed-loop latent collaboration'。
RecursiveMAS 与现有 MAS 的本质区别在于把 agent 之间的通信介质从 text 换成 latent state,并让整个系统图本身形成一条闭环。具体创新体现在三处:(1) RecursiveLink 设计遵循 'dense-to-shallow transition' 与 'cross-model transition' 两种语义,只学一个轻量残差 delta 而非线性投影,使其能在异构 agent(不同底座、不同 hidden dim)间无缝传递;(2) 端到端的训练范式 inner-outer loop:先并行预热每个 agent 的 inner link(监督信号是其自己文本答案经 input embedding 后的分布),再对所有 outer link 做共享 credit 的系统级 BP,从而让整条环路一起进化;(3) 通过递归轮次 $r$ 提供 test-time scaling axis——同一个训练好的系统可在推理时增加递归轮数继续提升精度,且训练-推理递归构成互补型 scaling law。
方法步骤详情
训练分两阶段。Inner-Loop Training:对每个 agent $A_i$ 冻结参数 $\theta_i$,把 ground-truth 文本 $y$ 通过 agent 自己的 input embedding 层 $\text{Emb}_{\theta_i}$ 得到参考语义向量 $\text{Emb}_{\theta_i}(y)$,再用同 agent 当前的前向 hidden states $H$ 通过 $R_{in}$ 与它计算 cosine 距离,损失 $\mathcal{L}_{in} = 1 - \cos(R_{in}(H), \text{Emb}_{\theta_i}(y))$,从而让 inner link 学到 '把内部隐状态重新对齐到输入嵌入空间' 的能力;这一步对所有 agent 并行执行,为后续系统级递归做 warm-up。Outer-Loop Training:解除连接顺序限制,把所有 agent 通过 outer link 串成环,unroll 整个系统 $n$ 轮做前向传播——第 $r$ 轮的输入 = 第 $r-1$ 轮所有 agent 的 latent output 经过 outer link 投影后的拼接——只在最后一轮把最后一个 agent 的输出经 LM Head 解码成文本,并以交叉熵 $\mathcal{L}_{out} = \text{CE}(S^{(n)}(S^{(n-1)}(\cdots S^{(1)}(x))), y)$ 同时反向传播到所有 outer link。所有 LLM 主干仍冻结,唯一可训练的是 inner/outer link(合计约 13.12M 参数,0.31%)。推理时执行相同的 $n$ 轮递归,仅最后一轮解码文本,输出 token 数相较文本递归基线降低 34.6% 至 75.6%。
技术新颖性
技术新颖性体现在三个方面:第一,把 RLM 「shared layer $\to$ shared agent」的类比是首创——之前 LoopLM、TiR 等工作都局限在「单模型内部复用权重」,本文首次让异构 LLM 组成的多智能体系统作为一个统一的递归计算图被优化;第二,RecursiveLink 同时承担 (i) 同 agent 的 'dense→shallow' 自递归 和 (ii) 异 agent 的 'cross-model' 对齐 两个任务,仅用一个 2-layer GELU residual projection 就能覆盖维度差异,并通过 ablation 验证 (Table 4) 残差分支比纯 2-layer 提升 3.6% on GPQA-D;第三,inner-outer 两阶段训练 + shared gradient credit 是首个让 MAS 实现 'whole-system co-optimization' 的方案。理论层面,论文给出 Proposition 3.1 的运行时复杂度证明(递归步把 $m|V|d_h$ 词表解码降为 $m d_h^2$ 隐空间变换)和 Theorem 4.1 的梯度稳定性证明(在 $\epsilon$-confident tokens 条件下,文本递归梯度范数 $\leq O(\epsilon) \ll 1$,而 RecursiveLink 梯度范数 $\geq \Omega(1 - \sqrt{\log(1/\delta)/d_h})$,从而避免 gradient vanishing)——这两条理论联合支撑了「为什么必须是 latent-space recursion 而不是 text recursion」。
实验结果
在 9 个 benchmark(数学 MATH500/AIME2025/AIME2026,科学 GPQA-Diamond,医学 MedQA,代码 LiveCodeBench-v6/MBPP+,检索 HotpotQA/Bamboogle)上,RecursiveMAS 全面超过基线。Accuracy:当 $r=1$ 时比 Recursive-TextMAS 平均高 8.1%,$r=2$ 时提升 19.6%,$r=3$ 时进一步达到 20.2%,说明递归越深优势越显著(Table 2)。例如在 Light Sequential 上 MATH500 从 71.9%/72.5%/69.1% 提升至 75.8%/76.6%/77.8%;在 Scaled Sequential 上 AIME2025 从 71.3%/70.7%/73.3% 跃升到 80.0%/86.0%/86.7%($r=3$ 几乎追平 GPT-4 级别水平)。Inference Time:递归轮数越大加速比越大,$r=1/2/3$ 时平均分别快 $1.2\times/1.9\times/2.4\times$(Figure 5),与 Proposition 3.1 的复杂度分析吻合。Token Reduction:同样从 34.6% ($r=1$) 提升到 65.5% ($r=2$) 再到 75.6% ($r=3$),意味着大部分中间通信留在 latent 空间(Figure 6)。Scaling Law (Figure 1 Up):训练-推理递归存在互补效应,对角线方向(左下→右上)的 accuracy 持续上行,说明训练递归教系统生成 'refinement-ready' 的 latent states,推理递归则把这些可递归结构进一步释放成测试时增益。跨架构泛化 (Figure 1 Down):在 Mixture Style 下比最强单个 specialist 平均提升 6.2%;在 Deliberation Style 上比原生 Tool-Caller 提升 4.8%(含 HotpotQA 41.4% vs 39.6%、Bamboogle 53.7% vs 49.8%,Table 8);在 Distillation Style 下把 Learner agent 平均提升 8.0%(AIME2026 由 76.7% 升至 83.3%),且相对 Expert 仍维持 $1.5\times$ 端到端速度优势(Table 6),证明复杂模型的能力可被潜空间蒸馏到轻量学生模型而不损失推理效率。资源成本 (Table 5):RecursiveMAS 显存峰值 15.29 GB(vs Full-SFT 41.40 GB / LoRA 21.67 GB)、可训参数仅 13.12M、估算成本 4.27 美元(vs Full-SFT 9.67 美元),同时拿到下游任务 74.9% 平均准确率(高于 LoRA 66.9% 与 Full-SFT 68.6%)。Ablation Studies:Table 4 验证 RecursiveLink 的 2-layer+residual 设计在 MATH500/GPQA-D/LiveCodeBench 上分别达到 88.0%/66.2%/42.9%,均高于 1-layer、res+1-layer 和 plain 2-layer;Table 9 / Figure 8 显示潜空间步数 $m$ 在 80 左右时各 benchmark 已饱和,超过 80 后基本无收益,说明只需适度的 latent-thought 预算即可。Case Study (Appendix F):在 MATH500 的 '224 is a perfect $n$th power' 题上,$r=1$ 因把 $m$、$n$ 当 factor pairs 列出而错答 6;$r=2/3$ 改为枚举 24 的所有 >1 的因子输出正确答案 7,表明递归轮次执行的是逐步 refinement。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| Math500 (Light) | Accuracy (%) | 77.8 (r=3) | Recursive-TextMAS 69.1 (r=3) | +8.7 |
| Math500 (Scaled) | Accuracy (%) | 88.2 (r=3) | Recursive-TextMAS 85.8 (r=3) | +2.4 |
| AIME2025 (Scaled) | Accuracy (%) | 86.7 (r=3) | Recursive-TextMAS 73.3 (r=3) | +13.4 |
| AIME2026 (Light) | Accuracy (%) | 34.0 (r=3) | Recursive-TextMAS 18.0 (r=3) | +16.0 |
| AIME2026 (Scaled, Pass@10) | Accuracy (%) | 86.0 (r=3) | Recursive-TextMAS 74.7 (r=3) | +11.3 |
| GPQA-Diamond (Scaled) | Accuracy (%) | 66.2 (r=3) | Recursive-TextMAS 58.6 (r=3) | +7.6 |
| MedQA (Light) | Accuracy (%) | 31.7 (r=3) | Recursive-TextMAS 28.5 (r=3) | +3.2 |
| LiveCodeBench-v6 (Scaled) | Accuracy (%) | 42.9 (r=3) | Recursive-TextMAS 38.7 (r=3) | +4.2 |
| MBPP+ (Light) | Accuracy (%) | 37.4 (r=3) | Recursive-TextMAS 29.3 (r=3) | +8.1 |
局限与改进
作者自己在 Section 6 与多处实验设置中承认了若干局限。第一,RecursiveMAS 的性能强烈依赖递归轮次 $r$,实验最优值在 $r=3$——再继续增大 $r$ 是否仍能单调提升、有无收益拐点,论文未给出 $r>3$ 的系统报告。第二,递归深度对某些任务仍有限制:在 Light Sequential 上 AIME2025 即使 $r=3$ 也只有 34.0%,相对 Scaled 86.7% 出现巨大鸿沟,说明模型规模(1-1.7B vs 4B)在挑战性数学题上的能力天花板无法被递归本身弥补。第三,论文中 RecursiveLink 仅以 $W_1, W_2, W_3$ 全连接+残差结构实现,没有探索 LoRA-style low-rank 或 attention-based richer adapter,可能在异构 agent 维度差异极大时受限。第四,作者提示 '当 $m$ 达到 80 后提升迅速饱和'(Figure 8),意味着 expressive latent thought 长度存在预算上限,复杂的中间推理仍需借助显式文本。第五,附录 B.3 提到所有实验在 H100/A100 上完成,但未给出端到端 wall-clock 训练时长与跨硬件复现细节。我自己的观察是:RecursiveMAS 当前仅在 2-3 轮的浅递归下评估,对于 '更深递归是否会出现性能衰减/语义漂移' 缺乏诊断;并且 inner-outer link 的权重数仅占全参 0.31%,对超大规模 LLM(如 >70B)的扩展性是否仍然线性可证 $\Theta(N(m d_h^2 + (t+m)d_h^2 + (t+m)^2 d_h))$,仍有未充分验证。
独立分析的弱点
W1:RecursiveLink 结构过于简化——目前只用了 2-layer MLP+残差,但面对差距悬殊的异构 agent(如 1.5B 与 10B),单纯线性投影难以充分对齐语义分布,可能更优设计是给 outer link 加一个 cross-attention 模块,让查询侧的 agent 主动去 select relevant latent tokens。W2:递归轮次对下游任务鲁棒性不足——Light Sequential 在 MedQA 上仅从 30.3% ($r=1$) 提升到 31.7% ($r=3$),说明并非所有领域都能从递归中获得线性增益,建议引入 'adaptive recursion depth' 机制,根据答案置信度(如 entropy)决定是否提前终止。W3:训练目标单一——inner loop 用 cosine similarity 对齐 embedding 分布,但 latent thought 与 ground truth 之间未必是简单 cosine distance 度量;当 agent 内部能力较弱时,可能产生误导性梯度;可以考虑加一个 KL 散度项或 distillation 项。W4:缺乏对失败的递归诊断——表 9 中 $m>80$ 后饱和,但 $r=3$ 与 $r>3$ 仍存在空缺,作者未讨论 '递归饱和点' 与 '信息损失';建议后续可视化 latent thought 在每一轮的 drift,并监控 potential semantic collapse。W5:可解释性受限——递归好处几乎全部留在 latent 空间,仅最后一轮文本输出,缺乏对 '哪一轮哪个 agent 的修正最重要' 的 attribution 分析;可以引入 attention rollout 或 probing classifier 来回溯每个递归步的边际贡献。
未来方向
F1:作者提出的方向:(a) 把 RecursiveMAS 推广到更大 agent 群体(>5 个)与更长递归轮次($r>3$),并系统研究 scalability 是否仍近似线性;(b) 设计 self-evolve 机制,让 agent 网络拓扑本身也可被递归优化。F2:基于成果可延伸:(c) 将 RecursiveLink 与 mixture-of-experts 的 router 联动,使 outer link 同时承担 routing 与 transfer 两类功能;(d) 与 RAG / tool-use 深度融合,因为 Deliberation Style 已经做了 first-step,可以探索 '递归调用工具' 的范式,让每个 recursion 步都决定是否调用 tool;(e) 把 latent recursion 扩展到 multimodal agent 之间(例如 VLM + LLM 协作),outer link 需对齐 vision encoder 与 language model 的表示空间;(f) 与 RLHF / DPO 类对齐目标结合,把递归优化目标从 next-token CE 替换为 preference-aware loss,可能进一步提升 reasoning-heavy 任务表现。
复现评估
开源情况:项目主页为 https://recursivemas.github.io,目前正文未给出完整代码仓库链接,但表 1 详细列出所有 agent 用到的 checkpoint(Qwen3-1.7B、Llama3.2-1B-Instruct、Qwen2.5-Math-1.5B-Instruct、Gemma3-4B-it、Llama3.2-3B-Instruct、Qwen3.5-4B、Qwen2.5-Coder-3B-Instruct、BioMistral-7B、DeepSeek-R1-Distill-Qwen-1.5B、Qwen3.5-2B、Qwen3.5-9B 等),均为可下载的 HuggingFace 模型。训练数据:基于 s1K、m1K、OpenCodeReasoning、ARPO-SFT 四个公开数据集,附录 B.3 公开了 role-specific 数据构造方法(包括如何用 Qwen3.5-397B-A17B rewrite 出 Planner/Critic/Solver 的角色级监督)。超参与算力:训练使用 AdamW + cosine LR scheduler,lr=5e-4、batch=4、max seq len=4096,全部在 H100/A100 上完成;推理时 top-p=0.95,温度 0.6(reasoning)或 0.2(code),最大生成长度按任务设定 2k/4k/16k。实现细节:附录 E 公开了所有 4 种协作模式的 prompt template,但 Appendix B.3 明确未给出 inner/outer link 的具体权重初始化方式、Kaiming normal 的 $\sigma$ 取值等微观参数。复现难度:中等偏易——只需 1-2 张 A100 即可在 Light Sequential 上复现 9 个 benchmark;Scaled Sequential 需要更多显存来 host 4B-10B agent 群。不确定项:RecursiveLink 的训练 step 数、recursive softmax 中的 temperature、跨 agent 拼接时的 token mask 策略等在正文中没有明示,建议在跑实验前先复现 Section 6 的 RecursiveLink Design ablation (Table 4) 来 sanity check。
论文图表