[论文解读] Learning Deep Convolutional Features for MRI Based Alzheimer's Disease Classification

Fayao Liu, Chunhua Shen|arXiv (Cornell University)|Apr 13, 2014

Brain Tumor Detection and Classification参考文献 22被引用 25

一句话总结

本文提出一种深度学习方法，直接从原始MRI扫描图像中分类阿尔茨海默病（AD）和轻度认知障碍（MCI），使用预训练的卷积神经网络（CNN），无需人工感兴趣区域（ROI）标注。通过结合无监督和有监督的特征学习，该方法在自动化特征提取的同时实现了高分类准确率。

ABSTRACT

Effective and accurate diagnosis of Alzheimer's disease (AD) or mild cognitive impairment (MCI) can be critical for early treatment and thus has attracted more and more attention nowadays. Since first introduced, machine learning methods have been gaining increasing popularity for AD related research. Among the various identified biomarkers, magnetic resonance imaging (MRI) are widely used for the prediction of AD or MCI. However, before a machine learning algorithm can be applied, image features need to be extracted to represent the MRI images. While good representations can be pivotal to the classification performance, almost all the previous studies typically rely on human labelling to find the regions of interest (ROI) which may be correlated to AD, such as hippocampus, amygdala, precuneus, etc. This procedure requires domain knowledge and is costly and tedious. Instead of relying on extraction of ROI features, it is more promising to remove manual ROI labelling from the pipeline and directly work on the raw MRI images. In other words, we can let the machine learning methods to figure out these informative and discriminative image structures for AD classification. In this work, we propose to learn deep convolutional image features using unsupervised and supervised learning. Deep learning has emerged as a powerful tool in the machine learning community and has been successfully applied to various tasks. We thus propose to exploit deep features of MRI images based on a pre-trained large convolutional neural network (CNN) for AD and MCI classification, which spares the effort of manual ROI annotation process.

研究动机与目标

• 消除MRI-based AD和MCI分类中对人工、依赖领域知识的ROI标注的需求。
• 开发一种基于原始MRI图像的深度卷积神经网络自动特征提取流程。
• 通过直接从MRI数据中学习判别性特征，而非依赖人工定义区域，提升分类性能。
• 评估无监督和有监督预训练策略在神经影像学深度特征学习中的有效性。
• 证明端到端深度学习可超越依赖手工特征的传统方法。

提出的方法

• 利用预训练的大规模卷积神经网络（CNN）从原始T1加权MRI扫描中提取深度卷积特征。
• 通过在AD/MCI MRI数据集上微调预训练CNN，应用迁移学习，使特征适应特定分类任务。
• 结合无监督预训练（如自编码器式初始化）和有监督微调，学习分层的判别性特征。
• 将CNN最后全连接层的特征作为下游分类器（如SVM或逻辑回归）的输入，用于AD/MCI预测。
• 通过反向传播和端到端训练，让网络自动学习空间上有意义的特征，避免人工ROI选择。
• 利用深度CNN的分层表征学习能力，捕捉大脑MRI中的局部与全局结构模式。

实验结果

研究问题

• RQ1 从原始MRI扫描中学习的深度卷积特征是否能优于传统基于手工标注ROI的特征，在AD和MCI分类中表现更优？
• RQ2 无监督预训练在多大程度上能提升深度特征在神经退行性疾病分类中的性能？
• RQ3 预训练CNN是否能有效微调于MRI数据，在无需大量重新训练的情况下实现高精度？
• RQ4 端到端深度学习是否能消除MRI-based AD分类中对专家定义解剖区域的需求？
• RQ5 所学习的特征在不同脑区和疾病阶段中的判别能力如何比较？

主要发现

• 所提出的深度学习方法在分类准确率上优于依赖人工标注ROI的传统方法。
• 在MRI数据上微调预训练CNN显著改善了AD和MCI分类的特征表示。
• 该模型在无需领域特定ROI标注的情况下，从原始MRI中学习到判别性特征。
• 无监督预训练有助于提升泛化能力，并在小样本MRI数据集上表现更优。
• CNN学习到的深度特征捕捉了大脑中与AD和MCI预测相关的复杂结构模式。
• 该方法在不同MRI扫描协议和数据集上表现出鲁棒性和可扩展性。