arXiv:2508.21300cs.LG2025-08被引 6

提升LoRA微调模型遗忘效率,实现更准更快的敏感信息删除。

Improving Fisher Information Estimation and Efficiency for LoRA-based LLM Unlearning

  • 基于鱼群信息重构机制,精准定位需遗忘参数
  • 相比FILA提升100倍参数效率、40倍训练速度
  • 无需访问全模型参数,适合大规模语言模型应用

大语言模型在多项任务中表现优异,但存在无意生成敏感内容的问题。传统方法通过剔除敏感数据后重新训练来解决,但计算成本过高。机器遗忘作为替代方案,可在不从头训练的前提下有效移除敏感信息。近期提出的FILA利用LoRA适配器结合鱼群信息(Fisher Information)识别与遗忘集相关的参数并进行更新。然而,FILA仍需访问全部模型参数,且未充分考虑鱼群信息的基础假设,导致重要性估计不准确。为此,本文提出VILA框架,显式考虑此前被忽略的假设,提升遗忘参数识别精度,并通过无需全局参数访问显著降低计算开销。实验表明,VILA相较FILA实现最高100倍的参数效率提升和40倍训练加速,在TOFU、WMDP、MUSE等基准上达到新最优性能。代码已开源:https://github.com/kyj93790/VILA。

原文摘要 · Abstract (English)

LLMs have demonstrated remarkable performance across various tasks but face challenges related to unintentionally generating outputs containing sensitive information. A straightforward approach to address this issue is to retrain the model after excluding the problematic data. However, this approach incurs prohibitively high computational costs. To overcome this limitation, machine unlearning has emerged as a promising solution that can effectively remove sensitive information without the need to retrain the model from scratch. Recently, FILA has been proposed as a parameter-efficient unlearning method by integrating LoRA adapters. Specifically, it calculates the Fisher information to identify parameters associated with the forget set and assigns them to LoRA adapters for updates. Despite its innovative approach, FILA still requires access to all model parameters and does not adequately account for fundamental assumptions underlying Fisher information, leading to inaccuracies in importance estimation. To address these limitations, we propose VILA, a novel unlearning framework that explicitly considers the assumptions overlooked in FILA, thereby enhancing the accuracy of parameter identification for the forget set. Moreover, VILA significantly reduces computational costs by enabling parameter identification without accessing the entire model. Our method achieves up to 100x higher parameter efficiency and 40x faster training speed compared to FILA, and sets new state-of-the-art performance on benchmarks including TOFU, WMDP, and MUSE. Our code is available at https://github.com/kyj93790/VILA.

模型遗忘LoRA高效训练鱼群信息

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