arXiv:2506.20941cs.LG2025-06中稿 · ICLR被引 1

利用模型训练历史状态实现大模型数据擦除,高效且精准。

Revisiting the Past: Data Unlearning with Model State History

  • 通过分析模型训练过程中的多个检查点,估算并抵消特定数据的影响。
  • 在多个基准测试中表现优于现有算法,部分指标提升显著。
  • 适合需要合规性或数据可控性的大模型应用开发人员使用。

大型语言模型基于海量网络数据训练,其中可能包含隐私数据、版权内容、事实错误信息或降低模型性能的数据。完全重训以移除这些数据的影响在计算上不可行。为此,研究者提出了无学习算法,在低计算成本下消除特定数据对模型的影响,同时保持其余模型结构不变。然而,精确消除大模型中特定数据的影响仍具挑战。本文提出一种新算法MSA(Model State Arithmetic),利用训练过程中保存的多个模型检查点,估算并抵消目标数据的影响。实验表明,MSA在多个基准、模型和评估指标上均表现优异,常优于现有机器无学习算法,表明其可作为实现灵活数据擦除能力的有效途径。

原文摘要 · Abstract (English)

Large language models are trained on massive corpora of web data, which may include private data, copyrighted material, factually inaccurate data, or data that degrades model performance. Eliminating the influence of such problematic datapoints on a model through complete retraining -- by repeatedly pretraining the model on datasets that exclude these specific instances -- is computationally prohibitive. To address this, unlearning algorithms have been proposed, that aim to eliminate the influence of particular datapoints at a low computational cost, while leaving the rest of the model intact. However, precisely unlearning the influence of data on a large language model has proven to be a major challenge. In this work, we propose a new algorithm, MSA (Model State Arithmetic), for unlearning datapoints in large language models. MSA utilizes prior model checkpoints -- artifacts that record model states at different stages of pretraining -- to estimate and counteract the effect of targeted datapoints. Our experimental results show that MSA achieves competitive performance and often outperforms existing machine unlearning algorithms across multiple benchmarks, models, and evaluation metrics, suggesting that MSA could be an effective approach towards more flexible large language models that are capable of data erasure.

大模型数据擦除无学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。