当前位置: 首页 > 专利查询>温州大学专利>正文

一种基于融合网络的阿尔兹海默病分类方法技术

技术编号:24411787 阅读:23 留言:0更新日期:2020-06-06 09:32
本发明专利技术公开了一种基于融合网络的阿尔兹海默病分类方法,包括以下步骤:1、给定被试的磁共振图像数据集,进行预处理;2、将样本划分训练集和测试集,进行图像扩增、归一化操作;3、将训练集输入训练网络;4、对样本进行特征提取及特征融合;5、对各基网络的分类决策和特征融合的分类决策进行融合;6、对输出标签进行误差计算,并通过反向传播进行参数更新;7、对分类模型评估并获得最优模型,重复第3‑7步直至迭代结束;8、训练完成的最优模型,输入通过预处理的被试磁共振影像数据,得到被试的标签。本发明专利技术使用了行卷积神经网络,实现了有效使用被试的磁共振影像数据进行阿尔兹海默病分类的方法,具有较好的鲁棒性。

A classification method of Alzheimer's disease based on fusion network

【技术实现步骤摘要】
一种基于融合网络的阿尔兹海默病分类方法
本专利技术属于脑影像与计算机科学的交叉领域,涉及图像分类
,尤其涉及一种基于融合网络的阿尔兹海默病分类方法。
技术介绍
随着人工智能和神经影像技术的快速发展,医学图像处理作为与医疗科学技术相关最为密切的领域之一正在蓬勃发展。阿尔兹海默病的分类是医学图像分类领域的一个重要研究方向,在计算机辅助诊断中具有重要意义。通过机器学习对磁共振图像进行分析,能够实现阿尔茨海默症(Alzheimer'sdisease,AD)的快速、准确分类。在某些情况下,这种方法表现出比临床医生更好的分类准确率。基于机器学习的计算机辅助诊断发展成为一个重要的研究领域,表现出巨大的优势和发展的前景。早期AD的分类方法依靠领域知识结合经典的机器学习方法,揭示阿尔兹海默病发生的病理细节。但是这些方法基于人为的先验假设,高度依赖于手工提取的特征,分类的效果受图像预处理步骤,如分割和配准等的影响。近年来,深度学习的成功进一步推动了AD分类的研究。深度学习作为一种先进的机器学习方法,避免了图像预处理步骤带来的负面影响,能够在大规模且复杂的高维数据中提取重要的信息,从而获得数据中隐含的层次化特征和数据的复杂结构。这些利用基于深度学习的基网络实现AD分类的模型,相较于传统的机器学习方法表现出良好的效果,但存在模型泛化性差的问题,以及准确率还有待进一步提升。
技术实现思路
针对以上问题,本专利技术提出一种基于融合网络的阿尔兹海默病分类方法,有效使用被试的结构核磁共振图像数据进行自动辅助疾病分类。本专利技术的技术方案:一种基于融合网络的阿尔兹海默病分类方法,包括以下步骤:步骤S1:给定被试的磁共振图像数据集,数据集包含磁共振图像和被试的标签,磁共振图像数据集记为C={(ai,bi)|i∈[1,R]},式中,a表示磁共振图像,b表示被试的对应标签,R表示样本数量,i表示样本下标。将磁共振图像全集C拷贝并划分成磁共振图像训练集E={(ai,bi)|i∈[1,M]}和磁共振图像测试集E={(ai,bi)|i∈[1,N]},其中N+M=R,且M和N分别表示划分后的样本数量;步骤S2:对磁共振图像训练集依次进行数据预处理、水平翻转扩增操作和数据归一化操作,得到处理后的磁共振图像训练集E'={(ai,bi)|i∈[1,M']},经过扩增后,其数量相应的增多至M';对磁共振图像测试集F依次进行数据预处理、数据归一化操作得到归一化的磁共振图像测试集F'={(ai,bi)|i∈[1,N]};步骤S3:将磁共振图像训练集E'作为输入,载入到计算机内存中,并将磁共振图像训练集E'={(ai,bi)|i∈[1,M']}随机打乱,为接下来的训练阶段做准备;步骤S4:分别构建基网络G1和G2。在Pytorch框架中使用SGD优化方法,对基网络G1和G2的网络参数进行初始化,得到基网络G1和G2的初始参数,并设定相关训练超参数,用于网络模型的训练优化。通过基网络G1和G2分别对训练磁共振图像进行特征提取,得到特征H1和H2,并对分别提取的特征进行特征融合得到融合特征H3;步骤S5:构建融合特征分类决策层P,将基网络G1和G2提取到的特征H1和H2以及融合特征H3分别输入到各网络的决策层,分别得到分类决策层的标签P1、P2和P3。对P1、P2和P3进行融合,得到对应输入样本最终的分类标签b';步骤S6:通过损失函数计算,计算训练数据通过网络分类得到的标签b'和训练真实标签b之间的误差,得出损失值。利用得出的损失值进行反向传播,对基网络G1和G2,以及融合网络分别进行网络参数调整;步骤S7:利用测试集F'对融合网络模型进行评估,保留网络模型G。同时在参数更新结束阶段进行判断,判断训练迭代次数是否已达到最大迭代次数,若已经达到最大迭代次数,则训练阶段结束,进入下一步骤。反之,将进入步骤S3进行循环迭代训练;步骤S8:训练完成的网络模型,输入一个通过数据预处理的被试磁共振图像数据,得到被试的输出标签。可选地,所述步骤S1中,对给定被试的磁共振图像数据集进行样本划分的具体方式为:根据被试的样本标签进行样本的随机划分,将被试对应的样本数据分为训练集和测试集。可选地,所述步骤S2中,对被试的磁共振图像数据预处理、水平翻转扩增操作和数据归一化操作具体为:使用基于体素的形态测量(VBM)对所有数据进行预处理,将图像归一化到一个标准模版空间中,然后提取数据的灰质信息,并对其进行平滑操作,最后通过水平翻转进行数据扩增。可选地,所述步骤S3中,输入模型训练的数据进行随机打乱。可选地,所述步骤S4中,在Pytorch框架中采用SGD优化方法对训练超参数动态调整,优化训练。基网络G1和G2分别对训练磁共振图像进行特征提取,得到的特征H1和H2,通过对特征H1和H2进行通道拼接,得到一个融合的特征H3。可选地,所述步骤S5中,对于基网络提取的特征H1和H2,以及融合的特征H3,在通过对应的决策层后,分别得到分类决策层的标签P1、P2和P3。对P1、P2和P3进行拼接,得到对应输入样本最终的分类标签b'。可选地,所述步骤S6中,对分类得到的标签b'和训练真实标签b进行误差计算,得出损失值,利用网络的反向传播,对模型进行参数调整。可选地,所述步骤S7中,通过输入磁共振图像测试样本计算模型分类的准确率,对训练的分类模型进行评估,保留最优模型参数。有益效果:本专利技术使用了新颖的行卷积神经网络,定义了行卷积操作,不同于传统方格卷积,该卷积操作可以有效获取图结构数据上的局部信息;定义了特征融合方法和决策融合方法,构造的基于深度融合网络的分类模型,能够有效利用多层次多分类器不同尺度的特征信息,从而获得更好的阿尔兹海默病分类效果。附图说明图1是本专利技术的总体流程图;图2是本专利技术的预处理流程图;图3是本专利技术的总体框架图;图4是本专利技术的网络结构框架图。具体实施方式为了将本专利技术实施例中的技术方案描述的完整、清楚,下面将结合本专利技术实施例中的附图展开进一步详细说明。应当理解,此处所描述的具体实施例仅用于解释本专利技术,并不限定本专利技术。请参阅图1,本专利技术提供了一种技术方案:一种基于融合网络的阿尔兹海默病分类方法,包括以下步骤:步骤S1:从ADNI(http://adni.loni.usc.edu/)公开数据集获取被试的磁共振图像数据集,该数据集为t1加权的结构MRI扫描,经过特定的图像预处理步骤,包括多平面重建(MPR)、灰度偏差(GradWarp)、B1非均匀性校正和N3强度归一化。数据集包含磁共振图像和被试的标签,其中标签是根据受教育程度和其他标准修正的记忆任务的得分来提取的。这些标签分别是AD(诊断为AD的轻度痴呆患者)、MCI(AD前驱期轻度认知障碍患者)和NC(老年对照组参与者)。磁共振图像数据集记为C={(ai,bi)|i∈[1,R]},式中,a表示磁共振图像,b表示被试的对应标签,R表示样本数量,i本文档来自技高网...

【技术保护点】
1.一种基于融合网络的阿尔兹海默病分类方法,其特征在于,所述方法包括以下步骤:/n步骤S1:给定被试的磁共振图像数据集,数据集包含磁共振图像和被试的标签,磁共振图像数据集记为C={(a

【技术特征摘要】
1.一种基于融合网络的阿尔兹海默病分类方法,其特征在于,所述方法包括以下步骤:
步骤S1:给定被试的磁共振图像数据集,数据集包含磁共振图像和被试的标签,磁共振图像数据集记为C={(ai,bi)|i∈[1,R]},式中,a表示磁共振图像,b表示被试的对应标签,R表示样本数量,i表示样本下标,将磁共振图像全集C拷贝并划分成磁共振图像训练集E={(ai,bi)|i∈[1,M]}和磁共振图像测试集E={(ai,bi)|i∈[1,N]},其中N+M=R,且M和N分别表示划分后的样本数量;
步骤S2:对磁共振图像训练集依次进行数据预处理、水平翻转扩增操作和数据归一化操作,得到处理后的磁共振图像训练集E'={(ai,bi)|i∈[1,M']},经过扩增后,其数量相应的增多至M';对磁共振图像测试集F依次进行数据预处理、数据归一化操作得到归一化的磁共振图像测试集F'={(ai,bi)|i∈[1,N]};
步骤S3:将磁共振图像训练集E'作为输入,载入到计算机内存中,并将磁共振图像训练集E'={(ai,bi)|i∈[1,M']}随机打乱,为接下来的训练阶段做准备;
步骤S4:分别构建基网络G1和G2,在Pytorch框架中使用SGD优化方法,对基网络G1和G2的网络参数进行初始化,得到基网络G1和G2的初始参数,并设定相关训练超参数,用于网络模型的训练优化,通过基网络G1和G2分别对训练磁共振图像进行特征提取,得到特征H1和H2,并对分别提取的特征进行特征融合得到融合特征H3;
步骤S5:构建融合特征分类决策层P,将基网络G1和G2提取到的特征H1和H2以及融合特征H3分别输入到各网络的决策层,分别得到分类决策层的标签P1、P2和P3,对P1、P2和P3进行融合,得到对应输入样本最终的分类标签b';
步骤S6:通过损失函数计算,计算训练数据通过网络分类得到的标签b'和训练真实标签b之间的误差,得出损失值,利用得出的损失值进行反向传播,对基网络G1和G2,以及融合网络分别进行网络参数调整;
步骤S7:利用测试集F'对融合网络模型进行评估,保留网络模型G,同时在参数更新结束阶段进行判断,判断训练迭代次数是否已达到最大迭代次数,若已经达到最大迭代次数,则训练阶段结束,进入下一步骤,反...

【专利技术属性】
技术研发人员:胡众义陈昌足吴奇
申请(专利权)人:温州大学
类型:发明
国别省市:浙江;33

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1