让预训练视觉模型持续删掉特定信息,影响小还高效。
Practical Continual Forgetting for Pre-trained Vision Models
- 用分组稀疏LoRA动态调整模型,逐个删知识
- 删特定类别时对其他类别性能影响小于2%
- 适合隐私保护场景,尤其数据缺失时仍有效
出于隐私与安全考虑,从预训练视觉模型中删除特定信息的需求日益显著。在真实场景中,擦除请求可能随时由用户或模型所有者提出,且通常呈序列性。因此,需要在不破坏其余知识的前提下,持续地选择性删除信息。我们定义该问题为持续遗忘,并识别出三大挑战:(i) 对于不想要的知识,需高效且有效地删除;(ii) 对于保留的知识,遗忘过程带来的影响应尽可能小;(iii) 在真实场景中,遗忘过程中训练样本可能稀缺或部分缺失。为此,我们首先提出分组稀疏LoRA(GS-LoRA)。具体而言,针对挑战 (i),我们在Transformer块的前馈网络(FFN)层中引入低秩适配(LoRA)模块,独立处理每个遗忘任务;针对挑战 (ii),采用简单的分组稀疏正则化,实现特定LoRA组的自动选择并置零其余组。为进一步拓展到更实际场景,我们引入原型信息作为额外监督,提出更实用的GS-LoRA++。对于被遗忘类别,我们将其输出logits远离原始原型;对于保留类别,则拉近其logits至各自原型。我们在人脸识别、目标检测和图像分类任务上进行了广泛实验,结果表明,该方法能在几乎不影响其他类别的前提下,成功删除特定类别知识。代码已开源:https://github.com/bjzhb666/GS-LoRA。
原文摘要 · Abstract (English)
For privacy and security concerns, the need to erase unwanted information from pre-trained vision models is becoming evident nowadays. In real-world scenarios, erasure requests originate at any time from both users and model owners, and these requests usually form a sequence. Therefore, under such a setting, selective information is expected to be continuously removed from a pre-trained model while maintaining the rest. We define this problem as continual forgetting and identify three key challenges. (i) For unwanted knowledge, efficient and effective deleting is crucial. (ii) For remaining knowledge, the impact brought by the forgetting procedure should be minimal. (iii) In real-world scenarios, the training samples may be scarce or partially missing during the process of forgetting. To address them, we first propose Group Sparse LoRA (GS-LoRA). Specifically, towards (i), we introduce Low-Rank Adaptation (LoRA) modules to fine-tune the Feed-Forward Network (FFN) layers in Transformer blocks for each forgetting task independently, and towards (ii), a simple group sparse regularization is adopted, enabling automatic selection of specific LoRA groups and zeroing out the others. To further extend GS-LoRA to more practical scenarios, we incorporate prototype information as additional supervision and introduce a more practical approach, GS-LoRA++. For each forgotten class, we move the logits away from its original prototype. For the remaining classes, we pull the logits closer to their respective prototypes. We conduct extensive experiments on face recognition, object detection, and image classification and demonstrate that our method manages to forget specific classes with minimal impact on other classes. Codes have been released on https://github.com/bjzhb666/GS-LoRA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。