用轻量CNN ensemble模型,从MRI图像中早期精准识别阿尔茨海默病。
Early Diagnosis of Alzheimer's Diseases and Dementia from MRI Images Using an Ensemble Deep Learning
- 设计两个轻量CNN模型,通过平均输出构建集成模型提升稳定性。
- 无SMOTE时准确率98.28%,使用SMOTE后达99.92%,显著提升小类识别效果。
- 模型参数少、性能优,适合临床早期筛查应用。
阿尔茨海默病(AD)是一种进行性神经退行性疾病,可导致严重认知障碍和痴呆。本研究提出两种低参数卷积神经网络(IR-BRAINNET和Modified-DEMNET),用于准确检测AD早期阶段。引入集成模型,通过平均两模型输出降低方差,增强诊断能力。所有模型在Kaggle数据库的MRI数据集上训练与评估,该数据集包含四期痴呆图像,存在类别分布不均问题。为缓解不平衡,采用合成少数类过采样技术(SMOTE)生成少数类样本。在无SMOTE情况下,集成模型准确率达98.28%,优于IR-BRAINNET(97.26%)和Modified-DEMNET(95.54%),Wilcoxon检验p值分别为2.9e-3和5.20e-6,表明集成显著提升预测精度。在使用SMOTE时,集成模型准确率达99.92%(较无SMOTE提升1.64%),IR-BRAINNET达99.80%(提升2.54%),Modified-DEMNET达99.72%(提升4.18%)。实验表明,模型输出平均化有效提升诊断性能,且SMOTE引入的数据多样性显著改善结果。所提轻量模型即使在类别不平衡下也优于以往研究。
原文摘要 · Abstract (English)
Alzheimer's Disease (AD) is a progressive neurological disorder that can result in significant cognitive impairment and dementia. Accurate and timely diagnosis is essential for effective treatment and management of this disease. In this study, we proposed two low-parameter Convolutional Neural Networks (CNNs), IR-BRAINNET and Modified-DEMNET, designed to detect the early stages of AD accurately. We also introduced an ensemble model that averages their outputs to reduce variance across the CNNs and enhance AD detection. Both CNNs are trained, and all models are evaluated using a Magnetic Resonance Imaging (MRI) dataset from the Kaggle database. The dataset includes images of four stages of dementia, with an uneven class distribution. To mitigate challenges stemming from the inherent imbalance in the dataset, we employed the Synthetic Minority Over-sampling Technique (SMOTE) to generate additional instances for minority classes. In the NO-SMOTE scenario, despite the imbalanced distribution, the ensemble model achieved 98.28% accuracy, outperforming IR-BRAINNET (97.26%) and Modified-DEMNET (95.54%), with Wilcoxon p-values of 2.9e-3 and 5.20e-6, respectively, indicating significant improvement in correct predictions through the use of the average function. In the SMOTE scenario, the ensemble model achieved 99.92% accuracy (1.64% improvement over NO-SMOTE), IR-BRAINNET reached 99.80% (2.54% improvement), and Modified-DEMNET attained 99.72% (4.18% improvement). Based on the experimental findings, averaging the models' outputs enhanced AD diagnosis in both scenarios, while the diversity in the dataset introduced by SMOTE-generated instances significantly improved performance. Furthermore, the compact models we proposed outperformed those from previous studies, even in the presence of an imbalanced distribution.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。