arXiv:2410.22952cs.CVcs.AI2024-10NeurIPS被引 10

用霍舍尔变换动态调整视觉Transformer适配矩阵秩,提升微调灵活性。

Efficient Adaptation of Pre-trained Vision Transformer via Householder Transformation

  • 基于奇异值分解思想,用霍舍尔变换构建可学习的正交矩阵。
  • 各层适配矩阵秩可变,灵活匹配不同层特征需求。
  • 性能媲美现有方法,参数量更少,适合资源受限场景。

预训练视觉Transformer的参数高效微调(PEFT)常通过学习低秩适配矩阵实现,该矩阵由降维和升维矩阵构成,瓶颈维度决定可训练参数数量。然而,现有方法多采用固定瓶颈维度,难以适应各层差异。为此,本文受奇异值分解(SVD)启发,提出新方法:将适配矩阵表示为左酉矩阵、对角缩放矩阵与右酉矩阵的乘积。利用霍舍尔变换构造正交矩阵,仅需向量即可高效生成;对角元素分层学习,能灵活捕捉各层特性。该设计使不同层的适配矩阵具有可变秩,显著提升适应灵活性。在标准下游视觉任务上的实验表明,该方法达到有竞争力的微调性能。

原文摘要 · Abstract (English)

A common strategy for Parameter-Efficient Fine-Tuning (PEFT) of pre-trained Vision Transformers (ViTs) involves adapting the model to downstream tasks by learning a low-rank adaptation matrix. This matrix is decomposed into a product of down-projection and up-projection matrices, with the bottleneck dimensionality being crucial for reducing the number of learnable parameters, as exemplified by prevalent methods like LoRA and Adapter. However, these low-rank strategies typically employ a fixed bottleneck dimensionality, which limits their flexibility in handling layer-wise variations. To address this limitation, we propose a novel PEFT approach inspired by Singular Value Decomposition (SVD) for representing the adaptation matrix. SVD decomposes a matrix into the product of a left unitary matrix, a diagonal matrix of scaling values, and a right unitary matrix. We utilize Householder transformations to construct orthogonal matrices that efficiently mimic the unitary matrices, requiring only a vector. The diagonal values are learned in a layer-wise manner, allowing them to flexibly capture the unique properties of each layer. This approach enables the generation of adaptation matrices with varying ranks across different layers, providing greater flexibility in adapting pre-trained models. Experiments on standard downstream vision tasks demonstrate that our method achieves promising fine-tuning performance.

视觉Transformer参数高效微调低秩适配霍舍尔变换

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