Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Tailoring Teaching to Aptitude: Direction-Adaptive Self-Distillation for LLM Reasoning

About

On-policy self-distillation (OPSD) is an emerging LLM post-training paradigm in which the model serves as its own teacher: conditioned on privileged information such as a reference trace or hint, the same policy provides dense token-level supervision on its own rollouts. However, recent studies show that OPSD degrades complex reasoning by suppressing predictive uncertainty, which supports exploration and hypothesis revision. Our token-level analysis shows that this failure arises from applying a uniform direction of teacher supervision across tokens with different uncertainty levels: conformity to the privileged self-teacher suppresses exploration at high entropy, while deviation from the teacher degrades step accuracy at low entropy. Accordingly, we propose \textbf{Direction-Adaptive Self-Distillation} (\textbf{DASD}), which reframes privileged self-distillation from uniform teacher imitation into entropy-routed directional supervision: high-entropy tokens are pushed away from the privileged teacher to preserve exploration, while low-entropy tokens are pulled toward the teacher to stabilize step-level execution. Across six mathematical reasoning benchmarks, DASD achieves the best macro Avg@16 over strong RLVR and self-distillation baselines. Pass@$k$, reasoning-health, and generalization analyses show that these average gains come from preserving exploration without sacrificing step-level execution.

Hongbin Zhang, Chaozheng Wang, Kehai Chen, Youcheng Pan, Yang Xiang, Jinpeng Wang, Min Zhang• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMinerva
Avg@1633
42
Mathematical ReasoningAIME 2025
pass@1683.3
32
Mathematical ReasoningAIME24
Pass@1671.7
30
Mathematical ReasoningHMMT25
Avg@1625
24
Mathematical ReasoningHMMT25
Avg@16 Accuracy25
24
Mathematical ReasoningAIME 24
Avg@16 Accuracy71.7
24
Mathematical ReasoningHMMT25
Pass@1653.3
24
Mathematical ReasoningOlympiadBench
Pass@1665.6
24
Mathematical ReasoningMathematical Reasoning Macro Average
Pass@16 (Avg)69
24
Mathematical ReasoningMath Family Average
Accuracy67
12
Showing 10 of 17 rows

Other info

Follow for update