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

Constitutional On-Policy Safe Distillation

About

On-policy self-distillation (OPSD) has emerged as an efficient post-training paradigm by using a teacher conditioned on privileged information to provide dense token-level supervision. Prior work has shown that OPSD can collapse in verifiable reasoning tasks, but safety alignment differs in that it is guided by high-level constitutions rather than explicit target answers, making it a natural setting to revisit dense distillation. However, our pilot study show that safety OPSD still suffers from severe collapse: constitutional conditioning contracts the teacher distribution toward short and overly conservative responses, and Reverse KL further amplifies this contraction into reduced expressiveness. We formalize this effect as geometric leakage under safety boundaries in a non-orthogonal semantic space, where safety pressure transfers into the expressiveness dimension. Based on this analysis, we propose Constitutional On-Policy Safe Distillation (COPSD), which first calibrates the teacher through a Cross-SFT cold-start and then performs constitution-conditioned on-policy distillation. Experiments on 12 benchmarks show that COPSD achieves a consistently stronger safety--helpfulness trade-off than baselines while substantially reducing the safety tax on general reasoning ability.

Ming Wen, Yuxuan Liu, Kun Yang, Yunhao Feng, Zhuoer Xu, Yuhao Sun, Shiwen Cui, Xiang Zheng, Guoyu Wang, Xingjun Ma, Yu-Gang Jiang• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMathVista
Score72.7
566
Multimodal Capability EvaluationMM-Vet
Score70.09
429
Visual Question AnsweringVQA v2
Accuracy80.67
257
Safety & Helpfulness EvaluationSPA-VL
Safety Score87.17
32
Safety & Helpfulness EvaluationVLGuard
Safety Score86.3
32
Safety AlignmentBeaverTails V
Safety Score90.49
27
Complex ReasoningLLaVA-Bench Wilder
Score74.1
14
Safety Alignment EvaluationSIUO
Safety Score81.44
14
Safety Alignment EvaluationVLSBench
Safety Score99.51
14
Safety Alignment EvaluationMSS-Bench
Safety Score83.58
14
Showing 10 of 10 rows

Other info

Follow for update