Paper Reading AI Learner

DP-OPD: Differentially Private On-Policy Distillation for Language Models

2026-04-06 06:18:03
Fatemeh Khadem, Sajad Mousavi, Yi Fang, Yuhong Liu

Abstract

Large language models (LLMs) are increasingly adapted to proprietary and domain-specific corpora that contain sensitive information, creating a tension between formal privacy guarantees and efficient deployment through model compression. Differential privacy (DP), typically enforced via DP-SGD, provides record-level protection but often incurs substantial utility loss in autoregressive generation, where optimization noise can amplify exposure bias and compounding errors along long rollouts. Existing approaches to private distillation either apply DP-SGD to both teacher and student, worsening computation and the privacy--utility tradeoff, or rely on DP synthetic text generation from a DP-trained teacher, avoiding DP on the student at the cost of DP-optimizing a large teacher and introducing an offline generation pipeline. We propose \textbf{Differentially Private On-Policy Distillation (DP-OPD)}, a synthesis-free framework that enforces privacy solely through DP-SGD on the student while leveraging a frozen teacher to provide dense token-level targets on \emph{student-generated} trajectories. DP-OPD instantiates this idea via \emph{private generalized knowledge distillation} on continuation tokens. Under a strict privacy budget ($\varepsilon=2.0$), DP-OPD improves perplexity over DP fine-tuning and off-policy DP distillation, and outperforms synthesis-based DP distillation (Yelp: 44.15$\rightarrow$41.68; BigPatent: 32.43$\rightarrow$30.63), while substantially simplifying the training pipeline. In particular, \textbf{DP-OPD collapses private compression into a single DP student-training loop} by eliminating DP teacher training and offline synthetic text generation. Code will be released upon publication at this https URL.

Abstract (translated)

大型语言模型(LLMs)正日益适配于包含敏感信息的专有和领域特定语料库,这造成了形式化隐私保证与通过模型压缩实现高效部署之间的紧张关系。通常通过DP-SGD实施的差分隐私(DP)提供记录级保护,但在自回归生成中常导致显著的效用损失,因为优化噪声可能加剧曝光偏差并沿长序列传播累积错误。现有的私有蒸馏方法要么对教师和学生均应用DP-SGD,从而恶化计算负担与隐私-效用权衡;要么依赖从DP训练教师生成DP合成文本,虽避免对学生进行DP处理,却需以DP优化大型教师为代价并引入离线生成流程。我们提出\textbf{差分隐私同策略蒸馏(DP-OPD)},一个无合成框架,该框架仅通过学生的DP-SGD实施隐私保护,同时利用冻结的教师为学生生成轨迹上的密集词级目标提供指导。DP-OPD通过在续写词上实施\textit{私有广义知识蒸馏}来实现这一构想。在严格隐私预算($\varepsilon=2.0$)下,DP-OPD相较于DP微调和离策略DP蒸馏提升了困惑度表现,并超越了基于合成的DP蒸馏(Yelp:44.15$\rightarrow$41.68;BigPatent:32.43$\rightarrow$30.63),同时大幅简化了训练流程。特别地,\textbf{DP-OPD通过消除DP教师训练和离线合成文本生成,将私有压缩整合至单一DP学生训练循环中}。代码将在发表后通过此https URL发布。

URL

https://arxiv.org/abs/2604.04461

PDF

https://arxiv.org/pdf/2604.04461.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot