arrow
Return

Deep active learning models for imbalanced image classification

delete2022-12-01
delete26
PRE
AI
Q
Qiuye Jin
M
Mingzhi Yuan
H
Haoran Wang
M
Manning Wang
宋志坚 (Zhijian Song) *
DOI:10.1016/j.knosys.2022.109817delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Active learning can query valuable samples in an unlabeled sample pool for annotation, thus building a more informative labeled dataset and reducing the annotation cost. However, traditional active learning methods are not effective in the task of imbalanced image classification for ignoring the distribution bias. In this study, we propose a Balanced Active Learning (BAL) method for imbalanced image classification. BAL estimates the probability of a sample belonging to minority or majority classes and compensates the annotation query for the minority classes, thus alleviating the class imbalance in the selected training samples. Experiments on three imbalanced image classification datasets, imbalanced CIFAR-10, ISIC2020, and Caltech256, showed that BAL achieved new state-of-the-art performance of active learning in a variety of classification tasks and different types of imbalance.(c) 2022 Elsevier B.V. All rights reserved.
Keywords:
Active learning
Class imbalance
Image classification

Journal

K
Knowledge-Based Systems
IF:
7.6
Papers:
1.2W
Citations:
4.5W

Organization

F
fudan university
Scholars:
11.7W
Papers: 7.7W
Citations: 121