arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1902.09720cs.CV
收藏

Learning a Deep ConvNet for Multi-label Classification with Partial Labels

Borealis AI(北极光人工智能);Simon Fraser University(西蒙弗雷泽大学)

Thibaut Durand, Nazanin Mehrasa, Greg Mori

英文摘要:

Deep ConvNets have shown great performance for single-label image classification (e.g. ImageNet), but it is necessary to move beyond the single-label classification task because pictures of everyday life are inherently multi-label. Multi-label classification is a more difficult task than single-label classification because both the input images and output label spaces are more complex. Furthermore, collecting clean multi-label annotations is more difficult to scale-up than single-label annotations. To reduce the annotation cost, we propose to train a model with partial labels i.e. only some labels are known per image. We first empirically compare different labeling strategies to show the potential for using partial labels on multi-label datasets. Then to learn with partial labels, we introduce a new classification loss that exploits the proportion of known labels per example. Our approach allows the use of the same training settings as when learning with all the annotations. We further explore several curriculum learning based strategies to predict missing labels. Experiments are performed on three large-scale multi-label datasets: MS COCO, NUS-WIDE and Open Images.

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

↑