中文

利用少量样本在预训练模型中实现类别级遗忘

计算机视觉与模式识别 2022-10-20 v1 机器学习

摘要

为解决现实世界问题,深度学习模型通常在许多类别上联合训练。然而未来,某些类别可能因隐私/伦理问题而受限,必须从已在其上训练过的模型中移除受限类别的知识。由于隐私/伦理问题,可用数据也可能有限,重新训练模型将不可行。我们提出一种新方法解决该问题,且不影响模型对剩余类别的预测能力。我们的方法识别与受限类别高度相关的模型参数,并利用有限的可用训练数据从中移除关于受限类别的知识。我们的方法显著更快,且性能与在剩余类别完整数据上重新训练的模型相近。

关键词

引用

@article{arxiv.2210.10670,
  title  = {Attaining Class-level Forgetting in Pretrained Model using Few Samples},
  author = {Pravendra Singh and Pratik Mazumder and Mohammed Asad Karim},
  journal= {arXiv preprint arXiv:2210.10670},
  year   = {2022}
}

备注

Accepted in ECCV 2022