DART-SD: Diamond-topology Aware Retrieval and Tuning for Self-Distillation of Multi-Turn Tool-Calling Agents
Teaching a tool-using AI agent by fixing only the exact spot where it goes wrong beats teaching it the whole answer
When a task has several sub-goals that can be done in any order, the correct solutions form a branching, diamond-shaped set of paths, but standard training forces models to copy one single path, punishing valid alternatives the student itself discovers. DART-SD instead builds a graph of the teacher's successful and failed attempts, finds the exact point (called CTB) where a student's failed attempt first leaves the recoverable region, and trains the model only on how to recover from that point onward. Tested on Qwen3-4B and Qwen3-8B, this approach beat existing supervised and reinforcement-learning baselines across five tool-use benchmarks, sometimes even outperforming the teacher model itself.
METAL MEDIA explanatory visual
Teaching a tool-using AI agent by fixing only the exact spot where it goes wrong beats teaching it the whole answer
- 01Problem: tasks with order-independent sub-goals form a diamond-shaped lattice of valid solution paths, but standard SFT and GRPO-style reinforcement learning flatten this into a single linear trajectory, wrongly penalizing valid alternative paths the student model finds on its own
- 02Method: teacher rollouts (both successful and failed) are turned into a graph called the Interaction-State Transition Graph (ISTG); a student's failed attempt is mapped onto this graph to find the earliest point where it left the region still reachable to success, called the Critical Topological Breakpoint (CTB), and training loss is applied only to the recovery continuation generated after that point, repeated over multiple rounds
- 03Evaluation: trained on the FTRL dataset and tested on five benchmarks (FTRL, BFCL, ToolHop, tau-bench, RoTBench), DART-SD outperformed existing distillation methods (SCoRe-SFT, OPSD) and reinforcement learning methods (FTRL-GRPO, ToolRL, MatchTIR) on both Qwen3-4B and Qwen3-8B backbones
- 04Efficiency gains: across training iterations, average tool calls in successful trajectories dropped from 4.23 to 3.55 while task success kept improving, ending up shorter than the human-constructed reference trajectories (average 4.02 calls); the point of first failure (CTB) also shifted later over iterations, showing the student's reliable range kept extending
- 05General capability tests (instruction following, math reasoning, general knowledge) improved on average from 43.92 to 49.89, showing that tool-use training didn't come at the cost of the model's other abilities
What they did
- Problem: tasks with order-independent sub-goals form a diamond-shaped lattice of valid solution paths, but standard SFT and GRPO-style reinforcement learning flatten this into a single linear trajectory, wrongly penalizing valid alternative paths the student model finds on its own
- Method: teacher rollouts (both successful and failed) are turned into a graph called the Interaction-State Transition Graph (ISTG); a student's failed attempt is mapped onto this graph to find the earliest point where it left the region still reachable to success, called the Critical Topological Breakpoint (CTB), and training loss is applied only to the recovery continuation generated after that point, repeated over multiple rounds
- Evaluation: trained on the FTRL dataset and tested on five benchmarks (FTRL, BFCL, ToolHop, tau-bench, RoTBench), DART-SD outperformed existing distillation methods (SCoRe-SFT, OPSD) and reinforcement learning methods (FTRL-GRPO, ToolRL, MatchTIR) on both Qwen3-4B and Qwen3-8B backbones
- Efficiency gains: across training iterations, average tool calls in successful trajectories dropped from 4.23 to 3.55 while task success kept improving, ending up shorter than the human-constructed reference trajectories (average 4.02 calls); the point of first failure (CTB) also shifted later over iterations, showing the student's reliable range kept extending
- General capability tests (instruction following, math reasoning, general knowledge) improved on average from 43.92 to 49.89, showing that tool-use training didn't come at the cost of the model's other abilities
| Methods | FTRL | BFCL | ToolHop | τ-bench | RoTBench | Avg. | ||||
|---|---|---|---|---|---|---|---|---|---|---|
| Solve-P | Solve-R | Solve-F1 | Multi-Turn | AC | Pass^1 | TS | PI | CF | ||
| Qwen3-4B | ||||||||||
| Base | 21.00 | 26.54 | 21.81 | 10.14 | 20.20 | 15.15 | 69.52 | 26.31 | 16.07 | 25.19 |
| ♠ SFT | 34.26 | 49.02 | 37.96 | 14.57 | 40.50 | 21.82 | 71.67 | 44.40 | 24.40 | 37.62 |
| ♠ SCoRe-SFT | 23.79 | 28.20 | 24.61 | 12.75 | 25.83 | 11.52 | 60.36 | 31.19 | 18.33 | 26.29 |
| ♠ OPSD | 22.37 | 29.21 | 23.50 | 11.00 | 20.20 | 23.03 | 68.93 | 35.24 | 21.67 | 28.35 |
| ♢ FTRL-GRPO | 36.83 | 41.71 | 37.84 | 13.50 | 29.25 | 20.61 | 70.36 | 32.62 | 20.24 | 33.66 |
| ♢ ToolRL | 26.89 | 33.23 | 28.47 | 9.88 | 20.60 | 18.18 | 69.64 | 38.21 | 24.05 | 29.91 |
| ♢ MatchTIR (OT) | 23.55 | 29.27 | 24.95 | 10.38 | 26.23 | 16.36 | 72.02 | 32.74 | 20.36 | 28.43 |
| ♢ MatchTIR (KM) | 25.54 | 31.10 | 26.50 | 10.00 | 26.63 | 21.82 | 72.02 | 33.81 | 21.43 | 29.87 |
| ♠ DART-SD (Ours) | 36.70 | 48.16 | 39.77 | 23.88 | 42.11 | 23.03 | 72.02 | 42.38 | 24.52 | 39.17 |
| Qwen3-8B | ||||||||||
| Base | 21.18 | 30.71 | 23.48 | 18.38 | 28.54 | 10.13 | 75.52 | 36.29 | 22.19 | 29.60 |
| ♠ SFT | 38.08 | 50.95 | 41.89 | 19.25 | 43.52 | 26.06 | 75.95 | 48.81 | 30.24 | 41.64 |
| ♠ SCoRe-SFT | 31.28 | 34.48 | 31.58 | 19.25 | 30.25 | 18.18 | 70.36 | 40.48 | 24.52 | 33.38 |
| ♠ OPSD | 24.37 | 34.88 | 26.68 | 20.50 | 41.11 | 21.21 | 75.24 | 43.21 | 27.38 | 34.95 |
| ♢ FTRL-GRPO | 37.66 | 45.49 | 40.22 | 35.25 | 34.57 | 23.03 | 77.02 | 42.74 | 27.02 | 40.33 |
| ♢ ToolRL | 32.49 | 41.07 | 35.00 | 20.50 | 44.72 | 25.45 | 75.71 | 51.43 | 33.10 | 39.94 |
| ♢ MatchTIR (OT) | 29.45 | 34.89 | 30.65 | 21.38 | 38.79 | 24.24 | 75.00 | 40.36 | 25.36 | 35.57 |
| ♢ MatchTIR (KM) | 33.07 | 40.42 | 35.37 | 23.25 | 41.71 | 26.06 | 75.71 | 42.26 | 26.90 | 38.31 |
| ♠ DART-SD (Ours) | 42.00 | 54.13 | 45.66 | 27.63 | 45.03 | 27.12 | 75.83 | 57.38 | 35.48 | 45.58 |
| Task | DART-SD | Golden | ||||
|---|---|---|---|---|---|---|
| Iter1 | Iter2 | Iter3 | Iter4 | Iter5 | ||
| Single | 1.23 | 1.15 | 1.10 | 1.09 | 1.07 | 1.00 |
| Multi | 3.83 | 4.63 | 4.54 | 4.49 | 4.41 | 4.71 |
| Para-Single | 3.29 | 2.65 | 2.27 | 2.26 | 2.22 | 2.11 |
| Para-Multi | 7.25 | 6.48 | 5.97 | 5.82 | 5.62 | 6.97 |
| Overall | 4.23 | 3.99 | 3.71 | 3.65 | 3.55 | 4.02 |
| Solve-F1 | 40.37 | 42.95 | 43.78 | 44.67 | 45.66 | – |
| Task | CTB Position | ||||
|---|---|---|---|---|---|
| Iter1 | Iter2 | Iter3 | Iter4 | Iter5 | |
| Single | 0.034 | 0.186 | 0.137 | 0.249 | 0.077 |
| Multi | 0.500 | 1.624 | 1.729 | 1.876 | 1.953 |
| Para-Single | 0.095 | 0.219 | 0.299 | 0.328 | 0.313 |
| Para-Multi | 0.395 | 1.524 | 1.692 | 1.791 | 1.816 |
| Overall | 0.348 | 1.185 | 1.310 | 1.421 | 1.452 |
| Δ vs. Iter1 | – | +0.837 | +0.962 | +1.073 | +1.104 |
| Method | FTRL | BFCL | ToolHop |
|---|---|---|---|
| Qwen3-8B | 29.74 | 40.00 | 42.21 |
| FTRL-GRPO | 32.85 | 41.50 | 36.72 |
| ToolRL | 26.72 | 34.25 | 32.93 |
| MatchTIR (KM) | 37.33 | 47.13 | 46.16 |
| DART-SD | 41.03 | 49.75 | 46.43 |
| Method | IFEval | AIME24 | AIME25 | MMLU | Avg. |
|---|---|---|---|---|---|
| Qwen3-8B | 34.75 | 46.67 | 23.33 | 70.94 | 43.92 |
| SFT | 35.30 | 43.33 | 26.67 | 71.43 | 44.18 |
| DART-SD | 45.29 | 50.00 | 30.00 | 74.27 | 49.89 |
| Method | Solve-P | Solve-R | Solve-F1 |
|---|---|---|---|
| Qwen3-8B | 21.18 | 30.71 | 23.48 |
| +SD | 36.23 | 44.46 | 38.10 |
| +CTB | 36.62 | 46.32 | 39.51 |
| +Progressive SFT | 41.32 | 49.65 | 43.93 |
| +ISTG (Ours) | 42.00 | 54.13 | 45.66 |
Why it matters
It shows that for AI agents that repeatedly call tools to solve problems, forcing them to memorize one fixed correct sequence is wasteful and can destroy useful strategies the model already has. Precisely locating and fixing only the failure point offers a more efficient way to build smaller, smarter, and more efficient tool-using agents.
Terms in this paper
- SFT (Supervised Fine-Tuning) · Additional training where a model is taught to copy given example answers
- GRPO · A reinforcement learning method that adjusts a model's behavior using reward signals
- ISTG (Interaction-State Transition Graph) · A graph representing tool-use progress as nodes based on what information has been acquired
- CTB (Critical Topological Breakpoint) · The first point where a student's attempt departs from a path that could still succeed
- Self-distillation · Retraining a model using outputs it generated itself
Figures we cannot republish
- Figure 1: Comparison of training paradigms. (a) SFT Teacher Boosting applies an indiscriminate global loss, which overwrites valid exploration. (b) Standard-RL (GRPO) misassigns credit through a uniformly distributed reward spread. (c) DART-SD dynamically identifies the Critical Topological Breakpoints (CTB) and applies localized correction while preserving valid exploration.
- Figure 2: Overview of DART-SD. (1) DART-SD constructs an ISTG from teacher rollouts, where main and auxiliary nodes model information acquisition and useless exploration. (2–3) Failed student rollouts are collected and replayed in the same interaction-state space, then projected onto the budget-filtered success-reachable region ℛx+, where the first projectable-to-non-projectable transition defines the CTB. (4–5) Conditioned on the retained student prefix and privileged teacher references, DART-SD generates a recovery continuation after the CTB and applies localized supervision only to the generated assistant tokens. (6) This CTB-guided SFT loop is progressively repeated across self-distillation rounds.
- Figure 3: Performance comparison of Qwen3-8B, DART-SD, and the teacher across five tool-use benchmarks. DART-SD improves upon Qwen3-8B on all benchmarks and surpasses the teacher on FTRL, ToolHop, and τ-bench.
Original abstract (English)
Equipping Large Language Models (LLMs) with multi-turn tool-calling capabilities is essential for building autonomous agents. However, progress is fundamentally limited by the reliance on full-length trajectory imitation. For tasks involving multiple order-independent sub-goals, the optimal solution space forms a vast combinatorial diamond lattice. Forcing this rich topology into monolithic trajectories causes a severe topological collapse, indiscriminately penalizing valid alternative explorations and severely degrading policy diversity. To address this, we propose DART-SD (Diamond-topology Aware Retrieval and Tuning for Self-Distillation), a novel framework that shifts the paradigm from global forcing to topology-guided localized correction. DART-SD first models the execution process as a converging Interaction-State Transition Graph (ISTG), faithfully capturing the inherent diamond topology of successful and failed exploratory paths. During autonomous rollouts, the framework identifies the Critical Topological Breakpoint (CTB) and retrieves success-supported recovery references. Finally, we introduce a progressive self-distillation paradigm through CTB-guided localized supervision, ensuring that the training loss is calculated exclusively on the generated recovery steps while strictly protecting the valid reasoning prefix from destructive gradient updates. Experiments on complex multi-turn tool-calling benchmarks demonstrate that DART-SD significantly outperforms traditional full-trajectory baselines.
Read on arXivLatest papers
- SWE-bench Science: Can Coding Agents Resolve Engineering Tasks in Science?AI coding agents were tested on fixing real scientific software, and even the best one failed more than half the time
- FlashPrefill V2: Block-Sparse Prefill Attention for Long-Context LLM ServingMaking sparse attention fast enough and accurate enough for real LLM serving, not just papers
- PolicyGuide: From Guarding One Action to Guiding the Whole Workflow for Policy-Compliant LLM AgentsMaking customer-service AI agents follow the whole procedure, not just avoid one bad action
- EXIMO: VLM Guided Exploration of VLA PoliciesTeaching a robot new chores without human teleoperation, by letting a chatty AI supervise it
- EnvHarness: Awakening Static Worlds for Agent LearningInstead of building new training worlds from scratch, this work adds a plug-in layer that reshapes existing ones around each agent's actual weaknesses
- Bounded Sovereignty and the Control Tax: Pricing AI Oversight When the Deployer Does Not Own the ModelCompanies that rent AI instead of owning it can only do half of AI safety oversight
- Beyond Imitation: Filtering On-Policy Distillation by Reasoning ProgressA fix for AI models that get penalized by their teacher even when they're reasoning correctly
- PersonalBench: Measuring the Authorship Gap in LLM PersonalizationAI can be prompted to write 'like someone,' but its own voice never fully disappears