Research on Retinal OCT Image Classification Based on Deep Learning


  • Mingxin Wang



EfficientNet, Retina, Optical coherence tomography, Image classification, Transfer learning


Presently, there persists an issue in convolutional neural network (CNN)-based classification methods for retinal optical coherence tomography (OCT) images, particularly in discerning small-scale lesion areas. This dilemma results in neural network models exhibiting diminished accuracy when determining the activity of age-related macular degeneration (AMD) regarding its dry and wet stages, as well as choroidal neovascularization (CNV). Accurate identification of lesion types is paramount for ophthalmologists in devising treatment strategies. To address this challenge, a transfer learning-based EfficientNet retinal OCT image classification algorithm is proposed. Initially, retinal OCT images undergo data augmentation and preprocessing procedures. Subsequently, the pre-trained EfficientNet-B3 model is trained via transfer learning, followed by fine-tuning training using partial oversampling and class weighting techniques. The ultimate classification accuracy reaches 99.2%, signifying the model's commendable classification recognition accuracy. This underscores the clinical guidance significance of this research endeavor.


FERRIS III F L, WILKINSON C P, BIRD A, et al. Clinical classification of age-related macular degeneration [J]. Ophthalmology, 2013, 120(4): 844-851.

ZHANG Hao-rui, GUI Xiao, ZHAO Na, et al. Research progress on pharmacotherapy for diabetic macular edema [J]. Chinese Journal of Ophthalmology and Otorhinolaryngology, 2021, 21(3): 226-229.

HUANG D, SWANSON E A, LIN C P, et al. Optical coherence tomography [J]. Science, 1991, 254(5035): 1178-1181,

LIU X, BAI Y, CAO J, et al. Joint disease classification and lesion segmentation via one-stage attention-based convolutional neural network in OCT images [J]. Biomedical Signal Processing and Control, 2022, 71: 103087.

Ajes h F, Abraha m A. Detection and classification of age-related macular degeneration using integration of densenet169andconvolutional neural network[C]. Innovations in Bio-Inspired Computing and Applications: Proceedings of the 12th International Conference on Innovations in Bio-Inspired Computing Proceedings of the 12th International Conference on Innovations in Bio-Inspired Computing and Applications (IBICA 2021). Cham: Springer International Publishing, 2022: 226-238.

Sun Y, Zhang H, Yao X. Automatic diagnosis of macular diseases from OCT volume based on its two-dimensional feature map and convolutional neural network with attention mechanism[J]. Journal of Biomedical Optics, 2020, 25(9): 096004-096004.

Karri S P K, Chakraborty D, Chatterjee J. Transfer learning based classification of optical coherence tomography images with diabetic macular edema and dry age-related macular degeneration[J]. Biomedical Optics Express, 2017, 8(2): 579-592.

TAN M, LE Q V. Efficientnet: rethinking model scaling for convolutional neural networks [C]. International Conference on Machine Learning, 2019:6105-6114.

HE K, ZHANG X, REN S, et al. Deep residual learning for image recognition [C]. Proceedings of IEEE Conference on Computer Vision and Pattern Recognition, 2016: 770-778.







How to Cite

Wang, M. (2024). Research on Retinal OCT Image Classification Based on Deep Learning. Journal of Computing and Electronic Information Management, 12(2), 34-37.

Similar Articles

1-10 of 78

You may also start an advanced similarity search for this article.