让大模型推理更准:根据错误演化动态调整监督强度。
DASH: Divergence-Adaptive Supervision Horizons for On-Policy Self-Distillation of Reasoning Models

- 根据局部误差的历史演变,动态调整每步的监督权重。
- 在三个数学推理基准上,所有模型规模下均超越基线方法。
- 无需额外计算,复用已有教师-学生分布即可提升效果。
基于可验证奖励的强化学习(RLVR)利用自动验证的结果信号提升大语言模型的推理能力,但这类信号通常稀疏且仅在序列末端提供。为缓解信号稀疏性,当前的在线策略自蒸馏(OPSD)通过在学生访问的前缀处查询特权教师,提供密集的词元级分布监督。然而,标准OPSD对每个局部差异统一赋予相同系数,忽视了生成过程中的时序结构。同一误差幅度可能对应不同的偏差演化路径,而单一标量无法区分这些上下文。为此,本文提出偏差自适应监督时域(DASH),将每一步的局部蒸馏信号与序列均值的差距映射为可调节的传播门控,并用其控制反向多步聚合。该机制使监督权重随生成过程中局部偏差的演化动态调整。在三个数学推理基准上,跨三种模型规模的实验表明,DASH在所有设置下均优于匹配的基线版本。DASH复用OPSD已有的教师与学生分布,增益无需额外前向计算。
原文摘要 · Abstract (English)
Reinforcement learning with verifiable rewards (RLVR) improves the reasoning capabilities of large language models using automatically verifiable outcome signals, but these signals are typically sparse and at the sequence-level. On-policy self-distillation (OPSD) mitigates this sparsity by querying a privileged teacher at student-visited prefixes and providing dense token-level distributional supervision. Although this dense supervision alleviates signal sparsity, we find that standard OPSD still underexploits the temporal structure of the rollout. It assigns every local divergence the same coefficient, regardless of its position or the divergence sequence in which it occurs. In on-policy autoregressive generation, the same divergence magnitude can follow different discrepancy histories, reflecting different evolutions of the mismatch between the teacher and student. Since the local scalar alone cannot distinguish these temporal contexts, standard OPSD cannot adapt its token-level weights to the realized discrepancy sequence. To address this limitation, we propose Divergence-Adaptive Supervision Horizons (DASH). DASH maps the gap between each local distillation signal and the sequence-level mean to an adaptive propagation gate and then uses these gates to control backward multi-step aggregation. By doing so, DASH adjusts token-level supervision weights according to how local divergences evolve during generation. Experiments on three mathematical reasoning benchmarks across three model scales show that DASH improves over our matched vanilla OPSD reruns on every benchmark at all three scales. DASH reuses the teacher and student distributions that OPSD already computes, so the gains require no additional teacher or student forward pass. Code: https://github.com/DBtxy/DASH-OPSD
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。