Enhancing generalization in Sickle Cell Disease diagnosis through ensemble methods and feature importance analysis
This work presents a novel approach for selecting the optimal ensemble-based classification method and features with a primarly focus on achieving generalization, based on the state-of-the-art, to provide diagnostic support for Sickle Cell Disease using peripheral blood smear images of red blood cells. We pre-processed and segmented the microscopic images to ensure the extraction of high-quality features. To ensure the reliability of our proposed system, we conducted an in-depth analysis of interpretability. Leveraging techniques established in the literature, we extracted features from blood cells and employed ensemble machine learning methods to classify their morphology. Furthermore, we have devised a methodology to identify the most critical features for classification, aimed at reducing complexity and training time and enhancing interpretability in opaque models. Lastly, we validated our results using a new dataset, where our model overperformed state-of-the-art models in terms of generalization. The results of classifier ensembled of Random Forest and Extra Trees classifier achieved an harmonic mean of precision and recall (F1-score) of 90.71% and a Sickle Cell Disease diagnosis support score (SDS-score) of 93.33%. These results demonstrate notable enhancement from previous ones with Gradient Boosting classifier (F1-score 87.32% and SDS-score 89.51%). To foster scientific progress, we have made available the parameters for each model, the implemented code library, and the confusion matrices with the raw data.
Authors
- Antoni Jaume-i-Capó (ORCID: https://orcid.org/0000-0003-3312-5347)
- Nataša Petrović (ORCID: https://orcid.org/0000-0002-9576-7102)
- Gabriel Moyà-Alcover (ORCID: https://orcid.org/0000-0002-3412-5499)
- José María Buades Rubio (ORCID: https://orcid.org/0000-0002-6137-9558)
Institutions
- Demos (US)
- Universitat de les Illes Balears (ES)
Publication Details
- Journal
- Engineering Applications of Artificial Intelligence
- Published
- 2024-12-27
- DOI
- https://doi.org/10.1016/j.engappai.2024.109875
- Citations
- 5
- Primary Topic
- Digital Imaging for Blood Diseases
- Type
- article
- Field-Weighted Citation Impact
- 1.25