Does On-Policy Distillation for Safety Pose Backdoor Risks?
cs.LG, cs.AI
Submitted: 2026-10-06
Updated: 2026-10-06
License: http://creativecommons.org/licenses/by/4.0/
The gist: On-policy distillation (OPD) has attracted growing attention as an effective way to transfer capabilities from teacher models to student models.
Terminology
Abstract
On-policy distillation (OPD) has attracted growing attention as an effective way to transfer capabilities from teacher models to student models. Recent studies further explore OPD as a tool for improving large language model safety with promising results. However, these approaches typically assume that the teacher and training data are trustworthy. In this paper, we uncover an overlooked threat to OPD for safety: a safety-aligned but backdoored teacher can propagate its hidden malicious behavior to an initially clean student. Under our threat model, a poisoning rate as low as 3% results in an attack success rate (ASR) of up to 70% on the distilled student. We further identify two training choices that can amplify this risk. First, increasing the number of training epochs can lead to high ASR even at low poisoning rates. With only 10 poisoned samples, ASR reaches 67% after 16 epochs. Second, the commonly used top-k KL can accelerate backdoor transfer, causing trigger-conditioned harmful behavior to emerge earlier than sampled-token KL in most settings. Alongside these findings, we explore a simple mitigation, Lazy Defense, which clips KL rewards to make student updates less aggressive, limiting aggressive updates and slowing backdoor learning. Experiments show that Lazy Defense delays backdoor transfer in low poisoning rate settings. Together, our findings reveal that OPD can propagate backdoors, highlighting the need to address the safety risks of OPD.
Sources
- Emergent and Subliminal Misalignment Through the Lens of Data-Mediated Transfer
- Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedback
- Transferring Backdoors between Large Language Models by Knowledge Distillation
- Pay Attention to the Triggers: Constructing Backdoors That Survive Distillation
- Reducing the Safety Tax in LLM Safety Alignment with On-Policy Self-Distillation
- Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixes
- Rethinking On-Policy Distillation of Large Language Models II: One Training Example
- On-Policy Distillation for LLM Safety: A Routing Approach to Template-Robust Realignment
- On-Policy Consistency Training Improves LLM Safety with Minimal Capability Degradation
- Sleeper Agents: Training Deceptive LLMs that Persist Through Safety Training
- bi-GRPO: Bidirectional Optimization for Jailbreak Backdoor Injection on LLMs
- SafeSteer: Localized On-Policy Distillation for Efficient Safety Alignment
- Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe
- INK: Inheritable Natural Backdoor Attack Against Model Distillation
- The Llama 3 Herd of Models
- It Takes Two: Complementary Self-Distillation for Contextual Integrity in LLMs
- Multilingual Safety Alignment via Self-Distillation
- Covert Influence Between Language Models
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks