一个新的可解释的基于注意力的超学习框架,用于不平衡的大脑中风预测
1Department of Information Technology, Faculty of Computers and Information Technology, University of Tabuk, Tabuk 47912, Saudi Arabia.
Sensors (Basel, Switzerland)
|April 28, 2025
概括
这项研究引入了一种新的超级学习框架,以使用混合重抽样和可解释AI (XAI) 来提高脑中风预测的准确性. 该方法有效地处理不平衡的医疗数据,增强关键条件的检测.
科学领域:
- 医疗信息学 医疗信息学
- 医疗保健中的人工智能
- 机器学习用于预测诊断.
背景情况:
- 用于脑中风预测的医疗数据集往往表现出阶级不平衡,阻碍了传统机器学习模型的性能.
- 准确的中风预测对于及时诊断和有效的患者管理至关重要.
- 医疗保健中人工智能模型的可解释性对于建立信任和临床采用至关重要.
研究的目的:
- 开发和评估一种新的超级学习框架,用于增强脑中风预测.
- 解决医疗预测任务中不平衡数据集的挑战.
- 整合可解释的人工智能 (XAI) 以实现透明的模型洞察力.
主要方法:
- 使用SMOTE和SMOTEENN来管理阶级不平衡的混合重新采样策略.
- 动态功能选择以最大限度地减少数据噪声并改善模型焦点.
- 一个结合Random Forest和LightGBM的元学习组合,通过深度学习元分类器来改进.
- 整合SHAP (沙普利添加式解释) 以实现特征解释性.
主要成果:
- 与最先进的方法相比,拟议的框架在三个数据集 (DF-1,DF-2,DF-3) 中表现出卓越的性能.
- 实现了高精度 (例如,DF-1上的0.992189) 和F1-分数 (例如,DF-1上的0.992579).
- 显著改善了少数类实例的检测,而不会影响整体性能.
结论:
- 超级学习框架为脑中风预测提供了强大而有效的解决方案,特别是在不平衡的数据中.
- 集成XAI增强了模型的透明度和可信度.
- 这种方法为应用先进的人工智能技术应对其他不平衡的医疗预测挑战提供了基础.


