让大模型软提示私密迁移,降低算力与数据泄露风险
Efficient and Privacy-Preserving Soft Prompt Transfer for LLMs
- 用小模型蒸馏大模型,提升软提示迁移能力
- 本地微调+差分隐私,保护用户数据不外泄
- 仅需少量公开数据即可完成高效迁移,适合云服务场景
提示工程已成为适配大语言模型的主流方法。尽管离散(文本)提示因可解释性强而广泛应用,但参数化(软)提示在API中更受青睐——因其能整合更多训练样本信息,同时减少用户输入占用的上下文空间。然而,软提示与特定大模型强绑定,难以泛化至其他模型,这在效率和隐私上带来挑战:(1) 在每个模型上单独调优软提示计算开销巨大;(2) 当模型由外部托管时,软提示调优常需向服务商共享私有数据(如NVIDIA NeMo API)。为此,我们提出POST框架,实现私密的软提示迁移:先从大模型蒸馏出小模型以增强提示可迁移性,本地调优软提示(支持差分隐私),再通过少量公开数据将提示转移回大模型。实验表明,POST显著降低计算成本,保障隐私,并有效传递高价值软提示。
原文摘要 · Abstract (English)
Prompting has become a dominant paradigm for adapting large language models (LLMs). While discrete (textual) prompts are widely used for their interpretability, soft (parameter) prompts have recently gained traction in APIs. This is because they can encode information from more training samples while minimizing the user's token usage, leaving more space in the context window for task-specific input. However, soft prompts are tightly coupled to the LLM they are tuned on, limiting their generalization to other LLMs. This constraint is particularly problematic for efficiency and privacy: (1) tuning prompts on each LLM incurs high computational costs, especially as LLMs continue to grow in size. Additionally, (2) when the LLM is hosted externally, soft prompt tuning often requires sharing private data with the LLM provider. For instance, this is the case with the NVIDIA NeMo API. To address these issues, we propose POST (Privacy Of Soft prompt Transfer), a framework that enables private tuning of soft prompts on a small model and subsequently transfers these prompts to a larger LLM. POST uses knowledge distillation to derive a small model directly from the large LLM to improve prompt transferability, tunes the soft prompt locally, optionally with differential privacy guarantees, and transfers it back to the larger LLM using a small public dataset. Our experiments show that POST reduces computational costs, preserves privacy, and effectively transfers high-utility soft prompts.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。