Multi-Classifier Adversarial Optimization for Active Learning
DOI:
https://doi.org/10.1609/aaai.v37i6.25932Keywords:
ML: Active Learning, CV: Object Detection & CategorizationAbstract
Active learning (AL) aims to find a better trade-off between labeling costs and model performance by consciously selecting more informative samples to label. Recently, adversarial approaches have emerged as effective solutions. Most of them leverage generative adversarial networks to align feature distributions of labeled and unlabeled data, upon which discriminators are trained to better distinguish between them. However, these methods fail to consider the relationship between unlabeled samples and decision boundaries, and their training processes are often complex and unstable. To this end, this paper proposes a novel adversarial AL method, namely multi-classifier adversarial optimization for active learning (MAOAL). MAOAL employs task-specific decision boundaries for data alignment while selecting the most informative samples to label. To fulfill this, we introduce a novel classifier class confusion (C3) metric, which represents the classifier discrepancy as the inter-class correlation of classifier outputs. Without any additional hyper-parameters, the C3 metric further reduces the negative impacts of ambiguous samples in the process of distribution alignment and sample selection. More concretely, the network is trained adversarially by adding two auxiliary classifiers, reducing the distribution bias of labeled and unlabeled samples by minimizing the C3 loss between classifiers, while learning tighter decision boundaries and highlighting hard samples by maximizing the C3 loss. Finally, the unlabeled samples with the highest C3 loss are selected to label. Extensive experiments demonstrate the superiority of our approach over state-of-the-art AL methods in terms of image classification and object detection.Downloads
Published
2023-06-26
How to Cite
Geng, L., Liu, N., & Qin, J. (2023). Multi-Classifier Adversarial Optimization for Active Learning. Proceedings of the AAAI Conference on Artificial Intelligence, 37(6), 7687-7695. https://doi.org/10.1609/aaai.v37i6.25932
Issue
Section
AAAI Technical Track on Machine Learning I