用局部补丁推理+自适应聚合,提升视觉语言模型多标签识别能力
[CLS] is Not Enough: Multi-Label Recognition via Patch-Level Inference and Adaptive Aggregation
![论文配图:[CLS] is Not Enough: Multi-Label Recognition via Patch-Level Inference and Adaptive Aggregation](https://arxiv.org/html/2605.25821v1/cls2.png)
- 通过增强补丁级表示,减少语义混淆,提升细节感知
- 在NUS-WIDE上比基线高出6%以上mAP,计算开销极低
- 完全无需训练,适合快速部署到现有模型
视觉语言模型如CLIP虽具备强大的零样本识别能力,但在多标签场景下表现不佳,主要因[CLS] token作为单一全局表征,难以准确捕捉不同尺度、上下文及共现模式的目标。为此,我们提出PIAA框架,将预测分解为补丁级推理与自适应聚合。首先从两个互补角度增强补丁级预测:(i) 减少视觉编码器中的语义纠缠,获得更区分性的补丁表征;(ii) 学习无监督视觉分类器以缩小视觉-语言模态差距。随后引入自适应聚合模块,融合补丁得分生成最终多标签结果。整个流程无需梯度更新或参数微调。实验表明,该方法在计算开销极小的前提下显著提升性能,在挑战性数据集NUS-WIDE上相较代表性基线实现超过6%的mAP提升。
原文摘要 · Abstract (English)
Vision-Language Models such as CLIP exhibit strong zero-shot recognition capability by aligning images with textual concepts, yet they often underperform on multi-label recognition where multiple objects co-exist. A key bottleneck is that the [CLS] token, as a single global visual representation, is insufficient to faithfully encode diverse targets with varying scales, contexts, and co-occurrence patterns. To address this limitation, we present a new multi-label image recognition framework, termed PIAA, which formulates prediction as Patch-level Inference followed by Adaptive Aggregation. Specifically, we first enhance patch-wise predictions from two complementary perspectives: (i) mitigating semantic entanglement in the visual encoder to obtain more discriminative patch representations, and (ii) learning an unsupervised visual classifier to narrow the vision-language modality gap. We then introduce an adaptive aggregation module that consolidates patch-level scores into the final multi-label prediction. Notably, the entire pipeline is fully training-free, requiring no gradient updates or parameter fine-tuning. Experiments show that our method achieves strong improvements with minimal extra computation, exceeding a 6% mAP gain on the challenging NUS-WIDE benchmark over representative baselines. Code is available at https://github.com/akang-wang/PIAA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。