Conditional Consistency Regularization for Semi-Supervised Multi-label Image Classification
Zhengning Wu,Tianyu He,Xiaobo Xia,Jun Yu,Xu Shen,Tongliang Liu
DOI: https://doi.org/10.1109/tmm.2023.3324132
IF: 7.3
2023-01-01
IEEE Transactions on Multimedia
Abstract:Consistency regularization has achieved great successes in Semi-Supervised Single-Label Image Classification (SS-SLC) with deep learning models, while few effort has been devoted to Semi-Supervised Multi-Label Image Classification (SS-MLC) with deep learning models. One intuitive solution for introducing consistency regularization to SS-MLC is to regularize model predictions to be invariant to different augmented data of the same input image. However, the solution lacks the consideration of label relations, which are key elements in multi-label image classification. In this paper, we go beyond the consistency regularization for multi-view input images, and propose Conditional Consistency Regularization (CCR) that is tailored for SS-MLC. Specifically, for two augmented input images, we make the two model predictions conditioned on different label states (i.e., positive, negative, or unknown for each class). By encouraging the two predictions to be consistent, the model is able to build relations between the given two different label states, which helps to make use of label relations for boosting image classification. The experiments on large-scale real-world SS-MLC benchmarks demonstrate that the proposed method can surpass state-of-the-art methods by a large margin.
computer science, information systems,telecommunications, software engineering