通过分离视觉模型的先验与任务特征,实现高效微调。
ViT-Split: Unleashing the Power of Vision Foundation Models via Efficient Splitting Heads
- 将ViT模型分层为先验提取和任务适配两部分,冻结主干
- 训练时间减少4倍,分割任务性能持平或更优
- 适合需要快速部署、资源受限的下游任务场景
视觉基础模型(VFMs)在众多下游任务中表现卓越。现有适配方法存在两个效率问题:一是卷积网络与VFM主干交互引发早期梯度反传;二是需调优全部组件,增加复杂性,且改变原模型特征,未能充分利用先验知识。本文提出ViT-Split方法,基于观察:如DINOv2等模型的层级可划分为低级特征提取器与任务特定适配器。我们移除卷积分支,仅在冻结的VFM上添加两个头:任务头用于学习任务特征,缓解梯度问题;先验头用于利用多尺度先验特征,减少调参量和过拟合。在分割、检测、深度估计和视觉问答等任务上实验验证了该方法的有效性与高效性。具体地,相比其他适配方法,ViT-Split在ADE20K数据集上实现相当或更优性能,同时训练时间最多减少4倍。
原文摘要 · Abstract (English)
Vision foundation models (VFMs) have demonstrated remarkable performance across a wide range of downstream tasks. While several VFM adapters have shown promising results by leveraging the prior knowledge of VFMs, we identify two inefficiencies in these approaches. First, the interaction between convolutional neural network (CNN) and VFM backbone triggers early layer gradient backpropagation. Second, existing methods require tuning all components, adding complexity. Besides, these adapters alter VFM features, underutilizing the prior knowledge. To tackle these challenges, we propose a new approach called ViT-Split, based on a key observation: the layers of several VFMs, like DINOv2, can be divided into two distinct components: an extractor for learning low-level features and an adapter for learning task-specific features. Leveraging this insight, we eliminate the CNN branch and introduce two heads, task head and prior head, to the frozen VFM. The task head is designed to learn task-specific features, mitigating the early gradient propagation issue. The prior head is used to leverage the multi-scale prior features from the frozen VFM, reducing tuning parameters and overfitting. Extensive experiments on various tasks (e.g., segmentation, detection, depth estimation, and visual question answering) validate the effectiveness and efficiency of ViT-Split. Specifically, ViT-Split reduces training time up to $4\times$ while achieving comparable or even better results on ADE20K, compared to other VFM adapters.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。