arXiv:2502.15895cs.LGcs.AI2025-02ICLR被引 11

新方法DiGraP用梯度方向提升大模型微调的鲁棒性,效果优于现有方法。

Directional Gradient Projection for Robust Fine-Tuning of Foundation Models

  • 基于梯度方向设计可训练投影,避免传统方法依赖繁琐调参。
  • 在图像分类和多模态任务中均提升分布外泛化能力,显著改善鲁棒性。
  • 适用于图像与视觉问答场景,尤其适合追求稳定性能的研究者。

鲁棒微调旨在将大型基础模型适配下游任务的同时保持其对分布偏移的鲁棒性。现有方法主要通过约束和投影当前模型以贴近预训练初始化,依据微调后与预训练权重间的大小差异,但常需大量超参数调优,且可能引发欠拟合。本文提出方向梯度投影(DiGraP),一种新型分层可训练方法,引入梯度的方向信息,实现正则化与多目标优化的融合。除在图像分类上验证外,我们还将该领域扩展至多模态评估设置。具体而言,通过重构图像分类为视觉问答(VQA)基准分析单模态与多模态之间的差距,并按分布偏移类型与程度(近域/远域OOD)对十个外部分布(OOD)VQA数据集进行分类。实验结果表明,DiGraP在图像分类与VQA任务中均持续优于现有基线,涵盖判别式与生成式骨干网络,在分布内(ID)泛化与分布外(OOD)鲁棒性方面均有提升。

原文摘要 · Abstract (English)

Robust fine-tuning aims to adapt large foundation models to downstream tasks while preserving their robustness to distribution shifts. Existing methods primarily focus on constraining and projecting current model towards the pre-trained initialization based on the magnitudes between fine-tuned and pre-trained weights, which often require extensive hyper-parameter tuning and can sometimes result in underfitting. In this work, we propose Directional Gradient Projection (DiGraP), a novel layer-wise trainable method that incorporates directional information from gradients to bridge regularization and multi-objective optimization. Besides demonstrating our method on image classification, as another contribution we generalize this area to the multi-modal evaluation settings for robust fine-tuning. Specifically, we first bridge the uni-modal and multi-modal gap by performing analysis on Image Classification reformulated Visual Question Answering (VQA) benchmarks and further categorize ten out-of-distribution (OOD) VQA datasets by distribution shift types and degree (i.e. near versus far OOD). Experimental results show that DiGraP consistently outperforms existing baselines across Image Classfication and VQA tasks with discriminative and generative backbones, improving both in-distribution (ID) generalization and OOD robustness.

微调鲁棒性多模态

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