arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2303.03052cs.CVcs.LG
收藏

掩码图像是用于鲁棒微调的反事实样本

Masked Images Are Counterfactual Samples for Robust Fine-tuning

Sun Yat-sen University(中山大学)

Yao Xiao, Ziyi Tang, Pengxu Wei, Cong Liu, Liang Lin

AI总结:

针对微调大型预训练模型时分布内性能与分布外鲁棒性之间的权衡问题,提出基于因果分析的微调方法,利用类激活图掩码图像生成反事实样本进行特征蒸馏,有效打破虚假相关性并提升OOD鲁棒性。

摘要:

深度学习模型受到训练数据与测试数据之间分布偏移的挑战。最近,在多样化数据上预训练的大型模型展示了对各种分布偏移前所未有的鲁棒性。然而,微调这些模型可能会导致分布内(ID)性能与分布外(OOD)鲁棒性之间的权衡。现有的解决这种权衡的方法并没有明确地解决OOD鲁棒性问题。在本文中,基于对上述问题的因果分析,我们提出了一种新颖的微调方法,该方法使用掩码图像作为反事实样本,帮助提高微调模型的鲁棒性。具体而言,我们基于类激活图掩码图像中与语义相关或与语义无关的图像块,以打破虚假相关性,并用来自其他图像的图像块填充被掩码的部分。生成的反事实样本用于与预训练模型的基于特征的蒸馏。大量实验证实,使用所提出的掩码图像对微调进行正则化,可以在ID和OOD性能之间实现更好的权衡,在OOD性能上超越了以前的方法。我们的代码可在https://github.com/Coxy7/robust-finetuning获取。

英文摘要:

Deep learning models are challenged by the distribution shift between the training data and test data. Recently, the large models pre-trained on diverse data have demonstrated unprecedented robustness to various distribution shifts. However, fine-tuning these models can lead to a trade-off between in-distribution (ID) performance and out-of-distribution (OOD) robustness. Existing methods for tackling this trade-off do not explicitly address the OOD robustness problem. In this paper, based on causal analysis of the aforementioned problems, we propose a novel fine-tuning method, which uses masked images as counterfactual samples that help improve the robustness of the fine-tuning model. Specifically, we mask either the semantics-related or semantics-unrelated patches of the images based on class activation map to break the spurious correlation, and refill the masked patches with patches from other images. The resulting counterfactual samples are used in feature-based distillation with the pre-trained model. Extensive experiments verify that regularizing the fine-tuning with the proposed masked images can achieve a better trade-off between ID and OOD performance, surpassing previous methods on the OOD performance. Our code is available at https://github.com/Coxy7/robust-finetuning.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑