Automated classification of choroidal neovascularization, diabetic macular edema, and drusen from retinal OCT images using vision transformers: a comparative study

Abstract

Classifying retinal diseases is a complex problem because the early problematic areas of retinal disorders are quite small and conservative. In recent years, Transformer architectures have been successfully applied to solve various retinal related health problems. Age-related macular degeneration (AMD) and diabetic macular edema (DME), two prevalent retinal diseases, can cause partial or total blindness. Diseases therefore require an early and accurate detection. In this study, we proposed Vision Transformer (ViT), Tokens-To-Token Vision Transformer (T2T-ViT) and Mobile Vision Transformer (Mobile-ViT) algorithms to detect choroidal neovascularization (CNV), drusen, and diabetic macular edema (DME), and normal using optical coherence tomography (OCT) images. The predictive accuracies of ViT, T2T-ViT and Mobile-ViT achieved on the dataset for the classification of OCT images are 95.14%, 96.07% and 99.17% respectively. Experimental results obtained from ViT approaches showed that Mobile-ViT have superior performance with regard to classification accuracy in comparison with the others. Overall, it has been observed that ViT architectures have the capacity to classify with high accuracy in the diagnosis of retinal diseases.