用生成数据缓解视觉语言模型持续学习中的遗忘问题。
Synthetic Data is an Elegant GIFT for Continual Vision-Language Models
- 利用扩散模型生成图像文本对,重建预训练与下游任务数据。
- 通过对比蒸馏和对齐约束,提升模型对旧知识的保留能力。
- 适合研究持续学习、视觉语言模型或生成式AI的学者参考。
预训练视觉语言模型(VLMs)需通过持续学习(CL)高效更新知识并适应下游任务,而无需从头训练。然而,在持续微调过程中,不仅下游任务知识会丢失,预训练知识也会被破坏,且由于原始预训练数据不可获取,导致模型泛化能力下降。本文提出GIFT,一种新颖的持续微调方法,利用合成数据缓解灾难性遗忘。借助文本到图像生成技术,我们使用预训练扩散模型重建预训练数据及已学下游任务数据。通过在匹配的生成图像与对应文本提示上进行知识蒸馏,使模型可重新访问过往知识。利用生成图像-文本对在特征空间中的广泛分布与高对齐性,设计对比蒸馏损失与图像-文本对齐约束。为应对分布内过拟合,并在生成数据有限时增强蒸馏效果,引入基于费舍尔信息的自适应权重固化,实现更优的稳定性-可塑性平衡。大量实验表明,该方法在多种设置下均优于现有最先进方法。
原文摘要 · Abstract (English)
Pre-trained Vision-Language Models (VLMs) require Continual Learning (CL) to efficiently update their knowledge and adapt to various downstream tasks without retraining from scratch. However, for VLMs, in addition to the loss of knowledge previously learned from downstream tasks, pre-training knowledge is also corrupted during continual fine-tuning. This issue is exacerbated by the unavailability of original pre-training data, leaving VLM's generalization ability degrading. In this paper, we propose GIFT, a novel continual fine-tuning approach that utilizes synthetic data to overcome catastrophic forgetting in VLMs. Taking advantage of recent advances in text-to-image synthesis, we employ a pre-trained diffusion model to recreate both pre-training and learned downstream task data. In this way, the VLM can revisit previous knowledge through distillation on matching diffusion-generated images and corresponding text prompts. Leveraging the broad distribution and high alignment between synthetic image-text pairs in VLM's feature space, we propose a contrastive distillation loss along with an image-text alignment constraint. To further combat in-distribution overfitting and enhance distillation performance with limited amount of generated data, we incorporate adaptive weight consolidation, utilizing Fisher information from these synthetic image-text pairs and achieving a better stability-plasticity balance. Extensive experiments demonstrate that our method consistently outperforms previous state-of-the-art approaches across various settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。