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

Restoring the Sweet Spot: Pass-Rate Weighted Self-Distillation for LLM Reasoning

About

Self-Distillation Policy Optimization (SDPO) provides dense token-level credit assignment for reinforcement learning with large language models by leveraging the model's own feedback-conditioned predictions as a self-teacher. Unlike GRPO, however, whose group-relative advantage naturally concentrates learning on a sweet spot of intermediate-difficulty questions, SDPO's KL-based advantage lacks an implicit notion of difficulty awareness. We analyze this gap through the lens of GRPO's advantage normalization. Extending the learnability framework to normalized rewards, we show that normalization absorbs the variance term $p(1-p)$, equalizing leading-order learnability across questions and leaving $\sqrt{p(1-p)}$ as the sole residual scaling factor in the per-question gradient. This analysis yields a simple prescription: weight each question's SDPO loss by $[\hat{p}(1-\hat{p})]^{1/2}$, resulting in SC-SDPO, a scale-consistent variant of SDPO. The proposed weights are obtained as a zero-cost byproduct of on-policy rollouts with batch-adaptive normalization, inducing an implicit curriculum that dynamically tracks the model's evolving competence. Experiments on scientific reasoning and tool-use benchmarks demonstrate that SC-SDPO consistently improves over SDPO, yielding gains of +3.2/+4.3 (mean@16/maj@16) on Qwen3-8B and +1.8/+3.0 on OLMo-3-7B, while preserving stable training dynamics throughout optimization.

Zehao Liu, Yuanpu Cao, Jinghui Chen, Vasant G. Honavar• 2026

Related benchmarks

TaskDatasetResultRank
Scientific ReasoningBiology
Avg@1665.4
24
ReasoningReasoning Benchmarks Biology, Chemistry, Material, Physics, Tool Use
Biology Accuracy55.9
11
Scientific ReasoningPhysics
Accuracy81.6
10
Scientific ReasoningMaterial
Accuracy79.3
10
Scientific ReasoningChemistry
Accuracy80.6
10
Science Question AnsweringChemistry
Mean@1680.6
9
Science Question AnsweringPhysics
Mean@1681.6
9
Science Question AnsweringBiology (Bio.)
Mean@1665.4
9
Science Question AnsweringMaterials (Mat.)
Mean@1679.3
9
Tool UseTool use
Accuracy0.673
8
Showing 10 of 14 rows

Other info

Follow for update