提出新方法缓解多模态模型指令微调中的视觉知识遗忘问题。
Mitigating Visual Knowledge Forgetting in MLLM Instruction-tuning via Modality-decoupled Gradient Descent
- 分离视觉与任务优化,通过梯度调控保持视觉表征丰富性。
- 在多个下游任务上显著降低视觉知识遗忘率,提升任务适应能力。
- 适合需要保留预训练视觉能力的轻量化多模态模型微调场景。
近期多模态大模型(MLLM)在大规模多模态数据预训练后展现出强大的视觉理解与推理能力。然而,与预训练阶段丰富的图文对齐不同,指令微调通常以文本为主导,视觉监督较弱,导致预训练的视觉理解能力退化,引发视觉遗忘。现有方法如直接微调和持续学习未能有效解决此问题,常压缩视觉表征并优先于任务对齐,加剧视觉遗忘。为此,本文从有效秩角度量化视觉表征丰富性的退化,基于信息瓶颈原理将其解释为过度压缩所致。据此提出模态解耦梯度下降(MDGD)方法,通过调控梯度更新,维持视觉表征的有效秩,缓解信息压缩。通过显式解耦视觉理解与任务对齐优化,MDGD在实现高效任务适配的同时保留了预训练视觉知识。为进一步实现轻量级微调,还提出基于梯度掩码的记忆高效微调方法,仅更新部分参数,支持参数高效微调(PEFT),降低计算开销。大量实验表明,该方法在多种下游任务和主干模型上均能有效缓解视觉遗忘,并实现强任务适应能力。
原文摘要 · Abstract (English)
Recent MLLMs have shown emerging visual understanding and reasoning abilities after being pre-trained on large-scale multimodal datasets. Unlike pre-training, where MLLMs receive rich visual-text alignment, instruction-tuning is often text-driven with weaker visual supervision, leading to the degradation of pre-trained visual understanding and causing visual forgetting. Existing approaches, such as direct fine-tuning and continual learning methods, fail to explicitly address this issue, often compressing visual representations and prioritizing task alignment over visual retention, which further worsens visual forgetting. To overcome this limitation, we introduce a novel perspective leveraging effective rank to quantify the degradation of visual representation richness, interpreting this degradation through the information bottleneck principle as excessive compression that leads to the degradation of crucial pre-trained visual knowledge. Building on this view, we propose a modality-decoupled gradient descent (MDGD) method that regulates gradient updates to maintain the effective rank of visual representations while mitigating the over-compression effects described by the information bottleneck. By explicitly disentangling the optimization of visual understanding from task-specific alignment, MDGD preserves pre-trained visual knowledge while enabling efficient task adaptation. To enable lightweight instruction-tuning, we further develop a memory-efficient fine-tuning approach using gradient masking, which selectively updates a subset of model parameters to enable parameter-efficient fine-tuning (PEFT), reducing computational overhead while preserving rich visual representations. Extensive experiments across various downstream tasks and backbone MLLMs demonstrate that MDGD effectively mitigates visual forgetting from pre-trained tasks while enabling strong adaptation to new tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。