[发明专利]一种阿尔兹海默病MRI图像分类方法

说明书

本发明公开了一种阿尔兹海默病MRI图像的分类方法。发明使用度量学习的思想构造损失函数并用于训练一组卷积神经网络。一方面，对于每一个3D数据在Axial、Sagittal、coronal三个方向上得到多个2D形式切片，同时输入到一组卷积神经网络中，既在一定程度上保留了3D的数据信息，又避免了使用3D卷积神经网络难以找到合适的数据集进行预训练而过拟合严重的问题；另一方面，以度量学习的思想构造损失函数训练得到的深度学习模型提取出的特征更具有可分性。最终可以得到更好的分类结果。

技术领域

本发明涉及图像分类技术领域，具体涉及一种阿尔兹海默病MRI图像分类方法。

背景技术

医学图像处理作为图像处理和机器学习的一个发展方向，是关系人类生活最为密切的领域之一。随着人口老龄化的发展，阿尔兹海默病作为最常见的老年痴呆症之一，给人们的生活尤其是患者及其家人的生活带来了极大的影响。阿尔兹海默病(Alzheimer’sDisease,AD)的分类作为医学图像分类领域的一个重要分支，对AD的计算机辅助诊断具有重要意义，特别是对于病情的前期诊断及病情恶化的及时控制尤为重要。

本发明采用的影像数据为MRI(Magnetic Resonance Imaging)图像。MRI图像能够提供丰富的脑组织形态特征，特别是与阿尔兹海默病相关的海马体、颞叶、额叶等部位的特征，同时阿尔兹海默病患者脑部会出现大量的萎缩，在MRI图像中可以直观看出。

传统的阿尔茨海默病分类方法主要分为特征提取和分类两个部分：(1)首先通过各种人工或半人工的手段从采集的原始影像数据(MRI，PET等)中提取特征例如灰质体积、皮层厚度、海马体积等；(2)将(1)中提取到的特征送入传统的分类器——支持向量机(SVM)、随机森林(RF)等中进行分类，但该分类方法存在特征提取困难、分类精度低的缺点。

发明内容

针对现有技术中的上述不足，本发明提供的一种阿尔兹海默病MRI图像分类方法解决了现有分类方法特征提取困难、分类精度低的问题。

为了达到上述发明目的，本发明采用的技术方案为：一种阿尔兹海默病MRI图像分类方法，包括以下步骤：

S1、采集足够数量的AD、MCI和NC三类人群的MRI图像；

S2、对MRI图像进行预处理，得到Axial、Sagittal和coronal三个方向上的多个2D形式的切片，并将图像数据集分为训练集和测试集；

S3、构建深度学习模型，并对深度学习模型中每个子网络进行预训练；所述深度学习模型中的每个子网络都是一个卷积神经网络，子网络个数与每个样本的切片数量相等；

S4、构建基于度量学习的损失函数，将训练集中的训练数据作为深度学习模型的输入，并通过反向传播及随机梯度下降法优化网络参数，得到测试模型；

S5、将测试集中的测试数据输入测试模型中，通过前向传播得到输出数据，将输出数据作为分类特征输入KNN分类器得到分类结果。

进一步地：所述步骤S3中的预训练具体为：在每个子网络的最后一层增加softmax层，通过ImageNet数据集对每个子网络单独进行预训练，预训练完成后，去除softmax层，并用各个子网络的参数预训练结果初始化此深度学习模型中的相应参数。

进一步地：所述步骤S4中的损失函数J具体为：

J＝J₁+α·J₂+regularization

上式中，J₁为第一损失函数，J₂为第二损失函数，α为大于0的常数，regularization为网络参数的正则化项。

其中，第一损失函数为：