Laser & Optoelectronics Progress, Volume. 61, Issue 22, 2237002(2024)

Adversarial Active Learning Method for Enhanced Object Detection Guided by Uncertainty

Shaiya Wang1, Hongzhong Tang1,2、*, Haifan Luo1, Shang Gao1, and Yuebing Xu3
Author Affiliations
  • 1College of Automation and Electronic Information, Xiangtan University, Xiangtan 411105, Hunan , China
  • 2The Key Laboratory of Intelligent Computing and Information Processing of Ministry of Education, Xiangtan University, Xiangtan 411105, Hunan , China
  • 3College of Physics and Electronic Engineering, Hengyang Normal University, Hengyang 421002, Hunan , China
  • show less

    Object detection based on active learning typically utilizes limited labeled data to enhance detection model performance. This method allows learners to select valuable samples from a large pool of unlabeled data for manual labeling and to iteratively train and optimize the model. However, existing object detection methods that use active learning often struggle to effectively balance sample uncertainty and diversity, which results in high redundancy of query samples. To address this issue, we propose an adversarial active learning method guided by uncertainty for object detection. First, we introduce a loss prediction module to evaluate the uncertainty of unlabeled samples. This uncertainty guides the adversarial network training and helps construct a query sample set that includes both uncertainty and diversity. Second, we evaluate sample diversity based on feature similarity to reduce redundancy of query samples. Finally, experimental results on the MS COCO and Pascal VOC datasets using multiple detection frameworks demonstrate that the proposed method can effectively improve object detection accuracy with fewer annotations.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Shaiya Wang, Hongzhong Tang, Haifan Luo, Shang Gao, Yuebing Xu. Adversarial Active Learning Method for Enhanced Object Detection Guided by Uncertainty[J]. Laser & Optoelectronics Progress, 2024, 61(22): 2237002

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Digital Image Processing

    Received: Jan. 29, 2024

    Accepted: Mar. 12, 2024

    Published Online: Nov. 19, 2024

    The Author Email: Hongzhong Tang (HongzhongTang@xtu.edu.cn)

    DOI:10.3788/LOP240632

    CSTR:32186.14.LOP240632

    Topics