Active Learning Method Based on Pseudo-labeling

被引:0
作者
Hou, Xiaonan [1 ]
Wang, Chunlei [1 ]
机构
[1] China Elect Technol Grp Corp, Res Inst 21, Shanghai, Peoples R China
来源
2024 8TH INTERNATIONAL CONFERENCE ON ROBOTICS, CONTROL AND AUTOMATION, ICRCA 2024 | 2024年
关键词
object detection; active learning; pseudo-labeling;
D O I
10.1109/ICRCA60878.2024.10649264
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
As an important fundamental task in the field of computer vision, object detection has a wide range of applications. The performance of object detection not only depends on excellent network structures but also requires the support of large and rich annotated datasets. However, annotating data requires a considerable amount of personnel and funding, and the quality of the annotated data greatly affects the accuracy of object detection algorithms. In order to solve the problem of labeling cost, researchers, we proposed a multi-scale feature fusion method is designed to fuse and reduce the dimensionality of the multi-scale features of unlabeled samples, while retaining the main features of the samples and reducing computational burden. Clustering is used to apply pseudo-label classification to unlabeled samples, and information entropy is introduced to calculate the uncertainty of unlabeled samples, considering both uncertainty and diversity of the unlabeled samples. The experiment results on Pascal VOC data set and MS COCO data set prove that our method is equivalent to the latest methods. The ablation experiment proves that our method can effectively improve the labeling efficiency and can achieve the effect that would otherwise be achieved in a larger data set.
引用
收藏
页码:453 / 458
页数:6
相关论文
共 13 条
  • [1] The power of ensembles for active learning in image classification
    Beluch, William H.
    Genewein, Tim
    Nuernberger, Andreas
    Koehler, Jan M.
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 9368 - 9377
  • [2] Active Learning for Deep Object Detection via Probabilistic Modeling
    Choi, Jiwoong
    Elezi, Ismail
    Lee, Hyuk-Jae
    Farabet, Clement
    Alvarez, Jose M.
    [J]. 2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 10244 - 10253
  • [3] Gal Y, 2017, PR MACH LEARN RES, V70
  • [4] Gal Y, 2016, PR MACH LEARN RES, V48
  • [5] Guo J, 2021, P IEEE CVF INT C COM, P2896
  • [6] Houlsby N, 2011, Arxiv, DOI arXiv:1112.5745
  • [7] Joshi AJ, 2009, PROC CVPR IEEE, P2364
  • [8] Kirsch A, 2019, ADV NEUR IN, V32
  • [9] Active Learning by Feature Mixing
    Parvaneh, Amin
    Abbasnejad, Ehsan
    Teney, Damien
    Haffari, Reza
    van den Hengel, Anton
    Shi, Javen Qinfeng
    [J]. 2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, : 12227 - 12236
  • [10] Settles B., 2008, P 2008 C EMP METH NA, P1070