Skip to main content
Search

# Self-Distillation

12 related articles

Subscribe to updates
  1. arXiv

    Agent RL stops scoring whole trajectories: ORSD writes its own rubrics to assign credit token by token

    Online Rubric Self-Distillation (ORSD) has the current policy generate natural-language process rubrics from differences among a group of its own trajectories, lets a frozen verifier pick the bundle whose rubric satisfaction best correlates with task success, and uses that bundle as privileged context: the log-likelihood gap for the same tokens with and without the rubric becomes a token-level correction to the trajectory-level GRPO advantage. On ALFWorld and WebShop, Qwen3-1.7B and Qwen3-4B reach the best overall results among the compared methods under the same environment-interaction budget.

  2. arXiv

    Masked diffusion models lock answers too early: HiLo-RD distills large-window trajectories, lifting GSM8K accuracy by 9.4 points

    Masked diffusion language models can fix answer fragments before the supporting reasoning is resolved when the response window is small, and those fragments then become context for later predictions. HiLo-RD uses the same model's large-window trajectory as a frozen teacher and aligns its hidden representations to cropped small-window views at committed positions. At the same small inference budget, GSM8K accuracy rises from 68.84% to 78.24% while mean readable length falls from 362.8 to 271.3 tokens, and the gains extend to code generation without code-domain training.

  3. arXiv

    Epilepsy detection across patients with different electrode layouts: MOSAIC infers channel graphs and steadies them by self-distillation

    Intracranial EEG electrode count and placement differ from patient to patient, so detectors with a fixed channel topology generalize poorly under subject-independent evaluation. MOSAIC encodes each channel with a shared convolutional encoder, then a single channel-attention layer infers a weighted inter-channel adjacency from the encoded features without any preset montage graph; an exponential moving average self-distillation term pulls each window's adjacency toward a slowly updated per-patient teacher. Across four public iEEG corpora, MOSAIC's F2 on both the SWEC-ETHZ short-term and long-term corpora exceeds the listed baselines, by 1.8 points over the strongest baseline on the short-term corpus and 3.3 points on the long-term one.

  4. arXiv

    Multi-token prediction carries a noisy gradient term: stripping it lifts a 1.2B model past next-token prediction on zero-shot accuracy

    Multi-token prediction (MTP), which trains a language model to predict several future tokens, is now a common pre-training ingredient, but its standard cross-entropy gradient contains a low signal-to-noise residual that grows noisier with scale. Soft-MTP removes that residual and backpropagates only the KL self-distillation signal hidden inside it, reusing cross-entropy gradients already computed and materializing no vocabulary-sized logit matrices, as a one-line code change. On LLaMA-style models from 167M to 1.2B parameters, Soft-MTP beats standard MTP at every scale with lower next-token perplexity, faster self-speculative decoding and higher zero-shot accuracy; from 689M it also lowers MTP perplexity, and at 1.2B it gives stronger math and code reasoning after fine-tuning.

  5. arXiv

    Self-distillation's reasoning gains don't survive new problems: coding help stays inside the training set, and gradients often point away from policy gradients

    Two self-distillation methods lose or reverse their reported gains on unseen problems across code, science, tool use and math, and the math improvement comes mainly from transferring thinking-mode behavior rather than from privileged information such as reference solutions.

  6. arXiv

    Agents stop waiting for an external teacher: ClawLoop has them write tasks, diagnose failures, and retrain in the same sandbox

    ClawLoop closes task synthesis, training, and failure diagnosis into one loop so harness-based agents improve without an external teacher model, beating the one-shot ClawGym pipeline on SkillsBench, Claw-Eval-Live, and ClawGym-Bench.

  7. arXiv

    UltraWorld self-distills an interactive ultrasound world model from untracked clinical videos, cutting terminal distance and orientation error by 29% and 38% versus visual servoing across nine simulated closed-loop planning episodes

    UltraWorld presents a self-distillation recipe that first adapts clinical ultrasound videos into a generator conditioned on reference images and anatomical masks, then synthesizes action-video pairs by sampling masks along programmable trajectories through 3D anatomy, distilling the generator into a world model that predicts future observations from local observations and actions alone, and introduces the Acoustic Sampling Map representing probe poses and imaging settings as pixel-wise 3D sampling positions, beam directions, and depths; experiments show improved prediction fidelity and action following, with mean terminal distance and orientation error reduced by 29% and 38% versus visual servoing across nine simulated closed-loop local planning episodes.

  8. arXiv

    Researchers introduce mathematical primitives and a four-dimension benchmark, identify Discovery as the dominant bottleneck in LLM mathematical reasoning, and improve reasoning with a primitive-privileged self-distillation framework

    The work introduces the notion of Mathematical Primitive and a benchmark that evaluates mathematical reasoning along four dimensions—Discovery, Generation, Digestion, and Execution—and a systematic diagnosis shows that solution accuracy masks distinct capability profiles, that primitives unlock substantial latent execution capacity, and that Discovery is the dominant bottleneck; building on the finding that discovery-limited failures are particularly amenable to repair, it proposes a primitive-privileged self-distillation framework that selectively transfers primitive-guided reasoning into the student model and consistently improves mathematical reasoning over baselines across model scales and challenging benchmarks.

  9. arXiv

    Forward KL is proved to inflate student entropy above the teacher, making divergence an implicit entropy regularizer in distillation

    Taking an entropic perspective on knowledge distillation, the work proves that forward KL inflates the student's entropy above the teacher's, with cross-entropy training as a special case yielding an identity verified quantitatively in pretraining and supervised finetuning; reverse KL deflates entropy until the student-teacher gap grows too large, interpolating between the two changes entropy smoothly early in training but abruptly at convergence; the lower entropy of on-policy distillation comes from token-level reverse KL rather than on-policy sampling, so the divergence acts as an implicit entropy regularizer, most clearly in self-distillation where the best divergence hyperparameters compensate for the entropy deflation caused by conditioning on privileged information.

  10. arXiv

    Pivot-SD supervises only high-impact commitments in denoising, letting LLaDA-8B-Instruct beat full-sequence SFT and budget-matched diffusion RL baselines on math and code with 200 questions

    The work introduces Pivot-SD, an offline self-distillation framework that selects high-impact commitments (pivots) during denoising via an information-gain metric measuring uncertainty reduction over remaining masked positions, trains pivots from successful trajectories with cross-entropy and pivots from failed trajectories with targeted unlikelihood while leaving the rest of the failed trajectory untouched, and with only 200 questions and four rollouts each improves LLaDA-8B-Instruct over full-sequence SFT and budget-matched diffusion RL baselines across math and code benchmarks.