SODA: Semi On-Policy Black-Box Distillation for Large Language Models
Organizations: Clemson University · LinkedIn · Washington University in St. Louis · Arizona State University ·
Abstract
Black-box knowledge distillation for large language models presents a strict trade-off. Simple off-policy methods (e.g., sequence-level knowledge distillation) struggle to correct the student's inherent errors. Fully on-policy methods (e.g., Generative Adversarial Distillation) solve this via adversarial training but introduce well-known training instability and crippling computational overhead. To address this dilemma, we propose SODA (Semi On-policy Distillation with Alignment), a highly efficient alternative motivated by the inherent capability gap between frontier teachers and much smaller base models. Because a compact student model's natural, zero-shot responses are almost strictly inferior to the powerful teacher's targets, we can construct a highly effective contrastive signal simply by pairing the teacher's optimal response with a one-time static snapshot of the student's outputs. This demonstrates that exposing the small student to its own static inferior behaviors is sufficient for high-quality distribution alignment, eliminating the need for costly dynamic rollouts and fragile adversarial balancing. Extensive evaluations across four compact Qwen2.5 and Llama-3 models validate this semi on-policy paradigm. SODA matches or outperforms the state-of-the-art methods on 15 out of 16 benchmark results. More importantly, it achieves this superior distillation quality while training 10 times faster, consuming 27% less peak GPU memory, and completely eliminating adversarial instability.
Figures & tables
| Model | Method | In-Dist. | Out-of-Distribution | ||
|---|---|---|---|---|---|
| LMSYS | Dolly | SelfInst | Vicuna | ||
| GPT-5-Chat | Teacher | 51.7 | 49.8 | 49.7 | 49.9 |
| Qwen2.5-3B-Instruct | Base | 45.8 | 45.1 | 45.6 | 47.3 |
| SeqKD | 47.5 | 44.8 | 45.7 | 48.0 | |
| GAD | 48.9 | 46.7 | 47.7 | 49.4 | |
| SODA | 49.2 | 46.1 | 48.2 | 49.8 | |
| Model | SODA (Static ) | Resample @ Step 100 | Resample @ Step 200 |
|---|---|---|---|
| Qwen2.5-3B | 49.2 | 48.5 | 47.8 |
| Llama-3.2-3B | 49.1 | 48.1 | 47.4 |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
Language Model Distillation
Language model distillation transfers a larger model’s capabilities to smaller or specialized models using outputs, probabilities, or internal representations.
See all 508 papers