Exploiting ensemble learning for automatic cataract detection and grading

被引:108
作者
Yang, Ji-Jiang [1 ]
Li, Jianqiang [2 ]
Shen, Ruifang [1 ]
Zeng, Yang [3 ]
He, Jian [2 ]
Bi, Jing [2 ]
Li, Yong [2 ]
Zhang, Qinyan [3 ]
Peng, Lihui [1 ]
Wang, Qing [4 ]
机构
[1] Tsinghua Univ, Tsinghua Natl Lab Informat Sci & Technol, Beijing 100084, Peoples R China
[2] Beijing Univ Technol, Sch Software Engn, Beijing, Peoples R China
[3] Beijing Univ Posts & Telecommun, Automat Sch, Beijing 100876, Peoples R China
[4] Tsinghua Univ, Res Inst Applicat Technol Wuxi, Suzhou, Jiangsu, Peoples R China
基金
北京市自然科学基金;
关键词
Cataract detection; Fundus image classification; Ensemble learning; Support vector machines; Neural network; VESSEL SEGMENTATION; IMAGES; CLASSIFICATION; SYSTEM;
D O I
10.1016/j.cmpb.2015.10.007
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Cataract is defined as a lenticular opacity presenting usually with poor visual acuity. It is one of the most common causes of visual impairment worldwide. Early diagnosis demands the expertise of trained healthcare professionals, which may present a barrier to early intervention due to underlying costs. To date, studies reported in the literature utilize a single learning model for retinal image classification in grading cataract severity. We present an ensemble learning based approach as a means to improving diagnostic accuracy. Three independent feature sets, i.e., wavelet-, sketch-, and texture-based features, are extracted from each fundus image. For each feature set, two base learning models, i.e., Support Vector Machine and Back Propagation Neural Network, are built. Then, the ensemble methods, majority voting and stacking, are investigated to combine the multiple base learning models for final fundus image classification. Empirical experiments are conducted for cataract detection (two-class task, i.e., cataract or non-cataractous) and cataract grading (four-class task, i.e., non-cataractous, mild, moderate or severe) tasks. The best performance of the ensemble classifier is 93.2% and 84.5% in terms of the correct classification rates for cataract detection and grading tasks, respectively. The results demonstrate that the ensemble classifier outperforms the single learning model significantly, which also illustrates the effectiveness of the proposed approach. (C) 2015 Elsevier Ireland Ltd. All rights reserved.
引用
收藏
页码:45 / 57
页数:13
相关论文
共 52 条
[1]   Fourier analysis of digital retinal images in estimation of cataract severity [J].
Abdul-Rahman, Anmar M. ;
Molteno, Tim ;
Molteno, Anthony C. B. .
CLINICAL AND EXPERIMENTAL OPHTHALMOLOGY, 2008, 36 (07) :637-645
[2]  
Abramoff M.D., 2005, IEEE T MED IMAGING, V24
[3]  
Abramoff Michael D, 2010, IEEE Rev Biomed Eng, V3, P169, DOI 10.1109/RBME.2010.2084567
[4]   Cataract and surgery for cataract [J].
Allen, David ;
Vasavada, Abhay .
BMJ-BRITISH MEDICAL JOURNAL, 2006, 333 (7559) :128-132
[5]  
Cherkauer K.J., 1996, WORKING NOTES AAAI W, P15
[6]  
Chow Yew Chung, 2011, ENG MED BIOL SOC EMB
[7]  
Chrastek R., 2005, MED IMAGE ANAL
[8]   THE LENS OPACITIES CLASSIFICATION SYSTEM-III [J].
CHYLACK, LT ;
WOLFE, JK ;
SINGER, DM ;
LESKE, MC ;
BULLIMORE, MA ;
BAILEY, IL ;
FRIEND, J ;
MCCARTHY, D ;
WU, SY .
ARCHIVES OF OPHTHALMOLOGY, 1993, 111 (06) :831-836
[9]  
David J., 2008, 2008 CISP 08 C IEEE, V2, P49
[10]  
Fausett L.V., 1994, Fundamentals of Neural Networks: Architectures, Algorithms and Applications