Prediction of the ectasia screening index from raw Casia2 volume data for keratoconus identification by using convolutional neural networks

Abstract

Purpose Prediction of the ectasia screening index, an estimator provided by the Casia2 instrument for identifying keratoconus, from raw optical coherence tomography data using convolutional neural networks. Methods Three convolutional neural networks models (ResNet18, DenseNet121 and EfficientNetB0) were employed to predict the ectasia screening index. Mean absolute error was used as the performance metric for predicting the ectasia screening index by the adapted convolutional neural network models on the test set. Scans with an ectasia screening index value higher than a certain threshold were classified as Keratoconus, while the remaining scans were classified as Not Keratoconus. The architectures’ performance was evaluated using metrics such as accuracy, sensitivity, specificity, positive predictive value and F1 score on data collected from patients examined at the eye clinic of the Homburg University Hospital. The raw data from the Casia2 instrument, in 3dv format, was converted into 16 images per examination of one eye. For the training, validation and testing phases, 3689, 1050 and 1078 scans (3dv files) were selected, respectively. Results In the prediction of the ectasia screening index, the mean absolute error values for the adapted ResNet18, the adapted DenseNet121 and the adapted EfficientNetB0, rounded to two decimal places, were 7.15, 6.64 and 5.86, respectively. In the classification task, the three networks yielded an accuracy of 94.80%, 95.27% and 95.83%, respectively; a sensitivity of 92.07%, 94.64% and 94.17%, respectively; a specificity of 96.61%, 95.69% and 96.92%, respectively; a positive predictive value of 94.72%, 93.55% and 95.28%, respectively; and a F1 score of 93.38%, 94.09% and 94.72%, respectively. Conclusions Our results show that the prediction of keratoconus based on the ectasia screening index values estimated from raw data outperforms previous approaches using processed data. adapted EfficientNetB0 outperformed both the other adapted models and those in state-of-the-art studies, with the highest accuracy and F1 score.