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

PVP:预训练视觉参数高效微调

PVP: Pre-trained Visual Parameter-Efficient Tuning

Defense Innovation Institute(国防创新研究院);National University of Defense Technology(国防科技大学)

Zhao Song, Ke Yang, Naiyang Guan, Junjie Zhu, Peng Qiao, Qingyong Hu

AI总结:

针对低数据场景下视觉 PETuning 因提示模块初始化不当导致性能不足的问题,提出 PVP 框架,先预训练高效微调模块再迁移至下游任务,并在 FGVC 与 VTAB-1k 上取得最优结果。

摘要:

大规模预训练 transformer 已在多种计算机视觉任务中取得显著成功。然而,由于高昂的计算和存储成本,针对下游任务对这些模型进行完全微调仍然极具挑战性。近年来,参数高效微调(PETuning)技术(例如 Visual Prompt Tuning(VPT)和 Low-Rank Adaptation(LoRA))通过在预训练模型中插入轻量级提示模块,并在保持 transformer 主干冻结的同时,仅用少量可训练参数来微调这些提示模块,显著降低了计算和存储成本。尽管只需要调整少量参数,但大多数 PETuning 方法仍需要大量下游任务训练数据才能取得良好效果。在低数据场景下,尤其是每个类别只有一两个样本时,其性能并不理想。为此,我们首先通过实验发现,性能不佳主要源于提示模块的初始化方式不当,这一点也已在预训练语言模型中得到验证。随后,我们提出了 Pre-trained Visual Parameter-efficient(PVP)Tuning 框架,该框架首先预训练参数高效微调模块,然后利用这些预训练模块连同预训练 transformer 主干,在下游任务上执行参数高效微调。在五个细粒度视觉分类(FGVC)数据集和 VTAB-1k 数据集上的实验结果表明,我们提出的方法显著优于最先进的 PETuning 方法。

英文摘要:

Large-scale pre-trained transformers have demonstrated remarkable success in various computer vision tasks. However, it is still highly challenging to fully fine-tune these models for downstream tasks due to their high computational and storage costs. Recently, Parameter-Efficient Tuning (PETuning) techniques, e.g., Visual Prompt Tuning (VPT) and Low-Rank Adaptation (LoRA), have significantly reduced the computation and storage cost by inserting lightweight prompt modules into the pre-trained models and tuning these prompt modules with a small number of trainable parameters, while keeping the transformer backbone frozen. Although only a few parameters need to be adjusted, most PETuning methods still require a significant amount of downstream task training data to achieve good results. The performance is inadequate on low-data regimes, especially when there are only one or two examples per class. To this end, we first empirically identify the poor performance is mainly due to the inappropriate way of initializing prompt modules, which has also been verified in the pre-trained language models. Next, we propose a Pre-trained Visual Parameter-efficient (PVP) Tuning framework, which pre-trains the parameter-efficient tuning modules first and then leverages the pre-trained modules along with the pre-trained transformer backbone to perform parameter-efficient tuning on downstream tasks. Experiment results on five Fine-Grained Visual Classification (FGVC) and VTAB-1k datasets demonstrate that our proposed method significantly outperforms state-of-the-art PETuning methods.

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

↑