Cost-Effective Active Learning for Deep Image Classification

被引:426
作者
Wang, Keze [1 ,2 ]
Zhang, Dongyu [1 ,2 ]
Li, Ya [3 ]
Zhang, Ruimao [1 ,2 ]
Lin, Liang [1 ,2 ]
机构
[1] Sun Yat Sen Univ, Sch Data & Comp Sci, Guangzhou 510006, Guangdong, Peoples R China
[2] Natl Univ Def Technol, Collaborat Innovat Ctr High Performance Comp, Changsha 410073, Hunan, Peoples R China
[3] Guangzhou Univ, Guangzhou 510182, Guangdong, Peoples R China
基金
中国国家自然科学基金;
关键词
Active learning (AL); deep neural nets; image classification; incremental learning;
D O I
10.1109/TCSVT.2016.2589879
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Recent successes in learning-based image classification, however, heavily rely on the large number of annotated training samples, which may require considerable human effort. In this paper, we propose a novel active learning (AL) 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 AL methods in two aspects. First, we incorporate deep convolutional neural networks into AL. 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 pseudolabels. We thus call our framework cost-effective AL (CEAL) standing for the two advantages. Extensive experiments demonstrate that the proposed CEAL framework can achieve promising results on two challenging image classification data sets, i.e., face recognition on the cross-age celebrity face recognition data set database and object categorization on Caltech-256.
引用
收藏
页码:2591 / 2600
页数:10
相关论文
共 42 条
[1]  
[Anonymous], P 31 INT C MACH LEAR
[2]  
[Anonymous], P ICML
[3]  
[Anonymous], P ICML
[4]  
[Anonymous], 2015, AAAI C ARTIF INTELL, DOI DOI 10.1109/IJCNN.2015.7280498
[5]  
Bengio Y., 2009, P 26 ANN INT C MACHI, P41, DOI DOI 10.1145/1553374.1553380
[6]   Minimax bounds for active learning [J].
Castro, Rui M. ;
Nowak, Robert D. .
IEEE TRANSACTIONS ON INFORMATION THEORY, 2008, 54 (05) :2339-2353
[7]  
Chen BC, 2014, LECT NOTES COMPUT SC, V8694, P768, DOI 10.1007/978-3-319-10599-4_49
[8]  
Ciresan D, 2012, PROC CVPR IEEE, P3642, DOI 10.1109/CVPR.2012.6248110
[9]   A Novel Active Learning Method in Relevance Feedback for Content-Based Remote Sensing Image Retrieval [J].
Demir, Beguem ;
Bruzzone, Lorenzo .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2015, 53 (05) :2323-2334
[10]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848