English

Cost-Effective Active Learning for Deep Image Classification

Computer Vision and Pattern Recognition 2017-01-16 v1

Abstract

Recent successes in learning-based image classification, however, heavily rely on the large number of annotated training samples, which may require considerable human efforts. In this paper, we propose a novel active learning framework, which is capable of building a competitive classifier with optimal feature representation via a limited amount of labeled training instances in an incremental learning manner. Our approach advances the existing active learning methods in two aspects. First, we incorporate deep convolutional neural networks into active learning. Through the properly designed framework, the feature representation and the classifier can be simultaneously updated with progressively annotated informative samples. Second, we present a cost-effective sample selection strategy to improve the classification performance with less manual annotations. Unlike traditional methods focusing on only the uncertain samples of low prediction confidence, we especially discover the large amount of high confidence samples from the unlabeled set for feature learning. Specifically, these high confidence samples are automatically selected and iteratively assigned pseudo-labels. We thus call our framework "Cost-Effective Active Learning" (CEAL) standing for the two advantages.Extensive experiments demonstrate that the proposed CEAL framework can achieve promising results on two challenging image classification datasets, i.e., face recognition on CACD database [1] and object categorization on Caltech-256 [2].

Keywords

Cite

@article{arxiv.1701.03551,
  title  = {Cost-Effective Active Learning for Deep Image Classification},
  author = {Keze Wang and Dongyu Zhang and Ya Li and Ruimao Zhang and Liang Lin},
  journal= {arXiv preprint arXiv:1701.03551},
  year   = {2017}
}

Comments

Accepted by IEEE Transactions on Circuits and Systems for Video Technology (TCSVT) 2016