Conditional Consistency Regularization for Semi-Supervised Multi-Label Image Classification

被引:2
|
作者
Wu, Zhengning [1 ]
He, Tianyu [2 ]
Xia, Xiaobo [1 ]
Yu, Jun [3 ]
Shen, Xu [4 ]
Liu, Tongliang [1 ]
机构
[1] Univ Sydney, Sydney AI Ctr, Darlington, NSW 2006, Australia
[2] Microsoft Res Asia MSRA, Beijing 100080, Peoples R China
[3] Univ Sci & Technol China, Dept Automat, Hefei, Peoples R China
[4] DAMO Acad, Alibaba Grp, Hangzhou 311121, Peoples R China
基金
澳大利亚研究理事会;
关键词
Predictive models; Image classification; Data models; Motorcycles; Manifolds; Training; Deep learning; semi-supervised multi-label learning; consistency regularization;
D O I
10.1109/TMM.2023.3324132
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
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 article, 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.
引用
收藏
页码:4206 / 4216
页数:11
相关论文
共 50 条
  • [1] Semi-supervised partial multi-label classification via consistency learning
    Tan, Anhui
    Liang, Jiye
    Wu, Wei-Zhi
    Zhang, Jia
    [J]. PATTERN RECOGNITION, 2022, 131
  • [2] Discrete Semi-supervised Multi-label Learning for Image Classification
    Xie, Liang
    He, Lang
    Shu, Haohao
    Hu, Shengyuan
    [J]. ADVANCES IN MULTIMEDIA INFORMATION PROCESSING, PT I, 2018, 11164 : 808 - 818
  • [3] Robust Multi-Label Semi-Supervised Classification
    Li, Sheng
    Fu, Yun
    [J]. 2017 IEEE INTERNATIONAL CONFERENCE ON BIG DATA (BIG DATA), 2017, : 27 - 36
  • [4] Semi-supervised imbalanced multi-label classification with label propagation
    Du, Guodong
    Zhang, Jia
    Zhang, Ning
    Wu, Hanrui
    Wu, Peiliang
    Li, Shaozi
    [J]. PATTERN RECOGNITION, 2024, 150
  • [5] Semi-Supervised Dimension Reduction for Multi-label Classification
    Qian, Buyue
    Davidson, Ian
    [J]. PROCEEDINGS OF THE TWENTY-FOURTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE (AAAI-10), 2010, : 569 - 574
  • [6] Semi-supervised multi-label image classification based on nearest neighbor editing
    Wei, Zhihua
    Wang, Hanli
    Zhao, Rui
    [J]. NEUROCOMPUTING, 2013, 119 : 462 - 468
  • [7] Semi-supervised robust deep neural networks for multi-label image classification
    Cevikalp, Hakan
    Benligiray, Burak
    Gerek, Omer Nezih
    [J]. PATTERN RECOGNITION, 2020, 100
  • [8] A survey of multi-label classification based on supervised and semi-supervised learning
    Han, Meng
    Wu, Hongxin
    Chen, Zhiqiang
    Li, Muhang
    Zhang, Xilong
    [J]. INTERNATIONAL JOURNAL OF MACHINE LEARNING AND CYBERNETICS, 2023, 14 (03) : 697 - 724
  • [9] A survey of multi-label classification based on supervised and semi-supervised learning
    Meng Han
    Hongxin Wu
    Zhiqiang Chen
    Muhang Li
    Xilong Zhang
    [J]. International Journal of Machine Learning and Cybernetics, 2023, 14 : 697 - 724
  • [10] Semi-supervised multi-label classification using incomplete label information
    Tan, Qiaoyu
    Yu, Yanming
    Yu, Guoxian
    Wang, Jun
    [J]. NEUROCOMPUTING, 2017, 260 : 192 - 202