Learning from Self-Generated Responses: Reliability--Uncertainty Self-Distillation for Reasoning
cs.AI
Submitted: 2026-10-13
Updated: 2026-10-13
Code: https://github.com/JP-25/RUSD
License: http://creativecommons.org/licenses/by/4.0/
The gist: Reference solutions and question-specific rubrics are not always available to guide the teacher in self-distillation.
Terminology
Abstract
Reference solutions and question-specific rubrics are not always available to guide the teacher in self-distillation. A large language model's own solution attempts offer an alternative, but their quality is difficult to assess in open-ended tasks without automatic correctness checks, and even correct attempts can involve uncertain token predictions. We propose Reliability--Uncertainty Self-Distillation to learn from these attempts. The same model serves as teacher and student: the teacher receives the question and completed attempt, and the student matches the teacher's token-level predictions using a question-only prompt. weights each attempt's distillation loss by additively combining judged reliability, estimated from a self-judge's score and confidence, with student predictive uncertainty. Across eleven reasoning benchmarks, outperforms all five baselines in average accuracy, reaching 70.47% on Qwen3-4B and 76.69% on Qwen3-8B (8.90 and 2.10 percentage points above the base models). Data and code are available at https://github.com/JP-25/RUSD.
Sources
- A General Language Assistant as a Laboratory for Alignment
- H squared SD: Hybrid Hindsight Self-Distillation
- Training Verifiers to Solve Math Word Problems
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- PAST: Privileged Adaptation from Complete Student Trajectories for On-Policy Self-Distillation
- Consensus as Privileged Context for Label-Free Self-Distillation
- Reinforced Self-Training (ReST) for Language Modeling
- Learning from Consensus and Disagreement: Unsupervised On-Policy Self-Distillation with Minority-Trajectory Contrast
- Self-Distillation Zero: Self-Revision Turns Binary Rewards into Dense Supervision
- Distilling the Knowledge in a Neural Network
- Overcoming Scaling Limits in On-Policy Self-Distillation for LLM Reasoning
- When Clipping Reverses Correction: Failure Dynamics of Pointwise Forward-KL On-Policy Self-Distillation
- Privileged Solutions or Context-Induced Teacher Behavior? Dissecting On-Policy Self-Distillation
- Respecting Self-Uncertainty in On-Policy Self-Distillation for Efficient LLM Reasoning
- On-Policy Self-Distillation without Any Supervision
- When Are Teacher Tokens Reliable? Position-Weighted On-Policy Self-Distillation for Reasoning
- Activation-Conditioned Self-Distillation
- On-Policy Self-Distillation with Sampled Demonstrations Reduces Output Diversity
- Negative Self-Distillation: Learning to Reason by Avoiding Flaws
- Maximizing Confidence Alone Improves Reasoning
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems