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的隐藏状态轨迹上:
- Capability Head(能力头):预测当前模型能否独立解决该实例。如果置信度低,触发早期交接(early handoff)——在完整生成前就将任务委托给更强的协作模型,省去无效推理的Token消耗
- Resolution Head(解析头):预测当前场景的最优决策路径,四选一:
- Direct Answering:直接回答,无需额外步骤
- Tool Use:需要调用外部工具
- Clarification:信息不足,需要追问用户
- 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代表了从"人工指定路由规则"到"从模型内部信号推断路由"的范式转变
可执行建议¶
- AndroidWorld场景直接相关:论文在AndroidWorld移动端基准上验证了90.7%的大模型用量减少,这对你的移动端Agent开发有直接借鉴——可以考虑在端侧部署小模型,通过类似控制头机制判断何时上云请求大模型
- 跨模型版本的成本优化:如果你的Agent系统使用了多个模型(DeepSeek V4 Flash+Pro),MHLC的"冻结LLM+外挂头"思路可以用来自动判断何时将问题从Flash升级到Pro
- 关注代码开源进度:论文作者来自阿尔伯塔大学,项目尚未明确开源但方法论可复现——控制头本质是一个小型MLP,训练不复杂
- 与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 |