为大模型提供高效且可保证遗忘的去隐私化框架。
DP2Unlearning: An Efficient and Guaranteed Unlearning Framework for LLMs
- 用ε-差分隐私训练数据,实现模型事后高效删除指定信息
- 性能接近从头重训,但成本仅为一半左右
- 适合需合规删除敏感数据的大模型应用
大语言模型虽大幅提升语言处理能力,却可能记忆训练数据中的私密或版权内容,导致推理时泄露。传统做法是剔除数据后从头重训,虽能确保遗忘,但成本极高。近似遗忘方法虽快,但无严格保证。本文提出DP2Unlearning框架:先以ε-差分隐私保护训练数据,再通过该隐私参数实现高效且有形式化保证的遗忘。实验表明,该方法在模型性能上接近从头重训(金标准),但未学习成本约为其一半;同时,在保持模型效用和有效遗忘目标信息方面,优于现有近似遗忘方法,计算开销合理。
原文摘要 · Abstract (English)
Large language models (LLMs) have recently revolutionized language processing tasks but have also brought ethical and legal issues. LLMs have a tendency to memorize potentially private or copyrighted information present in the training data, which might then be delivered to end users at inference time. When this happens, a naive solution is to retrain the model from scratch after excluding the undesired data. Although this guarantees that the target data have been forgotten, it is also prohibitively expensive for LLMs. Approximate unlearning offers a more efficient alternative, as it consists of ex post modifications of the trained model itself to prevent undesirable results, but it lacks forgetting guarantees because it relies solely on empirical evidence. In this work, we present DP2Unlearning, a novel LLM unlearning framework that offers formal forgetting guarantees at a significantly lower cost than retraining from scratch on the data to be retained. DP2Unlearning involves training LLMs on textual data protected using ε-differential privacy (DP), which later enables efficient unlearning with the guarantees against disclosure associated with the chosen ε. Our experiments demonstrate that DP2Unlearning achieves similar model performance post-unlearning, compared to an LLM retraining from scratch on retained data -- the gold standard exact unlearning -- but at approximately half the unlearning cost. In addition, with a reasonable computational cost, it outperforms approximate unlearning methods at both preserving the utility of the model post-unlearning and effectively forgetting the targeted information.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。