arXiv:2603.16063cs.CV2026-03被引 1

让视觉Transformer用线性注意力,高效处理长序列。

ViT-AdaLA: Adapting Vision Transformers with Linear Attention

  • 分三阶段对齐软注意力与线性注意力,保留原模型知识。
  • 在分类与分割任务上性能超越现有线性注意力方法。
  • 适合希望高效迁移大模型知识的研究者或工程师。

基于视觉基础模型(VFMs)的视觉Transformer(ViT)在多种视觉任务中表现卓越,但其二次复杂度限制了长序列的扩展。现有线性注意力方法通常需从头训练,消耗大量算力;而为语言模型设计的线性化方法难以迁移到ViT。为此,我们提出ViT-AdaLA,一种将先验知识有效适配并迁移至线性注意力ViT的新框架。该框架包含三个阶段:注意力对齐、特征对齐和监督微调。在注意力对齐阶段,通过逐层对齐原始softmax注意力与线性注意力,近似其行为。但残差误差会跨层累积,因此我们通过微调线性化ViT,使其最终层特征与冻结的softmax VFM教师模型对齐。最后,经监督微调将适配后的知识迁移到下游任务。大量实验表明,ViT-AdaLA在分类与分割任务上均显著优于当前主流线性注意力方法,具有广泛有效性。

原文摘要 · Abstract (English)

Vision Transformers (ViTs) based vision foundation models (VFMs) have achieved remarkable performance across diverse vision tasks, but suffer from quadratic complexity that limits scalability to long sequences. Existing linear attention approaches for ViTs are typically trained from scratch, requiring substantial computational resources, while linearization-based methods developed for large language model decoders do not transfer well to ViTs. To address these challenges, we propose ViT-AdaLA, a novel framework for effectively adapting and transferring prior knowledge from VFMs to linear attention ViTs. ViT-AdaLA consists of three stages: attention alignment, feature alignment, and supervised fine-tuning. In the attention alignment stage, we align vanilla linear attention with the original softmax-based attention in each block to approximate the behavior of softmax attention. However, residual approximation errors inevitably accumulate across layers. We mitigate this by fine-tuning the linearized ViT to align its final-layer features with a frozen softmax VFM teacher. Finally, the adapted prior knowledge is transferred to downstream tasks through supervised fine-tuning. Extensive experiments on classification and segmentation tasks demonstrate the effectiveness and generality of ViT-AdaLA over various state-of-the-art linear attention counterpart.

视觉Transformer线性注意力模型迁移

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。