Laser & Optoelectronics Progress, Volume. 59, Issue 6, 0617017(2022)
Automatic Detection of Retinal Diseases Based on Lightweight Convolutional Neural Network
One major method for detecting retinopathy in clinics is optical coherence tomography. However, this manual diagnostic model is affected by strong subjectivity and low efficiency. Therefore, this paper proposes a lightweight convolutional neural network for the automatic detection of retinopathy. The proposed network consists of two modules. The first module combines atrous convolutions and depth wise separable convolutions to reduce the number of parameters; the second module uses the decomposition convolution method to extend the depth by decomposing the conventional convolution layer into multilayer asymmetric convolution. Both modules are combined to form a feature extractor, and the Softmax function is used as the classifier to obtain a lightweight model with 44 layers deep and 9.2 MB parameters. The accuracy, sensitivity, specificity, and area under the receiver operating characteristic curve of the proposed network on the test set are 0.980, 0.954, 0.987, and 0.997, respectively. The visualization results show that the diagnostic basis of the model is consistent with that of ophthalmologists. These results show that the proposed network can accurately automate retinal disease detection.
Get Citation
Copy Citation Text
Lingxiao Wang, Jun Yang, Wensai Wang, Ting Li. Automatic Detection of Retinal Diseases Based on Lightweight Convolutional Neural Network[J]. Laser & Optoelectronics Progress, 2022, 59(6): 0617017
Category: Medical Optics and Biotechnology
Received: Jan. 13, 2022
Accepted: Feb. 10, 2022
Published Online: Mar. 8, 2022
The Author Email: Ting Li (liting@bme.cams.cn)