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

基于课程式消歧的偏多标签图像分类深度模型

A Deep Model for Partial Multi-Label Image Classification with Curriculum Based Disambiguation

College of Computer Science and Technology, Nanjing University of Aeronautics and Astronautics(南京航空航天大学计算机科学与技术学院);MIIT Key Laboratory of Pattern Analysis and Machine Intelligence(工业和信息化部模式分析与机器智能重点实验室)

Feng Sun, Ming-Kun Xie, Sheng-Jun Huang

AI总结:

本文首次提出面向偏多标签图像分类的深度模型,通过课程式消歧策略逐步识别真实标签,并引入一致性正则化平衡简单标签拟合与潜在相关标签挖掘,在常用基准数据集上显著超越现有最优方法。

摘要:

本文研究偏多标签(PML)图像分类问题,其中每张图像被标注一个候选标签集,该集合包含多个相关标签以及其他噪声标签。现有的PML方法通常设计一种消歧策略,通过利用带有额外假设的先验知识来过滤噪声标签,但不幸的是,这些先验知识在许多真实任务中并不可用。此外,由于消歧的目标函数通常是在整个训练集上精心设计的,因此很难在基于小批量的SGD深度模型中进行优化。本文首次提出了一种用于PML的深度模型,以增强表示能力和判别能力。一方面,我们提出了一种新颖的基于课程式的消歧策略,通过结合不同类别之间不同的难度,逐步识别真实标签。另一方面,引入一致性正则化用于模型重训练,以平衡拟合已识别的简单标签和挖掘潜在相关标签之间的关系。在常用基准数据集上的大量实验结果表明,所提出的方法显著优于SOTA方法。

英文摘要:

In this paper, we study the partial multi-label (PML) image classification problem, where each image is annotated with a candidate label set consists of multiple relevant labels and other noisy labels. Existing PML methods typically design a disambiguation strategy to filter out noisy labels by utilizing prior knowledge with extra assumptions, which unfortunately is unavailable in many real tasks. Furthermore, because the objective function for disambiguation is usually elaborately designed on the whole training set, it can be hardly optimized in a deep model with SGD on mini-batches. In this paper, for the first time we propose a deep model for PML to enhance the representation and discrimination ability. On one hand, we propose a novel curriculum based disambiguation strategy to progressively identify ground-truth labels by incorporating the varied difficulties of different classes. On the other hand, a consistency regularization is introduced for model retraining to balance fitting identified easy labels and exploiting potential relevant labels. Extensive experimental results on the commonly used benchmark datasets show the proposed method significantly outperforms the SOTA methods.

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

↑