arXiv:2411.01327cs.CVcs.AI2024-11NeurIPS被引 56

用傅里叶变换提升视觉提示调优,让大模型适应新数据更稳定高效。

Visual Fourier Prompt Tuning

  • 将快速傅里叶变换引入提示嵌入,同时利用空间与频率域信息
  • 在VTAB-1k上仅用0.57%参数即达73.20%平均准确率,超越现有方法
  • 特别适合预训练与微调数据差异大的场景,通用性强

随着基于视觉Transformer的模型规模持续增长,为新任务微调这些大规模预训练模型变得愈发参数密集。视觉提示调优作为参数高效微调(PEFT)方法应对此趋势。尽管取得成功,几乎所有PEFT方法仍面临显著挑战:当预训练与微调数据集存在较大差异时,性能明显下降。为此,我们借鉴人类视觉认知机制,提出视觉傅里叶提示调优(VFPT),一种通用且高效的大型Transformer模型适配方案。该方法创新性地将快速傅里叶变换融入提示嵌入,协同考虑空间与频率域信息。除内在简洁直观外,VFPT在所有数据集上均表现优异,有效应对数据差异带来的挑战。实验表明,该方法在两个基准测试中优于当前最先进基线,在VTAB-1k上仅使用0.57%模型参数便实现73.20%的平均准确率,显著提升性能。代码已开源。

原文摘要 · Abstract (English)

With the scale of vision Transformer-based models continuing to grow, finetuning these large-scale pretrained models for new tasks has become increasingly parameter-intensive. Visual prompt tuning is introduced as a parameter-efficient finetuning (PEFT) method to this trend. Despite its successes, a notable research challenge persists within almost all PEFT approaches: significant performance degradation is observed when there is a substantial disparity between the datasets applied in pretraining and finetuning phases. To address this challenge, we draw inspiration from human visual cognition, and propose the Visual Fourier Prompt Tuning (VFPT) method as a general and effective solution for adapting large-scale transformer-based models. Our approach innovatively incorporates the Fast Fourier Transform into prompt embeddings and harmoniously considers both spatial and frequency domain information. Apart from its inherent simplicity and intuitiveness, VFPT exhibits superior performance across all datasets, offering a general solution to dataset challenges, irrespective of data disparities. Empirical results demonstrate that our approach outperforms current state-of-the-art baselines on two benchmarks, with low parameter usage (e.g., 0.57% of model parameters on VTAB-1k) and notable performance enhancements (e.g., 73.20% of mean accuracy on VTAB-1k). Our code is avaliable at https://github.com/runtsang/VFPT.

视觉提示傅里叶变换参数高效

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