On-Policy Self-Distillation without Any Supervision

Yijiang Li, Bingyang Wang, Yijun Liang, Yunjie Tian, Di Fu, Nuno Vasconcelos

arXiv:2608.06296 · 2026-08-12 공개 · arXiv · PDF

large-language-models on-policy-distillation mathematical-reasoning qwen3 unsupervised-learning self-consistency model-self-correction pseudo-solution

Abstract

On-policy (Self-)Distillation (OPD / OPSD) has shown strong potential for post-training large language models (LLMs). However, existing methods still rely heavily on external supervision, including ground-truth signals, environmental feedback, or guidance from larger models, and therefore fall short of genuine "self"-distillation. In this study, we show that on-policy self-distillation can be achieved using only a model's own generations via internal consistency. We propose unsupervised on-policy self-distillation (U-OPSD). U-OPSD first samples multiple rollouts and constructs a pseudo solution by majority vote under a self-consistency threshold. It then conditions the model's distribution on the pseudo-solution and distills itself on the disagreeing completions, allowing the model to correct itself precisely where it is confidently wrong. Across diverse benchmarks, base models, and training settings, U-OPSD consistently improves over the base models and matches or surpasses supervised methods with ground truth (GT) such as OPSD and GRPO. On five mathematical reasoning benchmarks, i.e., AIME24, AIME25, HMMT25, MATH500, and AMC23, U-OPSD improves over the base model by 8.5% and 10.7% on Qwen3 non-thinking mode at 4B and 8B scales, and outperforms OPSD by 3.2% and 2.3% on average, respectively. In thinking mode, U-OPSD stays on par with OPSD, ahead by 0.9% at 4B and level at 8B and surpassing GRPO by 0.7% and 1.1%, respectively. Code is available at [https://github.com/williamium3000/u-opsd](https://github.com/williamium3000/u-opsd).

한국어 요약

한 줄 요약

U-OPSD는 외부 지도 없이 모델 자체의 생성물만으로 이루어지는 비지도 온-폴리시 자기-디스틸레이션 방법이다.

핵심 기여도

핵심 아이디어

기존 온-폴리시 디스틸레이션(OPD/OPSD)는 외부 지도(예: ground-truth, 강력한 teacher 모델)에 의존하여 ‘진정한’ 자기-디스틸레이션이 아니었다. 본 연구는 모델이 자체적으로 생성한 여러 롤아웃을 기반으로 내부 일관성을 활용하여 투표를 통해 의사결정을 생성하고, 이 투표 결과를 기반으로 모델 스스로 교정할 수 있음을 제시한다. 이는 모델이 자신이 확신하는 오류를 스스로 수정할 수 있는 구조를 만든다. 핵심 아이디어는 다음과 같다:

1. **다중 롤아웃 샘플링**: 문제 x에 대해 G개의 독립 롤아웃을 샘플링한다.
2. **투표 기반 의사결정 생성**: 각 롤아웃의 최종 답변을 추출하고, 다수결로 의사결정을 생성한다.
3. **일관성 기준 적용**: 투표 비율이 임계값 τ 이상일 경우, 가장 긴 일치 롤아웃을 투사된 ‘가짜 해법’으로 사용하고, 일치하지 않는 롤아웃을 학습 데이터로 활용한다.
4. **자기-디스틸레이션**: 투사된 해법을 조건으로 하여, 일치하지 않는 롤아웃의 prefix에 대해 KL-divergence 기반의 토큰 수준 디스틸레이션을 수행한다.

기술적 접근법

주요 결과

의의 및 한계

U-OPSD는 외부 지도 없이도 모델 스스로 자기 교정을 수행할 수 있음을 보여주며, 기존 지도 기반 온-폴리시 디스틸레이션과 동등 또는 우수한 성능을 달성한다. 특히, 투표 기반 의사결정과 내부 일관성 기준을 통해 모델의 오류를 효과적으로 식별하고 교정할 수 있다는 점에서 학술적·실용적 의의가 있다.

그러나, U-OPSD는 투표 기반 의사결정이 일정 수준의 일관성을 요구하므로, 모델이 생성한 롤아웃이 모두 불일치할 경우 학습이 어려울 수 있다. 또한, 투표 기준인 τ가 너무 낮거나 높을 경우, 잘못된 의사결정이 학습에 반영될 수 있다. 따라서 τ의 적절한 설정이 필요하며, 이는 모델과 문제에 따라 달라질 수 있다.

실용적 활용

U-OPSD는 외부 지도가 제한된 환경(예: 비정형 데이터, 비용이 높은 라벨링)에서 모델의 자기-개선을 가능하게 하므로, 대규모 언어 모델의 비지도 학습 및 자가 진단 시스템 구