arXiv:2411.06764cs.CVcs.LG2024-11被引 8

通过多阶段知识整合,提升视觉语言模型持续学习能力。

Multi-Stage Knowledge Integration of Vision-Language Models for Continual Learning

  • 分四阶段模拟人类学习过程,融合跨模态与多粒度知识
  • 在多个下游任务上保持零样本性能,避免灾难性遗忘
  • 无需额外数据,适合资源受限场景的模型持续优化

视觉语言模型(VLMs)在大规模图像-文本数据集上预训练后,可对未见数据进行零样本预测,但在特定未见任务上表现可能不佳。持续学习(CL)可在不联合训练的情况下帮助VLMs适应新数据分布,但面临灾难性遗忘和泛化遗忘的挑战。尽管基于知识蒸馏的方法取得进展,但仍存在两个严重局限:一是普遍采用的单教师范式难以传递全面知识;二是现有方法未能充分利用原始训练数据中的多模态信息,转而依赖额外数据进行蒸馏,增加计算与存储开销。为缓解上述问题,我们借鉴知识整合理论(KIT),提出多阶段知识整合网络(MulKI),模拟人类学习过程。MulKI包含四个阶段:激发想法、引入新知识、区分知识、建立关联。首先利用原型对齐跨模态,激发跨模态知识;然后通过原型构建细粒度的模内与模间关系,引入新知识;接着自适应区分并重加权来自两个教师模型的知识;最后在模内与模间任务间建立连接,整合旧知识与新知识。实验表明,该方法在保持零样本能力的同时,有效支持多样化下游任务的持续学习,展现了其在应对动态数据分布方面的潜力。

原文摘要 · Abstract (English)

Vision Language Models (VLMs), pre-trained on large-scale image-text datasets, enable zero-shot predictions for unseen data but may underperform on specific unseen tasks. Continual learning (CL) can help VLMs effectively adapt to new data distributions without joint training, but faces challenges of catastrophic forgetting and generalization forgetting. Although significant progress has been achieved by distillation-based methods, they exhibit two severe limitations. One is the popularly adopted single-teacher paradigm fails to impart comprehensive knowledge, The other is the existing methods inadequately leverage the multimodal information in the original training dataset, instead they rely on additional data for distillation, which increases computational and storage overhead. To mitigate both limitations, by drawing on Knowledge Integration Theory (KIT), we propose a Multi-Stage Knowledge Integration network (MulKI) to emulate the human learning process in distillation methods. MulKI achieves this through four stages, including Eliciting Ideas, Adding New Ideas, Distinguishing Ideas, and Making Connections. During the four stages, we first leverage prototypes to align across modalities, eliciting cross-modal knowledge, then adding new knowledge by constructing fine-grained intra- and inter-modality relationships with prototypes. After that, knowledge from two teacher models is adaptively distinguished and re-weighted. Finally, we connect between models from intra- and inter-task, integrating preceding and new knowledge. Our method demonstrates significant improvements in maintaining zero-shot capabilities while supporting continual learning across diverse downstream tasks, showcasing its potential in adapting VLMs to evolving data distributions.

持续学习视觉语言模型知识蒸馏多模态

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