arXiv:2507.22872cs.CV2025-07ICCV被引 11

针对大模型微调效率低问题,提出动态筛选关键参数与图像块的方法。

TR-PTS: Task-Relevant Parameter and Token Selection for Efficient Tuning

  • 基于FIM矩阵逐层筛选对任务重要的参数进行微调。
  • 同时动态保留关键图像块,合并冗余块,降低计算开销。
  • 在FGVC和VTAB-1k上超越全量微调,分别提升3.40%和10.35%。

大规模预训练模型在视觉任务中表现优异,但微调时因计算和存储成本过高而不实用。参数高效微调(PEFT)通过仅更新部分参数缓解此问题,但多数方法为任务无关,无法充分利用任务特异性适应,导致效率与性能欠佳。为此,我们提出任务相关参数与图像块选择(TR-PTS),一种任务驱动的框架,以提升计算效率与准确率。具体地,引入任务相关参数选择,利用费舍尔信息矩阵(FIM)逐层识别并微调最具有信息量的参数,其余参数保持冻结;同时,任务相关图像块选择动态保留最具信息量的图像块,并合并冗余块,降低计算开销。通过联合优化参数与图像块,TR-PTS使模型聚焦于任务区分性信息。我们在多个基准数据集上评估该方法,包括FGVC与VTAB-1k,结果表明其性能优于全量微调,在这两个数据集上分别提升3.40%与10.35%。代码已公开于https://github.com/synbol/TR-PTS。

原文摘要 · Abstract (English)

Large pre-trained models achieve remarkable performance in vision tasks but are impractical for fine-tuning due to high computational and storage costs. Parameter-Efficient Fine-Tuning (PEFT) methods mitigate this issue by updating only a subset of parameters; however, most existing approaches are task-agnostic, failing to fully exploit task-specific adaptations, which leads to suboptimal efficiency and performance. To address this limitation, we propose Task-Relevant Parameter and Token Selection (TR-PTS), a task-driven framework that enhances both computational efficiency and accuracy. Specifically, we introduce Task-Relevant Parameter Selection, which utilizes the Fisher Information Matrix (FIM) to identify and fine-tune only the most informative parameters in a layer-wise manner, while keeping the remaining parameters frozen. Simultaneously, Task-Relevant Token Selection dynamically preserves the most informative tokens and merges redundant ones, reducing computational overhead. By jointly optimizing parameters and tokens, TR-PTS enables the model to concentrate on task-discriminative information. We evaluate TR-PTS on benchmark, including FGVC and VTAB-1k, where it achieves state-of-the-art performance, surpassing full fine-tuning by 3.40% and 10.35%, respectively. The code are available at https://github.com/synbol/TR-PTS.

参数高效微调视觉任务FIM模型压缩

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