arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.01752cs.CV
收藏

FullLoRA:高效提升预训练视觉Transformer的鲁棒性

FullLoRA: Efficiently Boosting the Robustness of Pretrained Vision Transformers

Key Laboratory of AI Safety of CAS, Institute of Computing Technology, Chinese Academy of Sciences (CAS), Beijing 100190, China, and the University of Chinese Academy of Sciences, Beijing 100049, China(中国科学院计算技术研究所人工智能安全重点实验室,中国科学院(CAS)及中国科学院大学,北京)

Zheng Yuan, Jie Zhang, Shiguang Shan, Xilin Chen

AI总结:

本文提出FullLoRA框架,通过集成可学习LNLoRA模块到ViT所有关键组件,以约5%参数实现与全量微调相当的对抗鲁棒性,高效解决标准训练模型的鲁棒性提升问题。

摘要:

近年来,视觉Transformer(ViT)模型已逐渐成为各种计算机视觉任务的主流,模型的鲁棒性也受到越来越多的关注。然而,现有的大模型在训练过程中往往优先考虑性能,可能忽视了鲁棒性,这可能导致严重的安全问题。在本文中,我们建立了一个新的挑战:探索如何使用少量额外参数进行对抗性微调,以快速有效地增强标准训练模型的对抗鲁棒性。为了应对这一挑战,我们开发了新颖的LNLoRA模块,在传统LoRA模块之前引入了一个可学习的层归一化层,这有助于缓解对抗训练与标准训练范式之间参数幅度的差异。此外,我们提出了FullLoRA框架,通过将可学习的LNLoRA模块集成到基于ViT模型的所有关键组件中,同时保持预训练模型冻结,从而以参数高效的方式通过对抗性微调显著提升模型鲁棒性。在多个数据集上进行的大量实验证明了我们提出的FullLoRA框架的优越性。它在仅需约5%可学习参数的情况下,实现了与全量微调相当的鲁棒性。这也有效解决了对抗性微调带来的额外模型存储空间和巨大训练时间的担忧。

英文摘要:

In recent years, the Vision Transformer (ViT) model has gradually become mainstream in various computer vision tasks, and the robustness of the model has received increasing attention. However, existing large models tend to prioritize performance during training, potentially neglecting the robustness, which may lead to serious security concerns. In this paper, we establish a new challenge: exploring how to use a small number of additional parameters for adversarial finetuning to quickly and effectively enhance the adversarial robustness of a standardly trained model. To address this challenge, we develop novel LNLoRA module, incorporating a learnable layer normalization before the conventional LoRA module, which helps mitigate magnitude differences in parameters between the adversarial and standard training paradigms. Furthermore, we propose the FullLoRA framework by integrating the learnable LNLoRA modules into all key components of ViT-based models while keeping the pretrained model frozen, which can significantly improve the model robustness via adversarial finetuning in a parameter-efficient manner. Extensive experiments on several datasets demonstrate the superiority of our proposed FullLoRA framework. It achieves comparable robustness with full finetuning while only requiring about 5\% of the learnable parameters. This also effectively addresses concerns regarding extra model storage space and enormous training time caused by adversarial finetuning.

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

↑