Related Experiment Video
Updated: May 21, 2026
Combining Reflectance Confocal Microscopy with Optical Coherence Tomography for Noninvasive Diagnosis of Skin Cancers via Image Acquisition
Published on: August 18, 2022
Highly efficient stacking ensemble learning model for automated keratoconus screening
Zahra J Muhsin1, Rami Qahwaji2, Ibrahim Ghafir1
1Faculty of Engineering and Digital Technologies, University of Bradford, Bradford, BD7 1DP, UK.
Background:
Despite extensive research on keratoconus (KC) detection with traditional machine learning models, stacking ensemble learning approaches remain underexplored. This paper presents a stacking ensemble learning method to enhance automated KC screening.
Methods:
This study utilizes a clinical dataset containing detailed corneal data from 2491 cases classified as non-KC (NKC), subclinical KC (SCKC) and clinical KC (CKC). Each cornea is represented by 79 features extracted from Pentacam imaging. Following extensive pre-processing, key corneal features that are strongly correlated with the target diagnosis are identified. These features are the keratometry of the steepest anterior point, surface variance index, vertical asymmetry index, height decentration index, and height asymmetry index. A novel stacking ensemble model is developed using the selected features to improve corneal classification into NKC, SCKC, and CKC by integrating top tree-based classifiers (random forest, gradient boosting, decision trees) with a support vector machine meta-classifier.
Results:
The pre-processing and feature selection techniques reduced the model's parameters to just 6.33% of the original dataset, improving classification performance, and cutting over 85% of the training time. The performance of the developed model was validated and tested on unseen data. Experimental results showed that the model outperforms existing studies, achieving 99.72% accuracy, precision, sensitivity, F1, and F2 scores, with a Matthews correlation coefficient of 0.995. It accurately classified all NKC and CKC cases, with just one misclassification involving an SCKC case. The model also demonstrated consistent performance on 100 additional unseen test cases, underscoring its generalizability and robustness in KC screening.
Conclusions:
By combining the strengths of diverse base models and key Pentacam indices, the stacking ensemble approach ensures reliable, accurate KC screening, providing clinicians with an automated tool for early detection and better patient management.
