Alibaba I-SDPO: distill only all-incorrect GRPO groups, SciKnowEval 56.67% to 70.31%

I-SDPO: Instance-Level Adaptive Self-Distillation Policy Optimization

Yubo Zhang, Xinhong Ma, Zezhong Tan, Ziqiang Dong

cs.LG, cs.CL

2026-08-13

I-SDPO distills only when every rollout sample fails and keeps GRPO otherwise, raising Qwen3-8B SciKnowEval mean@16 accuracy from 56.67% to 70.31%.

What problem this solves

GRPO has become the default RL algorithm for LLM reasoning training: sample a group of responses per prompt, subtract group-relative advantages, skip the critic. Its blind spot is the all-incorrect group. On a hard problem where all 16 samples are wrong, rewards are nearly identical, advantages collapse toward zero, and the gradient vanishes. The paper quantifies this: at a per-sample success rate of 0.1 with K=16, roughly 18.5% of groups produce no relative signal at all. The weaker the policy and the harder the problem, the less signal GRPO gets, exactly where learning is most needed.

Privileged self-distillation fills the gap: a teacher that sees the ground-truth solution assigns token-level soft targets along the student's own trajectories. The signal is dense and low-variance, but the teacher is biased. It is an EMA copy of the student and inherits the student's systematic errors; it favors one demonstrated solution style while the reward may accept many valid paths. Distilling throughout training keeps pulling toward a teacher-specific distribution after reward-bearing trajectories become available.

The tension is that distillation helps early and hurts late, and "early or late" is not a global clock. It varies per prompt: an easy prompt may yield a success in the first batch, a hard one may stay all-incorrect until the end of training.

Method

I-SDPO lets each rollout group vote on whether to trust the teacher. One routing decision per instance, shared across the whole group:

The teacher is not an external model but an EMA copy of the student (update rate 0.05, mean lag of about 19 updates). Its privilege is in the input: the ground-truth solution is prepended to the prompt before it scores the student's generated tokens position by position. The distillation loss interpolates forward and reverse KL at α=0.5, with entropy-aware token weighting: positions where the teacher distribution has low entropy (high confidence) get larger weights. The authors are explicit that this is a confidence proxy only and cannot fix a confidently wrong teacher.

Two theoretical results back the design. Proposition 1: when the teacher target sits a bias b away from the reward optimum, a non-vanishing distillation weight λ parks the parameters at λ/(1+λ)·b from the optimum, with excess reward loss about ½[λ/(1+λ)]²‖b‖². Persistent distillation has a bias floor. Proposition 2: if the per-sample success probability is non-decreasing and the K rollouts are conditionally independent, the all-incorrect probability (1−pt)^K is non-increasing. Routing drives the effective λ toward zero as capability grows; the teacher withdraws without a hand-designed schedule.

This self-annealing is per prompt: at the same training step, easy prompts have long returned to GRPO while hard ones are still distilling. The sample-level alternative (SRPO) routes each incorrect sample individually, which dismantles the mixed group's contrast: in a 15-wrong-1-right group, SRPO converts 15 pieces of negative evidence into imitation targets.

Results

SciKnowEval, four scientific domains, Qwen3-8B, 2 epochs, mean@16 accuracy:

MethodBiologyMaterialChemistryPhysicsAvg
GRPO32.1270.7462.9260.8856.67
SDPO (pure distillation)45.9371.4176.6468.9865.74
SRPO (sample-level routing)44.2772.9478.6968.1266.01
I-SDPO50.2574.5381.1675.3170.31

Training dynamics match the self-annealing account: on biology, the GRPO-routed fraction rises from about 0.65 at step 50 to about 0.90 at step 350, while the all-wrong fraction falls from 0.22 to 0.03. In ablation, the balanced α=0.5 KL mix wins in all four domains, averaging 68.16% versus 60.39% for forward KL alone and 61.63% for reverse KL alone.

Why it matters

All-incorrect groups are not a cold-start-only problem. A harder benchmark, a smaller base model, or a stricter reward checker pushes their share right up, and GRPO spins its wheels in those settings. I-SDPO offers a patch with almost no extra hyperparameters: the routing criterion is "does the group contain one correct answer", binary and observable, no gradient-alignment estimation needed; the teacher is a free EMA copy with no inference-side deployment cost. It replaces a hand-tuned distillation schedule with the (1−p)^K curve, which adapts per prompt.

For anyone doing RL post-training, this is the "distillation and RL each cover a phase" intuition made mechanical: switch on capability evidence per prompt, not on training step. The SRPO comparison shows granularity matters, and that negative samples in a mixed group are valuable evidence, not imitation material.

For calibration on where the gain comes from: pure distillation already captures 9 of the 13 points. I-SDPO's contribution is extending that to 13.6 while avoiding the late-stage drag, so the novelty is in the scheduling, not the distillation.

Limitations

The authors list two: only Qwen3-8B was tested, and only SciKnowEval. Stronger base policies produce fewer all-incorrect groups, so the patching value shrinks; math or open-ended reasoning tasks have different reward structure and solution multiplicity.

Further concerns from reading the paper:

Terms

Source

What people are saying

Related papers

All paper explainers