Screening and grading of fundus images of diabetic retinopathy based on visual attention
10.3760/cma.j.issn.2095-0160.2019.08.008
- VernacularTitle:应用视觉注意力的糖尿病视网膜病变眼底影像筛查及分级
- Author:
Jialong WAN
1
;
Jianbin HU
;
Weidong JIN
;
Peng TANG
Author Information
1. 西南交通大学电气工程学院
- Keywords:
Diabetic retinopathy;
Visual attention;
Convolutional neural network;
Attention module;
Grading
- From:
Chinese Journal of Experimental Ophthalmology
2019;37(8):630-637
- CountryChina
- Language:Chinese
-
Abstract:
Objective To construct an intelligent analysis system based on visual attention for diabetic retinopathy ( DR) assistant diagnosis and to realize the automatic screening and grading of fundus images of DR. Methods Total of 35126 DR fundus images were downloaded from the Diabetic Retinopathy Detection competition in the Data Modeling and Data Analysis Competition Platform (Kaggle),and 1200 fundus images were downloaded from the Messidor website. Firstly,according to the characteristics of DR fundus images,a series of preprocessing was carried out for retina images. Then,on the basis of VGG16 network,visual attention SENet module was introduced to improve the saliency of lesion features,and a deep convolution neural network SEVGG with complex network structure was generated. The network basically inherited some structural parameters of VGG16,and the parameters of SENet module were adjusted according to the basic network and training data set. Finally, the SEVGG network model was used to screen the DR fundus image,and the fundus image was divided into different levels according to the degree of lesions of DR in different periods. Configure the training platform and environment and perform algorithm performance verification experiments. Results The method proposed in this study was tested on different open standard datasets,and finally achieved high accuracy in image-based classification. The accuracy of 5 classification in Kaggle dataset was 83％,the sensitivity of lesion detection was 99. 86％ and the specificity was 99. 63％. The accuracy rate of the 4 classification in the Messidor data set was up to 88％,the sensitivity of the lesion detection was 98. 17％,and the specificity was 96. 39％. The introduction of visual attention was more significant for the focus of the lesion,which helped the accurate detection of DR. Conclusions This method effectively avoids some shortcomings of traditional artificial feature extraction and fundus image classification,and is more accurate for lesion recognition. It is not only superior to the previous method,but also shows better robustness and generalization.
