arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2409.08381cs.CVcs.AIcs.CLcs.LGcs.MM

重新思考带部分标注的多标签识别的提示策略

Rethinking Prompting Strategies for Multi-Label Recognition with Partial Annotations

University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校)

Samyak Rawlekar, Shubhang Bhatnagar, Narendra Ahuja

AI总结:

针对带部分标注的多标签识别,研究发现VLM提示学习中负向提示会降低性能,仅学正向提示结合学习型负向嵌入的PositiveCoOp优于双提示方法,且低缺失标签比例下纯视觉基线性能可比双提示法,计算和参数成本更低。

摘要:

诸如CLIP之类的视觉语言模型(VLM)已通过提示学习被应用于带部分标注的多标签识别(MLR),该方法为每个类别学习正向和负向提示,以在共享的视觉-文本特征空间中将其嵌入与类别存在或不存在相关联。虽然这种方法借助VLM先验提升了MLR性能,但我们假设学习负向提示可能并非最优,因为用于训练VLM的数据集缺少明确聚焦于类别不存在的图像-字幕对。为分析正向和负向提示学习对MLR的影响,我们提出PositiveCoOp和NegativeCoOp,其中仅一种提示在VLM引导下学习,另一种则被直接在共享特征空间中学习的嵌入向量替代,无需依赖文本编码器。通过实证分析,我们观察到负向提示会降低MLR性能,仅学习正向提示并结合学习到的负向嵌入(PositiveCoOp)的效果优于双提示学习方法。此外,我们量化了提示学习相较于仅使用视觉特征的简单基线的性能优势,发现当缺失标签比例较低时,该基线表现出与双提示学习方法(DualCoOp)相当的强劲性能,同时仅需一半的训练计算量和1/16的参数数量。

英文摘要:

Vision-language models (VLMs) like CLIP have been adapted for Multi-Label Recognition (MLR) with partial annotations by leveraging prompt-learning, where positive and negative prompts are learned for each class to associate their embeddings with class presence or absence in the shared vision-text feature space. While this approach improves MLR performance by relying on VLM priors, we hypothesize that learning negative prompts may be suboptimal, as the datasets used to train VLMs lack image-caption pairs explicitly focusing on class absence. To analyze the impact of positive and negative prompt learning on MLR, we introduce PositiveCoOp and NegativeCoOp, where only one prompt is learned with VLM guidance while the other is replaced by an embedding vector learned directly in the shared feature space without relying on the text encoder. Through empirical analysis, we observe that negative prompts degrade MLR performance, and learning only positive prompts, combined with learned negative embeddings (PositiveCoOp), outperforms dual prompt learning approaches. Moreover, we quantify the performance benefits that prompt-learning offers over a simple vision-features-only baseline, observing that the baseline displays strong performance comparable to dual prompt learning approach (DualCoOp), when the proportion of missing labels is low, while requiring half the training compute and 16 times fewer parameters

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

↑