Machine learning approach to automatic exudate detection in retinal images from diabetic patients

被引:53
作者
Sopharak, Akara [1 ]
Dailey, Matthew N. [2 ]
Uyyanonvara, Bunyarit [1 ]
Barman, Sarah [3 ]
Williamson, Tom [4 ]
Nwe, Khine Thet [2 ]
Moe, Yin Aye [2 ]
机构
[1] Thammasat Univ, Sirindhorn Int Inst Technol, Bangkok, Thailand
[2] Asian Inst Technol, Bangkok, Thailand
[3] Kingston Univ, Digital Imaging Res Ctr, Kingston upon Thames KT1 2EE, Surrey, England
[4] St Thomas Hosp, London, England
关键词
exudate; diabetic retinopathy; naive Bayes classifier; support vector machine; nearest neighbour classifier; RETINOPATHY;
D O I
10.1080/09500340903118517
中图分类号
O43 [光学];
学科分类号
070207 ; 0803 ;
摘要
Exudates are among the preliminary signs of diabetic retinopathy, a major cause of vision loss in diabetic patients. Early detection of exudates could improve patients' chances to avoid blindness. In this paper, we present a series of experiments on feature selection and exudates classification using naive Bayes and support vector machine (SVM) classifiers. We first fit the naive Bayes model to a training set consisting of 15 features extracted from each of 115,867 positive examples of exudate pixels and an equal number of negative examples. We then perform feature selection on the naive Bayes model, repeatedly removing features from the classifier, one by one, until classification performance stops improving. To find the best SVM, we begin with the best feature set from the naive Bayes classifier, and repeatedly add the previously-removed features to the classifier. For each combination of features, we perform a grid search to determine the best combination of hyperparameters nu ( tolerance for training errors) and gamma ( radial basis function width). We compare the best naive Bayes and SVM classifiers to a baseline nearest neighbour (NN) classifier using the best feature sets from both classifiers. We find that the naive Bayes and SVM classifiers perform better than the NN classifier. The overall best sensitivity, specificity, precision, and accuracy are 92.28%, 98.52%, 53.05%, and 98.41%, respectively.
引用
收藏
页码:124 / 135
页数:12
相关论文
共 24 条