arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2305.19270cs.CVcs.LG

视觉语言模型的无遗忘学习

Learning without Forgetting for Vision-Language Models

Nanjing University(南京大学);Nanyang Technological University(南洋理工大学)

Da-Wei Zhou, Yuanhan Zhang, Yan Wang, Jingyi Ning, Han-Jia Ye, De-Chuan Zhan, Ziwei Liu

AI总结:

针对视觉语言模型(VLM)在类别增量学习(CIL)中存在的灾难性遗忘和多模态信息利用不足问题,提出PROOF方法,通过训练特定任务投影和融合模块,在九个基准数据集上实现了最先进性能。

摘要:

类别增量学习(CIL)或持续学习是现实世界中一种理想的能力,它要求学习系统在适应新任务的同时不遗忘先前的任务。传统CIL方法专注于视觉信息以掌握核心特征,而视觉语言模型(VLM)的最新进展表明,借助文本信息,模型在学习可泛化表示方面具有良好能力。然而,当持续训练新类别时,VLM往往会遭受对先前知识的灾难性遗忘。将VLM应用于CIL面临两大挑战:1)如何在不遗忘的情况下调整模型;2)如何充分利用多模态信息。为此,我们提出了PROjectiOn Fusion(PROOF)方法,使VLM能够无遗忘地学习。为应对第一个挑战,我们提出基于冻结的图像/文本编码器训练特定任务的投影。面对新任务时,扩展新投影并固定先前投影,从而减轻对旧概念的遗忘。针对第二个挑战,我们提出融合模块以更好地利用跨模态信息。通过联合调整视觉和文本特征,模型能够捕获具有更强表示能力的语义信息。在九个基准数据集上的大量实验验证了PROOF实现了最先进的性能。代码可在https://github.com/zhoudw-zdw/PROOF获取。

英文摘要:

Class-Incremental Learning (CIL) or continual learning is a desired capability in the real world, which requires a learning system to adapt to new tasks without forgetting former ones. While traditional CIL methods focus on visual information to grasp core features, recent advances in Vision-Language Models (VLM) have shown promising capabilities in learning generalizable representations with the aid of textual information. However, when continually trained with new classes, VLMs often suffer from catastrophic forgetting of former knowledge. Applying VLMs to CIL poses two major challenges: 1) how to adapt the model without forgetting; and 2) how to make full use of the multi-modal information. To this end, we propose PROjectiOn Fusion (PROOF) that enables VLMs to learn without forgetting. To handle the first challenge, we propose training task-specific projections based on the frozen image/text encoders. When facing new tasks, new projections are expanded and former projections are fixed, alleviating the forgetting of old concepts. For the second challenge, we propose the fusion module to better utilize the cross-modality information. By jointly adjusting visual and textual features, the model can capture semantic information with stronger representation ability. Extensive experiments on nine benchmark datasets validate PROOF achieves state-of-the-art performance. Code is available at https://github.com/zhoudw-zdw/PROOF

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑