在联邦学习中压缩大模型,兼顾隐私保护与性能
PPC-GPT: Federated Task-Specific Compression of Large Language Models via Pruning and Chain-of-Thought Distillation
- 通过差分隐私数据与思维链蒸馏生成合成数据
- 压缩后小模型性能接近全尺寸大模型
- 适合需要隐私保护的垂直领域应用
将大语言模型(LLMs)压缩为特定任务的小语言模型(SLMs)面临两大挑战:保护领域知识隐私和应对资源受限。为此,我们提出PPC-GPT,一个统一框架,在联邦设置下同时解决隐私保护与模型压缩问题。该框架采用服务器-客户端架构,客户端向服务器的LLM发送经过差分隐私(DP)扰动的任务特定数据,由LLM生成带推理过程的合成数据。这些合成数据用于后续的LLM剪枝与重训练。其核心创新在于将隐私保护机制、合成数据生成与任务特定压缩技术深度融合,各组件间产生协同增益。在多种文本生成任务上的实验表明,PPC-GPT成功实现双重目标:性能媲美全尺寸LLM,且通过联邦架构实现强隐私保护。代码已贡献至FATE开源项目,可公开获取于https://github.com/FederatedAI/FATE-LLM/tree/main/python/fate_llm/algo/ppc-gpt。
原文摘要 · Abstract (English)
Compressing Large Language Models (LLMs) into task-specific Small Language Models (SLMs) encounters two significant challenges: safeguarding domain-specific knowledge privacy and managing limited resources. To tackle these challenges, we propose PPC-GPT, a novel unified framework that systematically addresses both privacy preservation and model compression in federated settings. PPC-GPT works on a server-client federated architecture, where the client sends differentially private (DP) perturbed task-specific data to the server's LLM. The LLM then generates synthetic data along with their corresponding rationales. This synthetic data is subsequently used for both LLM pruning and retraining processes. Our framework's key innovation lies in its holistic integration of privacy-preserving mechanisms, synthetic data generation, and task-specific compression techniques, creating unique benefits through component interaction. Our experiments across diverse text generation tasks demonstrate that PPC-GPT successfully achieves dual objectives: maintaining competitive performance comparable to full-sized LLMs while ensuring robust privacy protection through its federated architecture. Our code has been contributed to the FATE open-source project and is now publicly accessible at \textit{https://github.com/FederatedAI/FATE-LLM/tree/main/python/fate_llm/algo/ppc-gpt}
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。