用视觉语言模型提升长尾多标签图像识别,解决类别不平衡问题
Unleashing the Power of Vision-Language Models for Long-Tailed Multi-Label Visual Recognition
- 通过文本编码器显式建模标签间相关性,构建可学习的软提示与图卷积网络
- 在VOC-LT、COCO-LT等数据集上显著优于现有方法,尾类准确率提升明显
- 适合关注真实场景中长尾多标签识别的研究者与开发者
长尾多标签图像识别面临严峻挑战:图像通常包含多个标签,且类别分布极度不均,导致模型偏向头部类别而忽视尾部类别。尽管已有研究利用CLIP等预训练视觉语言模型结合长尾学习技术,借助丰富的视觉-文本先验提升性能,但现有方法常直接从不平衡数据集中提取语义类间关系,因尾部类别数据稀少而产生不可靠关联。此外,CLIP的零样本范式针对单标签图文匹配优化,难以胜任多标签任务。为此,我们提出相关性自适应提示网络(CAPNET),一种端到端框架,显式从CLIP文本编码器中建模标签相关性。该框架引入标签感知传播的图卷积网络和可学习的软提示以生成优化嵌入,并采用分布平衡的焦点损失与类别感知重加权进行训练。同时,通过测试时集成与参数高效微调实现视觉-文本模态对齐,防止尾部类别过拟合,又不损害头部类别表现。在VOC-LT、COCO-LT、NUS-WIDE等多个基准上的大量实验与消融分析表明,CAPNET显著超越当前最优方法,验证了其在真实世界长尾多标签识别中的有效性。
原文摘要 · Abstract (English)
Long-tailed multi-label visual recognition poses a significant challenge, as images typically contain multiple labels with highly imbalanced class distributions, leading to biased models that favor head classes while underperforming on tail classes. Recent efforts have leveraged pre-trained vision-language models, such as CLIP, alongside long-tailed learning techniques to exploit rich visual-textual priors for improved performance. However, existing methods often derive semantic inter-class relationships directly from imbalanced datasets, resulting in unreliable correlations for tail classes due to data scarcity. Moreover, CLIP's zero-shot paradigm is optimized for single-label image-text matching, making it suboptimal for multi-label tasks. To address these issues, we propose the correlation adaptation prompt network (CAPNET), a novel end-to-end framework that explicitly models label correlations from CLIP's textual encoder. The framework incorporates a graph convolutional network for label-aware propagation and learnable soft prompts for refined embeddings. It utilizes a distribution-balanced Focal loss with class-aware re-weighting for optimized training under imbalance. Moreover, it improves generalization through test-time ensembling and realigns visual-textual modalities using parameter-efficient fine-tuning to avert overfitting on tail classes without compromising head class performance. Extensive experiments and ablation studies on benchmarks including VOC-LT, COCO-LT, and NUS-WIDE demonstrate that CAPNET achieves substantial improvements over state-of-the-art methods, validating its effectiveness for real-world long-tailed multi-label visual recognition.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。