Multilabel Image Classification via Feature/Label Co-Projection

TitleMultilabel Image Classification via Feature/Label Co-Projection
Publication TypeJournal Article
Year of Publication2021
AuthorsS Wen, W Liu, Y Yang, P Zhou, Z Guo, Z Yan, Y Chen, and T Huang
JournalIeee Transactions on Systems, Man, and Cybernetics: Systems
Volume51
Start Page7250
Issue11
Pagination7250 - 7259
Date Published11/2021
Abstract

This article presents a simple and intuitive solution for multilabel image classification, which achieves the competitive performance on the popular COCO and PASCAL VOC benchmarks. The main idea is to capture how humans perform this task: We recognize both labels (i.e., objects and attributes) and the correlation of labels at the same time. Here, label recognition is performed by a standard ConvNet pipeline, whereas label correlation modeling is done by projecting both labels and image features extracted by the ConvNet to a common latent vector space. Specifically, we carefully design the loss function to ensure that: 1) labels and features that co-appear frequently are close to each other in the latent space and 2) conversely, labels/features that do not appear together are far apart. This information is then combined with the original ConvNet outputs to form the final prediction. The whole model is trained end-to-end, with no additional supervised information other than the image-level supervised information. Experiments show that the proposed method consistently outperforms previous approaches on COCO and PASCAL VOC in terms of mAP, macro/micro precision, recall, and F-measure. Further, our model is highly efficient at test time, with only a small number of additional weights compared to the base model for direct label recognition.

DOI10.1109/TSMC.2020.2967071
Short TitleIeee Transactions on Systems, Man, and Cybernetics: Systems