动态调整特征重要性,提升视觉语言模型对新类别的泛化能力
Dynamic Rank Adaptation for Vision-Language Models
- 按特征重要性分组并动态分配适配秩,保留关键信息
- 在多个基准上显著提升新类别识别准确率,尤其跨数据集场景
- 适合需要强泛化能力的下游任务,如开放世界识别
预训练的大规模视觉语言模型(如CLIP)展现出强大的泛化能力。现有基于提示和适配器的方法在微调中取得进展,但仍面临难以保持对未见新类别的泛化能力的问题。这主要源于这些方法对图像和文本编码器的所有标记一视同仁,易导致对低信息量特征(如背景噪声、模板词)过拟合,从而削弱对新概念识别至关重要的通用表示。为此,我们提出动态秩适配(DRA),一种新型适配器变体方法,专门用于增强新类别泛化能力。DRA首先利用序列注意力评估并按重要性分组标记;然后根据每组的重要性动态分配更高特征秩,优先适配重要标记。此外,设计新的通道响应机制,优先保留并适配每个实例中最具信息量的特征通道。同时引入L1正则项以稳定训练。大量实验表明,DRA在多个基准上优于现有方法,尤其在基础-新类别、跨数据集评估和领域泛化任务中表现突出。
原文摘要 · Abstract (English)
Pre-trained large vision-language models (VLMs) like CLIP demonstrate impressive generalization ability. Existing prompt-based and adapter-based works have made significant progress in fine-tuning VLMs but still face the challenges of maintaining strong generalization abilities, particularly towards unseen new classes. This limitation partly arises from these methods treating all tokens of the image and text encoder equally, which can lead to overfitting on less informative features (e.g., background noise, template words) and degrade the general representations that are crucial for novel concept recognition. To address this issue, we propose Dynamic Rank Adaptation (DRA), a novel adapter variant method, designed specifically to enhance new class generalization. DRA dynamically allocates adaptation ranks based on the importance of features during training to preserve general knowledge. DRA first employs token importance grouping, using sequence attention to evaluate and group tokens by their importance. Then, we adopt rank adaptation according to the importance of each token group dynamically by assigning higher feature ranks to the more important tokens. Also, we design a new channel response mechanism to prioritize the preservation and adaptation of feature channels identified as the most informative for each instance. In addition, a L1 regularization term is introduced to stabilize the training. Extensive experiments demonstrate the effectiveness and superiority of our proposed DRA over existing works, especially on enhancing the performance of new classes on various benchmarks, including base-new classes, cross-datasets evaluation and domain generalization. The source code will be published after the paper is received.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。