Everything K-culture — comebacks to K-beauty, straight to your inboxGet it in your inbox

METAL MEDIA

DART-SD: Diamond-topology Aware Retrieval and Tuning for Self-Distillation of Multi-Turn Tool-Calling Agents

arXiv:2608.185242026-08-20

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

  1. 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
  2. 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
  3. 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
  4. 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
  5. 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
An explanatory diagram made by METAL MEDIA, not a figure supplied by the paper's authors.

What they did

  1. 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
  2. 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
  3. 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
  4. 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
  5. 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
Table 1: Performance comparison of different training methods on five tool-use benchmarks using Qwen3-4B and Qwen3-8B backbones. Training-based baselines are organized into distillation-based ♠ and reinforcement learning ♢ paradigms. All trainable methods are trained on FTRL and evaluated on both the in-domain FTRL test set and four out-of-domain benchmarks. The best and second-best results within each backbone are highlighted.
MethodsFTRLBFCLToolHopτ-benchRoTBenchAvg.
Solve-PSolve-RSolve-F1Multi-TurnACPass^1TSPICF
Qwen3-4B
Base21.0026.5421.8110.1420.2015.1569.5226.3116.0725.19
♠ SFT34.2649.0237.9614.5740.5021.8271.6744.4024.4037.62
♠ SCoRe-SFT23.7928.2024.6112.7525.8311.5260.3631.1918.3326.29
♠ OPSD22.3729.2123.5011.0020.2023.0368.9335.2421.6728.35
♢ FTRL-GRPO36.8341.7137.8413.5029.2520.6170.3632.6220.2433.66
♢ ToolRL26.8933.2328.479.8820.6018.1869.6438.2124.0529.91
♢ MatchTIR (OT)23.5529.2724.9510.3826.2316.3672.0232.7420.3628.43
♢ MatchTIR (KM)25.5431.1026.5010.0026.6321.8272.0233.8121.4329.87
♠ DART-SD (Ours)36.7048.1639.7723.8842.1123.0372.0242.3824.5239.17
Qwen3-8B
Base21.1830.7123.4818.3828.5410.1375.5236.2922.1929.60
♠ SFT38.0850.9541.8919.2543.5226.0675.9548.8130.2441.64
♠ SCoRe-SFT31.2834.4831.5819.2530.2518.1870.3640.4824.5233.38
♠ OPSD24.3734.8826.6820.5041.1121.2175.2443.2127.3834.95
♢ FTRL-GRPO37.6645.4940.2235.2534.5723.0377.0242.7427.0240.33
♢ ToolRL32.4941.0735.0020.5044.7225.4575.7151.4333.1039.94
♢ MatchTIR (OT)29.4534.8930.6521.3838.7924.2475.0040.3625.3635.57
♢ MatchTIR (KM)33.0740.4235.3723.2541.7126.0675.7142.2626.9038.31
♠ DART-SD (Ours)42.0054.1345.6627.6345.0327.1275.8357.3835.4845.58
Table 2: Average tool-call length of successful trajectories across progressive SFT iterations on the FTRL test set. DART-SD progressively improves Solve-F1 while shortening its tool traces, eventually producing more efficient traces than the golden solutions given during data construction.
TaskDART-SDGolden
Iter1Iter2Iter3Iter4Iter5
Single1.231.151.101.091.071.00
Multi3.834.634.544.494.414.71
Para-Single3.292.652.272.262.222.11
Para-Multi7.256.485.975.825.626.97
Overall4.233.993.713.653.554.02
Solve-F140.3742.9543.7844.6745.66
Table 3: Average CTB positions of failed training trajectories across progressive SFT. Larger values indicate that the first departure from empirically recoverable behavior occurs later, meaning that the model correctly executes a longer trajectory prefix before localized recovery is required.
TaskCTB Position
Iter1Iter2Iter3Iter4Iter5
Single0.0340.1860.1370.2490.077
Multi0.5001.6241.7291.8761.953
Para-Single0.0950.2190.2990.3280.313
Para-Multi0.3951.5241.6921.7911.816
Overall0.3481.1851.3101.4211.452
Δ vs. Iter1+0.837+0.962+1.073+1.104
Table 4: Performance comparison under the thinking setting. Results are reported using FTRL Solve-F1, the average BFCL Multi-Turn score, and ToolHop AC.
MethodFTRLBFCLToolHop
Qwen3-8B29.7440.0042.21
FTRL-GRPO32.8541.5036.72
ToolRL26.7234.2532.93
MatchTIR (KM)37.3347.1346.16
DART-SD41.0349.7546.43
Table 5: General capability evaluation on representative benchmarks. The best results are highlighted.
MethodIFEvalAIME24AIME25MMLUAvg.
Qwen3-8B34.7546.6723.3370.9443.92
SFT35.3043.3326.6771.4344.18
DART-SD45.2950.0030.0074.2749.89
Table 6: Component ablation of DART-SD on the FTRL test set using Solve-P, Solve-R, and Solve-F1.
MethodSolve-PSolve-RSolve-F1
Qwen3-8B21.1830.7123.48
+SD36.2344.4638.10
+CTB36.6246.3239.51
+Progressive SFT41.3249.6543.93
+ISTG (Ours)42.0054.1345.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.
See the figures in the original paper →

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.

Authors · Hangrui Xu, Jiarui Wang, Yang Yang, Chuanbo Zhu, Fangda Chen, Ziqi Wu, Jingming Cai, Yan Song

Read on arXiv

Latest papers

All papers →

Latest from METAL MEDIA