TRACER: Trajectory-Aware Tool Retrieval for LLM Agents
TRACER:面向 LLM Agent 的轨迹感知工具检索
🧭 Agent 工具检索与轨迹条件化 ★ 深度解读
Microsoft
利用历史工具调用轨迹的 trie 索引和状态自适应双先验混合,在保持准确率的同时大幅降低工具选择上下文成本
📌 工作参考价值(5 点)
- 在电商客服 Agent 中,可将历史成功调用轨迹构建为 trie,按当前执行前缀(如已调用 get_user_details)约束下一步候选工具,把工具选择从独立检索变为序列预测,减少无关工具干扰。
- 对每个转移边存储历史对话嵌入,计算当前上下文与该嵌入的 cosine 相似度,能区分描述相似但用途不同的工具(例如退货标签 vs 退货操作),避免模型误选危险操作。
- 使用轻量线性头学习每个状态的混合权重 w 和温度 T,动态调节经验先验(转移频率)与上下文先验(语义相似度)的重要性,比固定权重(如 α=0.5)更鲁棒,尤其适合工具分布不均匀的注册表。
- 若注册表中存在大量与状态无关的通用工具(如 KB_search、grep),可引入一个小型全局工具集 G(按前缀熵选取 top 工具)作为兜底,能显著提升召回。
- 工程实现上,trie 索引和边记忆查询开销极低,可完全离线构建,在线只需前缀查找和点积计算,适合大规模工具注册表的高并发场景。