arXiv:2607.17751cs.IRcs.AI2026-07

让智能体更准地选工具,解决指令模糊和外部干扰问题。

MagicSelector: Joint Optimization for Agent Tool Selection via Counterfactual Decomposition and Progressive Reranking

论文配图:MagicSelector: Joint Optimization for Agent Tool Selection via Counterfactual Decomposition and Progressive Reranking
图 1 · 摘自论文原文
  • 用反事实分解细化任务逻辑,提升指令理解精度
  • 通过渐进重排和动态筛选,提高工具召回率与去噪能力
  • 适合需要高精度工具选择的多轮交互系统

我们提出 MagicSelector,一个联合优化框架,融合反事实任务分解、渐进式重排与动态 Top-K 机制,以应对智能体工具检索中的核心挑战。该框架能将模糊用户指令转化为可执行的原子子任务,在跨域(OOD)场景下有效缓解冗余噪声与严重上下文干扰。其三大贡献为:(1)基于偏好的反事实任务分解机制,利用反事实奖励量化分解对检索排序的边际因果收益,实现细粒度结构监督;(2)基于自蒸馏难负样本挖掘的渐进式工具重排方法,同时优化点级与列表级相关性,增强高度相似工具间的判别力;(3)双语义边界感知的动态 Top-K 策略,自适应监测重排分数突变与工具间语义偏移,动态截断候选列表,在最大化相关工具召回的同时过滤长尾噪声。在我们构建的首个面向移动端多轮交互、带过程级标注的 MTDTool 基准上评估,MagicSelector 在工具检索准确率、跨域泛化能力及整体令牌效率方面显著优于现有最优方法,验证了框架的有效性。

原文摘要 · Abstract (English)

We present MagicSelector, a joint optimization framework integrating Counterfactual task decomposition, Progressive reranking, and Dynamic Top-K, designed to address the fundamental challenges of tool retrieval in agents. MagicSelector is a specialized framework capable of translating ambiguous user instructions into executable atomic subtasks and guiding high-precision tool retrieval, effectively mitigating redundant noise and severe context distraction in out-of-domain (OOD) scenarios. We empower MagicSelector with these capabilities through three key contributions: (1) a preference-guided counterfactual task decomposition mechanism that utilizes a counterfactual reward to quantify the marginal causal gain of decomposition on retrieval ranking, effectively imposing fine-grained structural supervision on logical coherence; (2) a progressive tool reranking method driven by self-distillation hard negative mining, which optimizes both point-wise and list-wise relevance to enhance fine-grained discrimination among highly similar tools; and (3) a dual semantic boundary-aware dynamic Top-K strategy that adaptively monitors reranking score cliffs and inter-tool semantic shifts to dynamically truncate the candidate list, maximizing relevant tool recall while filtering long-tail noise. Evaluated on MTDTool, the first task decomposition benchmark we constructed tailored for mobile multi-turn interactions with process-level annotations, MagicSelector yields promising performance. Extensive experiments demonstrate that MagicSelector significantly outperforms state-of-the-art methods in terms of tool retrieval accuracy, OOD generalization capability, and overall token efficiency, thereby demonstrating the effectiveness of our proposed framework.

工具选择智能体任务分解重排

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。