用集成模型与特征分析提升镰状细胞病诊断泛化能力。
Enhancing Generalization in Sickle Cell Disease Diagnosis through Ensemble Methods and Feature Importance Analysis
- 结合随机森林与额外树构建集成模型,优化分类性能。
- 在新数据集上达90.71%的F1分数与93.33%的诊断支持得分。
- 可解释性分析帮助识别关键特征,适合医学影像研究者参考。
本文提出一种基于集成学习与特征重要性分析的新方法,用于通过红细胞外周血涂片图像支持镰状细胞病诊断,并重点提升模型泛化能力。通过对显微图像进行预处理与分割,提取高质量特征。采用文献中成熟的特征提取与集成机器学习方法对红细胞形态进行分类,并设计了识别关键特征的流程,以降低模型复杂度、缩短训练时间并增强黑箱模型的可解释性。在新数据集上的验证结果显示,随机森林与额外树集成分类器的谐平均精度与召回率(F1-score)达90.71%,镰状细胞病诊断支持评分(SDS-score)为93.33%,显著优于以往梯度提升模型(F1-score 87.32%,SDS-score 89.51%)。为推动科学进步,论文公开了各模型参数、代码库及原始数据的混淆矩阵。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。