Vessel Density Features of Optical Coherence Tomography Angiography for Classification of Optic Neuropathies Using Machine Learning

Abstract

Background: To evaluate the classification performance of machine learning based on the 4 vessel density features of peripapillary optical coherence tomography angiography (OCT-A) for classifying healthy, nonarteritic anterior ischemic optic neuropathy (NAION), and optic neuritis (ON) eyes.

Methods: Forty-five eyes of 45 NAION patients, 32 eyes of 32 ON patients, and 76 eyes of 76 healthy individuals with optic nerve head OCT-A were included. Four vessel density features of OCT-A images were developed using a threshold-based segmentation method and were integrated in 3 models of machine learning classifiers. Classification performances of support vector machine (SVM), random forest, and Gaussian Naive Bayes (GNB) models were evaluated with the area under the receiver-operating-characteristic curve (AUC) and accuracy.

Results: We divided 121 images into a 70% training set and 30% test set. For ON-NAION classification, best results were achieved with 50% threshold, in which 3 classifiers (SVM, RF, and GNB) discriminated ON from NAION with an AUC of 1 and accuracy of 1. For ON-Normal classification, with 100% threshold, SVM and RF classifiers were able to discriminate normal from ON with AUCs of 1 and accuracies of 1. For NAION-normal classification, with 50% threshold, the SVM and RF classified the NAION from normal with AUC and accuracy of 1.

Conclusions: ML based on the combined peripapillary vessel density features of total vessels and capillaries in the whole image and ring image could provide excellent performance for NAION and ON distinction.