动态调整微调参数,提升遥感图像小样本表现
Meta-Learning Hyperparameters for Parameter Efficient Fine-Tuning
- 用自适应缩放器动态调节模块插入位置与学习率
- 在五个数据集上显著提升尾部类别准确率
- 适合资源有限的遥感图像迁移学习场景
由于数据限制和长尾分布,为特定领域(如遥感)从头训练大型基础模型几乎不可行。将自然图像预训练模型微调至遥感图像是一种直接方案。为降低计算成本并提升尾部类别性能,现有方法采用参数高效微调(PEFT)技术,如LoRA和AdaptFormer。然而我们发现,固定超参数(如层内位置、层深度、缩放因子)会显著制约PEFT性能,因遥感图像微调对这些设置高度敏感。为此,我们提出MetaPEFT,引入自适应缩放器,在微调过程中动态调整模块影响。MetaPEFT动态调节三个关键因素:模块插入位置、层选择与模块级学习率,共同控制PEFT模块在全网络中的影响。我们在三种迁移学习场景和五个数据集(涵盖遥感与自然图像)上进行大量实验。结果表明,MetaPEFT在跨光谱适应任务中达到当前最优性能,仅需少量可训练参数,并显著提升尾部类别准确率。
原文摘要 · Abstract (English)
Training large foundation models from scratch for domain-specific applications is almost impossible due to data limits and long-tailed distributions -- taking remote sensing (RS) as an example. Fine-tuning natural image pre-trained models on RS images is a straightforward solution. To reduce computational costs and improve performance on tail classes, existing methods apply parameter-efficient fine-tuning (PEFT) techniques, such as LoRA and AdaptFormer. However, we observe that fixed hyperparameters -- such as intra-layer positions, layer depth, and scaling factors, can considerably hinder PEFT performance, as fine-tuning on RS images proves highly sensitive to these settings. To address this, we propose MetaPEFT, a method incorporating adaptive scalers that dynamically adjust module influence during fine-tuning. MetaPEFT dynamically adjusts three key factors of PEFT on RS images: module insertion, layer selection, and module-wise learning rates, which collectively control the influence of PEFT modules across the network. We conduct extensive experiments on three transfer-learning scenarios and five datasets in both RS and natural image domains. The results show that MetaPEFT achieves state-of-the-art performance in cross-spectral adaptation, requiring only a small amount of trainable parameters and improving tail-class accuracy significantly.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。