首次为主题模型提供可证明的删减理论保障,支持预训练与微调场景。
Provable unlearning in topic modeling and downstream tasks
- 设计高效算法实现主题模型的可证明删减,计算开销与原始数据量无关。
- 量化模型删减容量:可无损删除大量样本,性能下降可控。
- 微调后更易删减预训练数据,且无需修改基础模型,适合合规需求者。
随着训练数据来源的法律争议增多,机器删减算法的重要性日益凸显,但验证删减效果往往困难。现有可证明删减理论多局限于监督学习场景。本文首次在预训练-微调范式下,通过研究主题模型(topic models)——一种简单的词袋语言模型,可适配检索与分类等下游任务——提供了可证明的删减理论保障。首先,我们设计了一种可证明有效的主题模型删减算法,其计算开销独立于原始数据集规模。分析还量化了模型的删减容量,即在不显著影响模型性能的前提下可删减的样本数量。最后,我们正式扩展分析以涵盖对下游任务的适应性。特别地,设计了一种高效算法,在通过线性头微调主题模型后执行删减。值得注意的是,我们证明了对已微调至特定任务的模型,删减预训练数据更简单,且可在不修改基础模型的情况下完成。
原文摘要 · Abstract (English)
Machine unlearning algorithms are increasingly important as legal concerns arise around the provenance of training data, but verifying the success of unlearning is often difficult. Provable guarantees for unlearning are often limited to supervised learning settings. In this paper, we provide the first theoretical guarantees for unlearning in the pre-training and fine-tuning paradigm by studying topic models, simple bag-of-words language models that can be adapted to solve downstream tasks like retrieval and classification. First, we design a provably effective unlearning algorithm for topic models that incurs a computational overhead independent of the size of the original dataset. Our analysis additionally quantifies the deletion capacity of the model -- i.e., the number of examples that can be unlearned without incurring a significant cost in model performance. Finally, we formally extend our analyses to account for adaptation to a given downstream task. In particular, we design an efficient algorithm to perform unlearning after fine-tuning the topic model via a linear head. Notably, we show that it is easier to unlearn pre-training data from models that have been fine-tuned to a particular task, and one can unlearn this data without modifying the base model.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。