用语义引导微调大模型,提升长尾视觉识别准确率
Semantic-guided Fine-tuning of Foundation Model for Long-tailed Visual Recognition
- 引入语义适配器,利用文本描述指导视觉编码器微调
- 在多个基准数据集上显著提升尾部类别的识别精度
- 适合关注长尾分布、多模态对齐的研究者和工程师
长尾分布中类别样本数量差异导致稀有类别性能下降。得益于在大规模开放数据集上预训练的通用表示能力,基础模型为解决该问题提供了潜力。现有先进微调方法通常仅调整视觉编码器,忽略冻结文本编码器提取的语义信息,忽视了视文本模态间的对齐。为此,本文提出Sage(Semantic-guided fine-tuning of foundation model for long-tailed visual recognition),将文本模态的语义引导融入视觉微调过程。具体地,设计SG-Adapter,将类别描述作为语义引导,通过注意力机制传递,使模型更关注语义相关内容,增强模态对齐。由于现有损失函数未考虑类别条件分布不一致,即使模态对齐增强,仍存在预测偏差,导致尾部类别提升不如头部。为此,基于理论分析提出分布不匹配感知补偿因子,有效校正该偏差,并无缝集成至损失函数。大量实验表明,Sage在多个基准数据集上均有效提升了长尾学习性能。
原文摘要 · Abstract (English)
The variance in class-wise sample sizes within long-tailed scenarios often results in degraded performance in less frequent classes. Fortunately, foundation models, pre-trained on vast open-world datasets, demonstrate strong potential for this task due to their generalizable representation, which promotes the development of adaptive strategies on pre-trained models in long-tailed learning. Advanced fine-tuning methods typically adjust visual encoders while neglecting the semantics derived from the frozen text encoder, overlooking the visual and textual alignment. To strengthen this alignment, we propose a novel approach, Semantic-guided fine-tuning of foundation model for long-tailed visual recognition (Sage), which incorporates semantic guidance derived from textual modality into the visual fine-tuning process. Specifically, we introduce an SG-Adapter that integrates class descriptions as semantic guidance to guide the fine-tuning of the visual encoder. The introduced guidance is passesed through the attention mechanism and enables the model to focus more on semantically relevant content, strengthening the alignment between the visual and textual modalities. Due to the inconsistent class-conditional distributions neglected by the existing loss function, the resulting prediction bias causes performance improvements for the tail class less than for the head class, even when the multi-modal alignment is enhanced. To address this challenge, we propose a novel distribution mismatch-aware compensation factor, which is specifically designed to rectify the prediction bias caused by the ignored inconsistent distribution based on our theoretical analysis, and is seamlessly integrated into the loss function. Extensive experiments on benchmark datasets demonstrate the effectiveness of the proposed Sage in enhancing performance in long-tailed learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。