arXiv:2505.24088cs.LGcs.CV2025-05ICML被引 1

用动态代理增强特征分布对齐,防止视觉大模型微调时遗忘旧知识。

Proxy-FDA: Proxy-based Feature Distribution Alignment for Fine-tuning Vision Foundation Models without Forgetting

  • 通过构建近邻图对齐预训练与微调后的特征分布结构。
  • 在图像分类、图文生成等任务中显著降低概念遗忘率。
  • 适用于端到端、少样本及持续学习等多种微调场景。

视觉基础模型在大规模数据上预训练后,可通过对下游任务进行微调实现适应。然而,在单一任务上微调常导致其他任务的知识遗忘。现有鲁棒微调方法通常通过匹配原始与微调后模型权重或特征对来保留知识,但这种逐点匹配过于严格,未考虑编码丰富知识的特征邻域结构。本文提出新型正则化方法 Proxy-FDA,显式保留特征空间中的结构化知识。Proxy-FDA 在预训练与微调特征空间间执行基于最近邻图的特征分布对齐,并通过动态生成的高信息量代理进一步提升数据多样性。实验表明,Proxy-FDA 显著减少微调过程中的概念遗忘,且遗忘程度与分布距离度量(相比 L2 距离)呈强相关性。我们还验证了其在端到端、少样本及持续微调等多场景下的有效性,涵盖图像分类、图像描述生成与视觉问答等多种任务。

原文摘要 · Abstract (English)

Vision foundation models pre-trained on massive data encode rich representations of real-world concepts, which can be adapted to downstream tasks by fine-tuning. However, fine-tuning foundation models on one task often leads to the issue of concept forgetting on other tasks. Recent methods of robust fine-tuning aim to mitigate forgetting of prior knowledge without affecting the fine-tuning performance. Knowledge is often preserved by matching the original and fine-tuned model weights or feature pairs. However, such point-wise matching can be too strong, without explicit awareness of the feature neighborhood structures that encode rich knowledge as well. We propose a novel regularization method Proxy-FDA that explicitly preserves the structural knowledge in feature space. Proxy-FDA performs Feature Distribution Alignment (using nearest neighbor graphs) between the pre-trained and fine-tuned feature spaces, and the alignment is further improved by informative proxies that are generated dynamically to increase data diversity. Experiments show that Proxy-FDA significantly reduces concept forgetting during fine-tuning, and we find a strong correlation between forgetting and a distributional distance metric (in comparison to L2 distance). We further demonstrate Proxy-FDA's benefits in various fine-tuning settings (end-to-end, few-shot and continual tuning) and across different tasks like image classification, captioning and VQA.

视觉模型微调知识保持特征对齐

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