融合2D与3D卷积网络,提升阿尔茨海默病早期诊断准确率
AlzhiNet: Traversing from 2DCNN to 3DCNN, Towards Early Detection and Diagnosis of Alzheimer's Disease
- 结合2D与3D CNN提取多维脑影像特征,增强判别能力
- 在Kaggle和MIRIAD数据集上准确率达98.9%至99.99%,AUC为100%
- 对噪声、亮度等扰动具有强鲁棒性,适合真实医疗场景
阿尔茨海默病(AD)是随老龄化加剧的进行性神经退行性疾病,亟需早期精准诊断以实现有效管理。本文提出一种新型混合深度学习框架AlzhiNet,融合2D卷积神经网络(2D-CNN)与3D卷积神经网络(3D-CNN),并引入定制损失函数和体积分割数据增强技术,以提升特征提取能力并优化分类性能。大量实验表明,该框架优于独立的2D或3D模型,凸显了多模态表示融合的重要性。由增强2D切片生成的3D体积深度与质量显著影响模型表现。结果还显示,合理设定混合预测中的权重因子对取得最优效果至关重要。AlzhiNet在Kaggle与MIRIAD的磁共振成像(MRI)数据集上分别达到98.9%与99.99%的准确率,AUC均为100%。进一步在阿尔茨海默病Kaggle数据集上测试了高斯噪声、亮度变化、对比度调整、椒盐噪声、色彩抖动及遮挡等多种扰动场景,结果表明AlzhiNet比ResNet-18更具鲁棒性,适用于实际临床应用。该方法为阿尔茨海默病的早期诊断与治疗规划提供了有力支持。
原文摘要 · Abstract (English)
Alzheimer's disease (AD) is a progressive neurodegenerative disorder with increasing prevalence among the aging population, necessitating early and accurate diagnosis for effective disease management. In this study, we present a novel hybrid deep learning framework that integrates both 2D Convolutional Neural Networks (2D-CNN) and 3D Convolutional Neural Networks (3D-CNN), along with a custom loss function and volumetric data augmentation, to enhance feature extraction and improve classification performance in AD diagnosis. According to extensive experiments, AlzhiNet outperforms standalone 2D and 3D models, highlighting the importance of combining these complementary representations of data. The depth and quality of 3D volumes derived from the augmented 2D slices also significantly influence the model's performance. The results indicate that carefully selecting weighting factors in hybrid predictions is imperative for achieving optimal results. Our framework has been validated on the Magnetic Resonance Imaging (MRI) from Kaggle and MIRIAD datasets, obtaining accuracies of 98.9% and 99.99%, respectively, with an AUC of 100%. Furthermore, AlzhiNet was studied under a variety of perturbation scenarios on the Alzheimer's Kaggle dataset, including Gaussian noise, brightness, contrast, salt and pepper noise, color jitter, and occlusion. The results obtained show that AlzhiNet is more robust to perturbations than ResNet-18, making it an excellent choice for real-world applications. This approach represents a promising advancement in the early diagnosis and treatment planning for Alzheimer's disease.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。