arXiv:2409.08170eess.IVcs.CV2024-09被引 7

轻量级CNN模型AD-Lite Net,提升阿尔茨海默病早期MRI诊断准确率

AD-Lite Net: A Lightweight and Concatenated CNN Model for Alzheimer's Detection from MRI Images

  • 采用深度可分离卷积与全局平均池化,构建轻量级网络结构
  • 引入并行拼接模块,加速收敛并有效缓解数据不平衡问题
  • 在多个数据集上超越传统CNN和ViT模型,适合医疗影像分析场景

阿尔茨海默病(AD)是一种不可治愈的进行性神经退行性疾病,导致记忆力、认知能力下降,最终影响日常活动。人工通过MRI图像诊断敏感性低且耗时。因此亟需高精度自动辅助诊断系统实现早期检测。本文提出一种全新轻量级CNN模型AD-Lite Net(从零训练),核心创新包括:(I) 结合深度可分离卷积(DWSC)与全局平均池化(GAP)层,大幅降低模型复杂度;(II) 设计并行拼接块(PCB),包含变换层(Tx-layer)后接双卷积层,将转换后的特征与原始特征拼接输入,显著增强特征表达能力。该设计使模型收敛更快,并以通用方式自动缓解MRI数据集中的类别不平衡问题。我们在三个不同MRI数据集上验证模型性能,同时合并ADNI与AD数据集,进行10折交叉验证以评估泛化能力。实验结果表明,本模型在分类准确率等指标上显著优于所有现有CNN模型及近期主流视觉变压器(ViT)模型。

原文摘要 · Abstract (English)

Alzheimer's Disease (AD) is a non-curable progressive neurodegenerative disorder that affects the human brain, leading to a decline in memory, cognitive abilities, and eventually, the ability to carry out daily tasks. Manual diagnosis of Alzheimer's disease from MRI images is fraught with less sensitivity and it is a very tedious process for neurologists. Therefore, there is a need for an automatic Computer Assisted Diagnosis (CAD) system, which can detect AD at early stages with higher accuracy. In this research, we have proposed a novel AD-Lite Net model (trained from scratch), that could alleviate the aforementioned problem. The novelties we bring here in this research are, (I) We have proposed a very lightweight CNN model by incorporating Depth Wise Separable Convolutional (DWSC) layers and Global Average Pooling (GAP) layers. (II) We have leveraged a ``parallel concatenation block'' (pcb), in the proposed AD-Lite Net model. This pcb consists of a Transformation layer (Tx-layer), followed by two convolutional layers, which are thereby concatenated with the original base model. This Tx-layer converts the features into very distinct kind of features, which are imperative for the Alzheimer's disease. As a consequence, the proposed AD-Lite Net model with ``parallel concatenation'' converges faster and automatically mitigates the class imbalance problem from the MRI datasets in a very generalized way. For the validity of our proposed model, we have implemented it on three different MRI datasets. Furthermore, we have combined the ADNI and AD datasets and subsequently performed a 10-fold cross-validation experiment to verify the model's generalization ability. Extensive experimental results showed that our proposed model has outperformed all the existing CNN models, and one recent trend Vision Transformer (ViT) model by a significant margin.

阿尔茨海默病医学影像轻量级网络CNN

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。