Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Rubric-based On-policy Distillation

About

On-policy distillation (OPD) is a powerful paradigm for model alignment, yet its reliance on teacher logits restricts its application to white-box scenarios. We contend that structured semantic rubrics can serve as a scalable alternative to teacher logits, enabling OPD using only teacher-generated responses. To prove it, we introduce ROPD, a simple yet foundational framework for rubric-based OPD. Specifically, ROPD induces prompt-specific rubrics from teacher-student contrasts, and then utilizes these rubrics to score the student rollouts for on-policy optimization. Empirically, ROPD outperforms the advanced logit-based OPD methods across most scenarios, and achieving up to a 10x gain in sample efficiency. These results position rubric-based OPD as a flexible, black-box-compatible alternative to the prevailing logit-based OPD, offering a simple yet strong baseline for scalable distillation across proprietary and open-source LLMs. Code is available at https://github.com/Peregrine123/ROPD_official.

Junfeng Fang, Zhepei Hong, Mao Zheng, Mingyang Song, Gengsheng Li, Houcheng Jiang, Dan Zhang, Haiyun Guo, Xiang Wang, Tat-Seng Chua• 2026

Related benchmarks

TaskDatasetResultRank
Instruction FollowingIFEval--
836
Mathematical ReasoningAIME 25
Pass@1 Accuracy68.75
178
Mathematical ReasoningAIME 24
Pass@1 Accuracy63.33
103
Scientific ReasoningGPQA Diamond
Pass@1 Accuracy55.05
67
Mathematical ReasoningHMMT Feb25
Pass@139.16
27
Mathematical ReasoningHMMT25 Nov.
Pass@154.17
17
Medical KnowledgeHealthBench
Pass@186.87
11
Mathematical ReasoningAverage 4 Math Benchmarks
Pass@1 Score45.87
6
Showing 8 of 8 rows

Other info

GitHub

Follow for update