KeratoEL: Detection of keratoconus using corneal parameters with ensemble learning

Background and aims: Keratoconus is a progressive eye condition in which the normally round cornea thins and bulges outwards into a cone shape. This irregular shape causes light to scatter in multiple directions as it enters the eye, leading to distorted vision, increased sensitivity to light and frequent changes in the prescription of glasses or contact lenses. Detecting keratoconus at an early stage is not only difficult but also challenging.

Methods: The study has proposed an ensemble-based machine learning (ML) technique named KeratoEL to detect keratoconus at an early stage. The proposed KeratoEL model combines the basic machine learning algorithms, namely support vector machine (SVM), decision tree (DT), random forest (RF) and artificial neural network (ANN). Before employing the ML model for keratoconus detection, the data set is first preprocessed manually by eliminating some features that don't contribute any significant value to predict the exact class. Moreover, the output features are labelled into three different classes and Extra Trees Classifier is used to find out the important features. Then, the features are sorted in descending order and top 45, 30, and 15 features are taken as input datasets against the output. Finally, different machine learning models are tested using the input datasets and performance metrics are measured.

Results: The proposed model obtains 98.0%, 98.9% and 99.8% accuracy for top 45, 30, and 15 number of features respectively. Overall experimental results show that the proposed ensemble model outperforms the existing machine learning models.

Conclusion: The proposed KeratoEL model effectively detects keratoconus at an early stage by combining SVM, DT, RF, and ANN algorithms, demonstrating superior performance over existing models. These results underscore the potential of the KeratoEL ensemble approach in enhancing early detection and treatment of keratoconus.