返回文章列表
推理与部署

Nexus:让智能体不必每轮重读工具说明书

阅读约 3 分钟

导语

在 MCP 等工具调用框架中,智能体每轮对话都可能重新处理一批冗长的工具 schema。工具数量增加后,预填充阶段的计算和上下文占用会迅速成为 time-to-first-token(TTFT)的主要瓶颈。Nexus 试图把“找到该用哪个工具”和“理解全部工具说明”拆开,从而减少不必要的 schema 计算。

核心做法

  • 先检索,后生成参数。 Nexus 建立 INT8 语义查找缓冲区(SLB),并用经过校准的交叉编码器 margin gate 筛选工具。模型不再每轮把整个工具注册表放入主上下文,而是根据请求检索候选项。
  • 使用压缩签名。 参数生成依赖工具的文本签名,而不是依赖拼接进上下文的完整 KV-cache。该签名的中位长度约为 19 个 token,因此可以显著减少主上下文中的 schema 内容。
  • 将 KV 拼接作为辅助路径。 对已经编译的 schema KV block,Nexus 尝试直接移植到当前上下文。但这条路径并非无条件安全:RoPE 会让不同位置的相位发生变化,偏离原锚点后可能破坏注意力结果。
  • 出现风险时逐步修复。 当上下文深度超过 P=256,系统采用深度自适应的后缀重新解码;必要时升级为完整重预填充。这里的“永不退化”主要指输出保真度,而不是延迟始终更低。

实验读数与边界

在 Qwen2.5-14B-Instruct Q4_K_M、Apple Silicon 统一内存环境中,检索式路由在工具规模扩展到 250 个时,准确率仍接近 89%;相比完整 schema 重预填充,首个参数 token 提前到来约 1.66 倍,主上下文 token 约减少 80%。同样的规模下,简单拼接全部 schema 已无法容纳于上下文窗口。

KV 拼接在中等深度可以带来约 1.1 至 1.7 倍 TTFT 加速,但随着上下文变深,优势会收窄至与基线相当。修复机制还能保持 top-1 输出一致性,KL 散度接近 0,不过个别阶段延迟可能先降至基线的 0.98 倍。论文还报告,单纯依赖无参考 drift gate 预测位置漂移并不可靠,Spearman 相关系数只有 0.193。

意义与影响

Nexus 的关键价值不在于证明 KV-cache 可以随意搬运,而在于给出了更稳妥的优先级:工具路由应尽量与完整 schema 预填充解耦,KV 拼接则应被视为有明确边界、需要回退机制的优化。对于拥有大型工具注册表的智能体,这种设计可能降低 TTFT 和上下文压力;但其数字表现来自单一模型、量化配置与硬件组合,不能直接外推到其他模型。未来评估还需要覆盖不同 RoPE 实现、上下文长度、模型规模和真实工具调用分布。

arXiv

评论

正在确认登录状态……

正在加载评论……

相关文章

CCTest · Blog
Daedalus-150M:为普通 CPU 推理反向设计的卷积—注意力混合模型
推理与部署
cctest.ai
推理与部署

Daedalus-150M:为普通 CPU 推理反向设计的卷积—注意力混合模型

Daedalus-150M 没有把大模型架构缩小后再适配 CPU,而是先锁定单用户、单 token、4-bit 权重等部署条件,再决定网络结构。它用短卷积替代三分之二的全注意力模块,在长上下文下实现更稳定的缓存开销和更高解码速度。

阅读全文
CCTest · Blog
Llama-Mobile:用2.7比特量化把视觉语言模型带到移动端
推理与部署
cctest.ai
推理与部署

Llama-Mobile:用2.7比特量化把视觉语言模型带到移动端

Llama-Mobile提出一套面向移动部署的视觉语言模型量化方案,结合模型自生成数据与适配Arm CPU的2.7比特格式,将Llama 3.2 11B Vision Instruct压缩至3.7 GB。实验显示,在8比特激活下,模型在多项标准视觉问答任务中仍保持较强表现。

阅读全文