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

Finding the Evidence: Discovering Decision-Supporting Tokens for On-Policy Reasoning Distillation

About

On-policy distillation transfers reasoning ability through dense token-level supervision, yet the nature of the transferable signal remains unclear. We discover that reasoning chains contain two types of knowledge that require different discovery mechanisms: decisions (where to branch), which surface through student uncertainty, and evidence (intermediate steps that justify decisions), which hides in positions where the student is confident yet wrong. Current methods capture only decisions; the substantive knowledge in evidence tokens remains untransferred. We propose DEAR(Decision-Evidence Aware Reasoning Distillation), which first identifies decisions via student entropy, then discovers their supporting evidence through hidden-state cosine similarity to decision anchors, boosted by teacher-student divergence to prioritize the largest knowledge gaps. Across three student-teacher configurations on math and code benchmarks, DEAR consistently outperforms standard OPD, with up to +2.5pp on competition math and +5.7pp on code generation.

Jinwei Xiao, Zhuowen Han, Yueqing Sun, Zhengxi Lu, Yuxin Liu, Zhiyuan Yao, Wentao Chen, Qi Gu, Xunliang Cai• 2026

Related benchmarks

TaskDatasetResultRank
Code GenerationHumanEval (test)--
701
Code GenerationMBPP Plus (test)
Accuracy57.05
93
Code GenerationAPPS (test)--
36
Mathematical ReasoningMinerva
Mean@828.68
25
Mathematical ReasoningMATH 500
Avg@8 Accuracy84.35
12
Mathematical ReasoningAMC23
Avg@8 Accuracy71.25
12
Mathematical ReasoningOlympiad
Avg@8 Accuracy50.1
12
Mathematical ReasoningAIME24
Avg@8 Accuracy34.18
12
Showing 8 of 8 rows

Other info

Follow for update