Skip to content

Multi-Head Latent Control:基于LLM隐藏状态的Agent决策路由

tags: #Agent-Decision #Latent-Control #Model-Routing #Tool-Calling #LLM-Agent source: arXiv 2607.14277 | AI论文速递2026-07-28 score: 技术深度8/10 | 实用价值8/10 | 时效性9/10 | 领域匹配9/10 | 综合 8.5/10

核心概念

Multi-Head Latent Control (MHLC) 是一种在冻结LLM上外挂轻量控制头的Agent决策方法。核心思想:不需要修改模型权重,也不依赖prompt-level路由或外部编排器,而是直接从LLM生成过程中的隐藏状态轨迹(hidden-state trajectories)中读取信号,判断当前模型能否解决该任务、是否需要委托给更强的模型、是否需要调用工具还是直接回答。

设计原理

为什么要从隐藏状态推断控制决策

现有Agent决策方法有三类,各有痛点:

方法 代表 核心问题
Prompt-level路由 在prompt中嵌入决策指令 依赖输入侧信号,无法感知模型内部的困惑度/不确定性
外部编排器 LangGraph、CrewAI 每次路由需要额外LLM调用,增加延迟和成本
任务特定微调 LoRA/QLoRA 模型升级时需要重新微调,维护成本高

MHLC的关键洞察:LLM在生成过程中的隐藏状态已经包含了"这个任务我能不能做""我是否在胡编"的信号,只是这些信号没有被显式提取和利用。

双头架构

MHLC包含两个轻量级控制头,都只训练在冻结LLM的隐藏状态轨迹上:

  1. Capability Head(能力头):预测当前模型能否独立解决该实例。如果置信度低,触发早期交接(early handoff)——在完整生成前就将任务委托给更强的协作模型,省去无效推理的Token消耗
  2. Resolution Head(解析头):预测当前场景的最优决策路径,四选一:
  3. Direct Answering:直接回答,无需额外步骤
  4. Tool Use:需要调用外部工具
  5. Clarification:信息不足,需要追问用户
  6. Abstention:超出能力范围,应该拒绝回答

核心优势:两个头只训练在冻结LLM backbone的隐藏状态上,实现事后适配(post hoc adaptation)——模型升级后无需重新训练控制头,只需在新的隐藏状态上做一次推理。

Trade-off分析

  • 为什么用冻结LLM而不是微调:微调虽然可能更精准,但模型升级时需要重新训练。冻结方案牺牲了理论上限,换来了跨模型版本的通用性——这在生产环境中价值更大
  • 额外推理成本:控制头本身参数量极小(相比LLM backbone),增加的推理延迟可忽略。但Capability Head需要读取部分生成轨迹,这意味着在做出"是否交接"决定前已经消耗了一些Token——不过比完整生成了错误答案再修复要省得多
  • 数据需求:只训练控制头,不需要修改LLM,训练数据只需"隐藏状态→正确决策标签"对,远少于一版模型微调

关键实现

实验结果

AndroidWorld(移动端Agent基准)上的路由执行(小模型+大模型)场景: - 大模型用量减少最高90.7% - 跨基准平均减少27-53% - 同时保留了大部分大模型性能

在语言和视觉语言场景下均有效,跨模态泛化性经过验证。

与现有Agent路由方案对比

方案 信号来源 模型修改 跨版本通用
Prompt路由 输入文本 部分
外部编排器 独立LLM调用
LoRA微调 模型权重
MHLC 隐藏状态 否(外挂头)

关联分析

  • Agent-Control-Flow — MHLC提供了一种新的控制流决策机制,区别于Prompt路由和外部编排
  • Claude-Agent-Harness-Patterns — MHLC的Capability Head可作为Harness层面的能力感知层
  • NVIDIA-OO-Agents — 在OO Agent框架中,控制头可以作为Agent类的方法级决策开关
  • Agent Model Routing方向:MHLC代表了从"人工指定路由规则"到"从模型内部信号推断路由"的范式转变

可执行建议

  1. AndroidWorld场景直接相关:论文在AndroidWorld移动端基准上验证了90.7%的大模型用量减少,这对你的移动端Agent开发有直接借鉴——可以考虑在端侧部署小模型,通过类似控制头机制判断何时上云请求大模型
  2. 跨模型版本的成本优化:如果你的Agent系统使用了多个模型(DeepSeek V4 Flash+Pro),MHLC的"冻结LLM+外挂头"思路可以用来自动判断何时将问题从Flash升级到Pro
  3. 关注代码开源进度:论文作者来自阿尔伯塔大学,项目尚未明确开源但方法论可复现——控制头本质是一个小型MLP,训练不复杂
  4. 与Tool Calling结合:Resolution Head的四分类(直接回答/工具调用/追问/拒绝)可以直接集成到Agent的工具决策层,减少无效的工具调用

自评

维度 分数 权重 加权
摘要质量 8 0.25 2.00
技术深度 8 0.25 2.00
相关性 9 0.20 1.80
原创性 7 0.15 1.05
格式规范 9 0.15 1.35
加权总分 8.20