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

Rethinking Reward Supervision: Rubric-Conditioned Self-Distillation

About

Post-training of reasoning language models is commonly driven by supervised distillation and reinforcement learning with verifiable rewards. Distillation often relies on chain-of-thought annotations that are expensive to obtain and may themselves be noisy, incomplete, or partially incorrect; even when the final solution is correct, an imperfect rationale can interfere with learning. Reinforcement learning with verified rewards, on the other hand, typically compresses evaluative feedback into a scalar signal, obscuring which aspects of a response should be improved. We propose \textbf{Rubric-Conditioned Self-Distillation}, a framework that incorporates rubrics as structured, fine-grained feedback for on-policy self-distillation. Our method conditions the teacher model on criterion-level rubrics and uses it to provide token-level guidance on the student's own sampled trajectories. This design avoids treating a single reference rationale as the sole supervision target. Instead, rubrics specify what a strong response should satisfy, enabling more fine-grained credit assignment over the reasoning process than scalar reward optimization. We instantiate this framework with a two-stage pipeline that first learns to generate task-specific rubrics and then trains a rubric-guided reasoner. We evaluate on a diverse suite of science reasoning benchmarks and results show that rubric-conditioned self-distillation effectively converts rubric-level criteria into token-level guidance over the reasoning process, surpassing GRPO by 1.0 points and OPSD by 0.9 points on average.

Siyi Gu, Jialin Chen, Sophia Zhou, Arman Cohan, Rex Ying• 2026

Related benchmarks

TaskDatasetResultRank
Medical Question AnsweringMedMCQA
Accuracy65.8
591
Medical Question AnsweringPubMedQA
Accuracy75.1
122
Scientific ReasoningGPQA Diamond (test)
Accuracy64.5
88
Scientific ReasoningSCIBENCH (test)
Accuracy70.8
6
Scientific ReasoningPIQA (test)
Accuracy90.8
6
Scientific ReasoningResearchQA (test)
Accuracy73.1
6
Scientific ReasoningRUBRICHUB (test)
Accuracy55.7
6
Scientific ReasoningRAR-SCIENCE (test)
Accuracy68.6
6
Showing 8 of 8 rows

Other info

Follow for update