基于GoogLeNet网络模型实现医学影像辅助诊断的方法、装置、设备及存储介质与流程

文档序号:20192914发布日期:2020-03-27 19:50阅读:134来源:国知局
基于GoogLeNet网络模型实现医学影像辅助诊断的方法、装置、设备及存储介质与流程

本发明属于医疗设备技术领域,具体地涉及一种基于googlenet网络模型实现医学影像辅助诊断的方法、装置、设备及存储介质。



背景技术:

目前医疗器械市场上的医疗摄影设备都是可以看到医疗影像的,例如血管摄影图像(angiography)、心血管造影图像(cardiacangiography)、电脑断层扫描图像(ct,computerizedtomography)、乳房摄影图像(mammography)、正子发射断层扫描图像(pet,positronemissiontomography)、核磁共振成像图像(nmri,nuclearmagneticresonanceimaging)和医学超音波检查图像(medicalultrasonography)等。但是当前影像诊断行业却存在如下几个问题:(1)诊断大夫的培养周期长成本高,目前高端人才的缺口很大;(2)大夫个人主观性差别很大,很难做到诊断标准完全统一;(3)大夫仅凭肉眼观察,很多底层信息不可见,信息利用率低;(4)相似疾病太多,阅片过程中存在大量重复性劳动;(5)大夫阅片耗时很长,至少几十分钟到数小时不等。

针对上述几个问题,当前的解决方案有两种:一是利用传统的图像处理方法,针对某种特定疾病和特定类型的影像片子,先手动提取特征,然后根据阈值做判断,该方法需要有大量的医学和图像处理经验,而且不具有通用性,在换了部位或影像类型之后,很难再有较高的准确率,甚至根本不能工作;二是利用机器学习模型或统计学习模型,在一定程度上,实现了自动化,也具备了一定的通用性,比第一种方法相比,在准确率上也有极大的提高,但是,模型依然存在表达能力不足的问题,不能涵盖各种复杂的情况,而且,很难进行迁移学习。



技术实现要素:

为了解决现有医疗设备在利用机器学习模型进行影像诊断时不能涵盖各种复杂情况的问题,本发明目的在于提供一种基于googlenet网络模型实现医学影像辅助诊断的方法、装置、设备及存储介质。

本发明所采用的技术方案为:

一种基于googlenet网络模型实现医学影像辅助诊断的方法,包括如下步骤:

s101.获取若干张样本医学影像和对应各张样本医学影像的诊断标签,其中,所述诊断标签为正例或负例;

s102.对各张样本医学影像进行图像预处理,得到相应的且呈正方形、尺寸大小一致和归一化的标准样本医学影像,同时对相应的诊断标签进行如下形式的数字编码;若为正例,标记为1,而若为负例,标记为0;

s103.从训练样本数据集中抽取2n张标准样本医学影像,然后将这些标准样本医学影像导入至googlenet网络模型中进行正向传播,得到对应各张标准样本医学影像的第一预测概率,其中,所述训练样本数据集包括有不少于2n张的所述标准样本医学影像,n为介于4~8之间的自然数,所述第一预测概率包括识别影像为正例的概率和识别影像为负例的概率;

s104.按照如下公式计算本次训练的平均损失值loss:

式中,i为介于1~2n之间的自然数,yi为第i张标准样本医学影像的诊断标签编码值,为第i张标准样本医学影像的且识别影像为正例的概率,κ为常数;

s105.通过梯度下降法将所述平均损失值loss逐层反向传播,进行模型参数的更新;

s106.返回执行步骤s103~s105,直到所述平均损失值loss降低到预设阈值,完成模型训练;

s107.获取待诊医学影像;

s108.对所述待诊医学影像进行图像预处理,得到呈正方形、尺寸大小与所述标准样本医学影像一致和归一化方式与所述标准样本医学影像一致的标准待诊医学影像;

s109.将所述标准待诊医学影像输入完成训练的googlenet网络模型中,得到对应所述标准待诊医学影像的第二预测概率,其中,所述第二预测概率包括识别影像为正例的概率和识别影像为负例的概率;

s110.根据所述第二预测概率得出诊断结果:若所述第二预测概率中识别影像为正例的概率不小于正例与负例的分割阈值,则判定诊断结果为正例和/或阳性,否则判定诊断结果为负例和/或阴性。

优化的,在所述步骤s102中,通过如下方式的图像预处理得到呈正方形的样本医学影像:

在样本医学影像的原始形状为非正方形的长方形时,对称地在宽度方向两侧填充像素值为0或像素均值的像素点,使最终的影像宽度等于影像长度,其中,像素均值为原样本医学影像中所有像素点的像素平均值。

进一步优化的,在所述步骤s102中且得到呈正方形的样本医学影像之后,通过如下方式的图像预处理得到尺寸大小一致的样本医学影像:

通过插值或压采样的方式使样本医学影像变成尺寸大小为2k*2k的影像,其中,k为介于7~12之间的自然数。

详细优化的,在所述步骤s102中且得到尺寸大小一致的样本医学影像之后,通过如下方式的图像预处理得到归一化的样本医学影像:

针对样本医学影像上的各个像素点,按照如下公式得到归一化后的像素值pnew:

式中,pold为归一化前的像素值,μ为归一化前所有像素点的像素平均值,σ为归一化前所有像素点的像素值标准差。

优化的,所述googlenet网络模型包括有conv2d_bn子模块,其中,所述conv2d_bn子模块沿正向传播方向依次包括有一个激活函数为relu激活函数且padding类型为same的卷积层和一个批归一化层;

所述googlenet网络模型沿正向传播方向依次包括有图像输入层、第一conv2d_bn模块、第一最大池化层、第二conv2d_bn模块、第二最大池化层、第一inception模块、第三最大池化层、第二inception模块、第四最大池化层、第三inception模块、平均池化层、扁平化层、第一丢弃层、第一全连接层、第二丢弃层、第二全连接层和第三全连接层;

所述第一conv2d_bn模块的输出通道数为64个,并包括有一个卷积核心为(7,7)且步长为(2,2)的所述conv2d_bn子模块;

所述第一最大池化层的池化核心为(2,2),步长为(2,2),padding类型为same;

所述第二conv2d_bn模块的输出通道数为128个,并包括有一个卷积核心为(3,3)且步长为(2,2)的所述conv2d_bn子模块;

所述第二最大池化层的池化核心为(2,2),步长为(2,2),padding类型为same;

所述第一inception模块包括有五个输出通道数分别为64个的且在通道的维度上按顺序拼接起来的并行分支:第一个并行分支包括有一个卷积核心为(1,1)的所述conv2d_bn子模块,第二个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(3,3)的所述conv2d_bn子模块,第三个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(5,5)的所述conv2d_bn子模块,第四个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(7,7)的所述conv2d_bn子模块,第五个并行分支沿正向传播方向依次包括有一个池化核心为(3,3)的最大池化层和一个卷积核心为(1,1)的所述conv2d_bn子模块;

所述第三最大池化层的池化核心为(2,2),步长为(2,2),padding类型为same;

所述第二inception模块也包括有五个输出通道数分别为96个的且在通道的维度上按顺序拼接起来的并行分支:第一个并行分支包括有一个卷积核心为(1,1)的所述conv2d_bn子模块,第二个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(3,3)的所述conv2d_bn子模块,第三个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(5,5)的所述conv2d_bn子模块,第四个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(7,7)的所述conv2d_bn子模块,第五个并行分支沿正向传播方向依次包括有一个池化核心为(3,3)的最大池化层和一个卷积核心为(1,1)的所述conv2d_bn子模块;

所述第四最大池化层的池化核心为(2,2),步长为(2,2),padding类型为same;

所述第三inception模块也包括有五个输出通道数分别为96个的且在通道的维度上按顺序拼接起来的并行分支:第一个并行分支包括有一个卷积核心为(1,1)的所述conv2d_bn子模块,第二个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(3,3)的所述conv2d_bn子模块,第三个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(5,5)的所述conv2d_bn子模块,第四个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(7,7)的所述conv2d_bn子模块,第五个并行分支沿正向传播方向依次包括有一个池化核心为(3,3)的最大池化层和一个卷积核心为(1,1)的所述conv2d_bn子模块;

所述平均池化层的池化核心为(3,3),步长为(2,2),padding类型为same;

所述扁平化层用于将特征图展成一行,便于后面接第一丢弃层;

所述第一丢弃层用于随机地使一半的神经元失活,避免过拟合,提高泛化能力;

所述第一全连接层的神经元数量为512,激活函数为relu激活函数;

所述第二丢弃层用于随机地使一半的神经元失活,避免过拟合,提高泛化能力;

所述第二全连接层的神经元数量为512,激活函数为relu激活函数;

所述第三全连接层的神经元数量为2,激活函数为softmax激活函数,其中,所述softmax激活函数用于识别影像为正例的概率和识别影像为负例的概率。

本发明所采用的另一种技术方案为:

一种基于googlenet网络模型实现医学影像辅助诊断的装置,包括样本获取模块、样本预处理模块、模型迭代训练模块、影像获取模块、影像预处理模块、影像识别模块和影像诊断模块,其中,所述模型迭代训练模块包括模型训练子模块、损失计算子模块、参数更新子模块和迭代控制子模块;

所述样本获取模块,用于获取若干张样本医学影像和对应各张样本医学影像的诊断标签,其中,所述诊断标签为正例或负例;

所述样本预处理模块,通信连接所述样本获取模块,用于对各张样本医学影像进行图像预处理,得到相应的且呈正方形、尺寸大小一致和归一化的标准样本医学影像,同时对相应的诊断标签进行如下形式的数字编码;若为正例,标记为1,而若为负例,标记为0;

所述模型训练子模块,通信连接所述样本预处理模块,用于从训练样本数据集中抽取2n张标准样本医学影像,然后将这些标准样本医学影像导入至googlenet网络模型中进行正向传播,得到对应各张标准样本医学影像的第一预测概率,其中,所述训练样本数据集包括有不少于2n张的所述标准样本医学影像,n为介于4~8之间的自然数,所述第一预测概率包括识别影像为正例的概率和识别影像为负例的概率;

所述损失计算子模块,分别通信连接所述模型训练子模块和所述样本预处理模块,用于按照如下公式计算本次训练的平均损失值loss:

式中,i为介于1~2n之间的自然数,yi为第i张标准样本医学影像的诊断标签编码值,为第i张标准样本医学影像的且识别影像为正例的概率,κ为常数;

所述参数更新子模块,通信连接所述损失计算子模块,用于通过梯度下降法将所述平均损失值loss逐层反向传播,进行模型参数的更新;

所述迭代控制子模块,通信连接所述损失计算子模块,用于循环地依次启动所述模型训练子模块、所述损失计算子模块和所述参数更新子模块,直到所述平均损失值loss降低到预设阈值,完成模型训练;

所述影像获取模块,用于获取待诊医学影像;

所述影像预处理模块,通信连接所述影像获取模块,用于对所述待诊医学影像进行图像预处理,得到呈正方形、尺寸大小与所述标准样本医学影像一致和归一化方式与所述标准样本医学影像一致的标准待诊医学影像;

所述影像识别模块,分别通信连接所述模型迭代训练模块和所述影像预处理模块,用于将所述标准待诊医学影像输入完成训练的googlenet网络模型中,得到对应所述标准待诊医学影像的第二预测概率,其中,所述第二预测概率包括识别影像为正例的概率和识别影像为负例的概率;

所述影像诊断模块,通信连接所述影像识别模块,用于根据所述第二预测概率得出诊断结果:若所述第二预测概率中识别影像为正例的概率不小于正例与负例的分割阈值,则判定诊断结果为正例和/或阳性,否则判定诊断结果为负例和/或阴性。

本发明所采用的另一种技术方案为:

一种基于googlenet网络模型实现医学影像辅助诊断的设备,包括通信相连的存储器和处理器,其中,所述存储器用于存储计算机程序,所述处理器用于执行所述计算机程序实现如前所述基于googlenet网络模型实现医学影像辅助诊断的方法步骤。

本发明所采用的另一种技术方案为:

一种存储介质,所述存储介质上存储有计算机程序,所述计算机程序被处理器执行时实现如前所述基于googlenet网络模型实现医学影像辅助诊断的方法步骤。

本发明的有益效果为:

(1)本发明创造提供了一种利用人工智能中最新深度学习技术实现医学影像辅助诊断的新方法,即在具有很好泛化能力的googlenet网络模型中,先用卷积神经网络提取各种复杂的特征,然后利用全连接网络对特征进行综合的判断,进而使模型可以涵盖各种复杂的情况,可利于在医学影像辅助诊断的全过程中,实现端到端的自动化诊断,效率和准确率都极高;

(2)还可以对新增部位或影像进行迁移学习,使模型升级变得简单易操作,而且能保证对原有知识的最大兼容性,便于实际应用和推广。

附图说明

为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。

图1是本发明提供的基于googlenet网络模型实现医学影像辅助诊断的方法流程示意图。

图2是本发明提供的在进行图像填充和统一大小的图像预处理的前后影像对比示例图。

图3是本发明提供的googlenet网络模型的多层结构上部区域示意图。

图4是本发明提供的googlenet网络模型的多层结构中部区域示意图。

图5是本发明提供的googlenet网络模型的多层结构下部区域示意图。

图6是本发明提供的基于googlenet网络模型实现医学影像辅助诊断的装置结构示意图。

图7是本发明提供的基于googlenet网络模型实现医学影像辅助诊断的设备结构示意图。

具体实施方式

下面结合附图及具体实施例来对本发明作进一步阐述。在此需要说明的是,对于这些实施例方式的说明虽然是用于帮助理解本发明,但并不构成对本发明的限定。本文公开的特定结构和功能细节仅用于描述本发明的示例实施例。然而,可用很多备选的形式来体现本发明,并且不应当理解为本发明限制在本文阐述的实施例中。

应当理解,尽管本文可能使用术语第一、第二等等来描述各种单元,但是这些单元不应当受到这些术语的限制。这些术语仅用于区分一个单元和另一个单元。例如可以将第一单元称作第二单元,并且类似地可以将第二单元称作第一单元,同时不脱离本发明的示例实施例的范围。

应当理解,对于本文中可能出现的术语“和/或”,其仅仅是一种描述关联对象的关联关系,表示可以存在三种关系,例如,a和/或b,可以表示:单独存在a,单独存在b,同时存在a和b三种情况;对于本文中可能出现的术语“/和”,其是描述另一种关联对象关系,表示可以存在两种关系,例如,a/和b,可以表示:单独存在a,单独存在a和b两种情况;另外,对于本文中可能出现的字符“/”,一般表示前后关联对象是一种“或”关系。

应当理解,在本文中若将单元称作与另一个单元“连接”、“相连”或“耦合”时,它可以与另一个单元直相连接或耦合,或中间单元可以存在。相対地,在本文中若将单元称作与另一个单元“直接相连”或“直接耦合”时,表示不存在中间单元。另外,应当以类似方式来解释用于描述单元之间的关系的其他单词(例如,“在……之间”对“直接在……之间”,“相邻”对“直接相邻”等等)。

应当理解,本文使用的术语仅用于描述特定实施例,并不意在限制本发明的示例实施例。若本文所使用的,单数形式“一”、“一个”以及“该”意在包括复数形式,除非上下文明确指示相反意思。还应当理解,若术语“包括”、“包括了”、“包含”和/或“包含了”在本文中被使用时,指定所声明的特征、整数、步骤、操作、单元和/或组件的存在性,并且不排除一个或多个其他特征、数量、步骤、操作、单元、组件和/或他们的组合存在性或增加。

应当理解,还应当注意到在一些备选实施例中,所出现的功能/动作可能与附图出现的顺序不同。例如,取决于所涉及的功能/动作,实际上可以实质上并发地执行,或者有时可以以相反的顺序来执行连续示出的两个图。

应当理解,在下面的描述中提供了特定的细节,以便于对示例实施例的完全理解。然而,本领域普通技术人员应当理解可以在没有这些特定细节的情况下实现示例实施例。例如可以在框图中示出系统,以避免用不必要的细节来使得示例不清楚。在其他实例中,可以不以不必要的细节来示出众所周知的过程、结构和技术,以避免使得示例实施例不清楚。

实施例一

如图1~5所示,本实施例提供的所述基于googlenet网络模型实现医学影像辅助诊断的方法,可以但不限于包括如下步骤s101~s110。

s101.获取若干张样本医学影像和对应各张样本医学影像的诊断标签,其中,所述诊断标签为正例或负例。

在所述步骤s101中,所述样本医学影像以及对应的诊断标签(其中的正例对应阳性,负例对应阴性,此为医学常识),可以但不限于从历史档案的数据库中获取,也可以从网络开源的数据库中获取,例如由斯坦福吴恩达团队开源的mura数据库(https://stanfordmlgroup.github.io/competitions/mura/),其包括有至少14982个病例的上肢肌肉骨骼x光片,每个病例包括一个或多个图像,均由放射科医师手动标志(即形成对应的诊断标签),并为了鼓舞医学影像诊断模型的提高,该mura数据库可以无偿使用。

s102.对各张样本医学影像进行图像预处理,得到相应的且呈正方形、尺寸大小一致和归一化的标准样本医学影像,同时对相应的诊断标签进行如下形式的数字编码;若为正例,标记为1,而若为负例,标记为0。

在所述步骤s102中,可以但不限于通过如下方式的图像预处理得到呈正方形的样本医学影像:在样本医学影像的原始形状为非正方形的长方形时,对称地在宽度方向两侧填充像素值为0或像素均值的像素点,使最终的影像宽度等于影像长度,其中,像素均值为原样本医学影像中所有像素点的像素平均值。如图2所示,所述样本医学影像均为单通道图像,其长宽可以是任意的,如果长宽不等,就需要通过前述方式使长宽相等,以便后续进行模型训练。此外,由于是在宽度方向两侧进行对称填充,可保证原始影像位于新影像的正中央区域。

在所述步骤s102中且得到呈正方形的样本医学影像之后,可以但不限于通过如下方式的图像预处理得到尺寸大小一致的样本医学影像:通过插值或压采样的方式使样本医学影像变成尺寸大小为2k*2k的影像,其中,k为介于7~12之间的自然数。例如得到尺寸大小为256*256像素(此时k为8)的影像、为512*512像素(此时k为9,常用尺寸大小)的影像或为1024*1024像素(此时k为10)的影像。

在所述步骤s102中且得到尺寸大小一致的样本医学影像之后,可以但不限于通过如下方式的图像预处理得到归一化的样本医学影像:针对样本医学影像上的各个像素点,按照如下公式得到归一化后的像素值pnew:

式中,pold为归一化前的像素值,μ为归一化前所有像素点的像素平均值,σ为归一化前所有像素点的像素值标准差。通过前述影像的归一化处理,可使后期训练时加快收敛,缩短模型训练时间和识别时间,利于快速得到诊断结果。

s103.从训练样本数据集中抽取2n张标准样本医学影像,然后将这些标准样本医学影像导入至googlenet网络模型中进行正向传播,得到对应各张标准样本医学影像的第一预测概率,其中,所述训练样本数据集包括有不少于2n张的所述标准样本医学影像,n为介于4~8之间的自然数,所述第一预测概率包括识别影像为正例的概率和识别影像为负例的概率。

在所述步骤s103中,2n一般为16、32、64或128等,视硬件的情况而定。若所述训练样本数据集中的标准样本医学影像少于2n张,则为了进行步骤s103和更真实地模拟现实场景,有必要在所述步骤s103之前对所述训练样本数据集做数据增强处理,以便充实所述训练样本数据集的影像数量。优化的,针对所述训练样本数据集,可以但不限于做如下方式(a)~(d)中任意一种或任意组合的数据增强处理:(a)对所述标准样本医学影像进行随机地上下和/或左右翻转处理,然后将得到的影像作为新样本添加到所述训练样本数据集中;(b)对所述标准样本医学影像进行随机地角度旋转处理,然后将得到的影像作为新样本添加到所述训练样本数据集中;(c)对所述标准样本医学影像进行随机地高斯噪声添加处理,然后将得到的影像作为新样本添加到所述训练样本数据集中;(d)对所述标准样本医学影像进行随机地图像裁剪处理,然后将得到的影像作为新样本添加到所述训练样本数据集中,其中,图像裁剪部分所占整个影像的比例不超过5%。前述翻转处理、旋转处理、高斯噪声添加处理和图像裁剪处理等均为常规图像处理方式,其中举例的,高斯噪声的均值为0,标准差为0.01。

在所述步骤s103中,所述googlenet网络模型是由谷歌(google)研究出来的深度网络结构模型,其继承了lenet网络模型以及alexnet网络模型的一些框架结构,并做了更加大胆的网络结构尝试,虽然深度只有22层,但大小却比alexnet网络模型和vggnet网络模型小很多:googlenet网络模型的参数为500万个,alexnet网络模型的参数个数是googlenet网络模型的12倍,而vggnet网络模型的参数又是alexnet网络模型的3倍,因此在内存或计算资源有限时,googlenet是比较好的选择;同时从模型结果来看,googlenet的性能却更加优越。由此优化的,可基于googlenet网络模型的层结构原理,搭建一个适用于样本医学影像训练和识别的改进型googlenet网络模型,即如图3~5所示,所述改进型googlenet网络模型包括有conv2d_bn子模块,其中,所述conv2d_bn子模块沿正向传播方向依次包括有一个激活函数为relu激活函数且padding类型为same的卷积层(conv2d)和一个批归一化层(batchnormalization);所述googlenet网络模型沿正向传播方向依次包括有图像输入层(input)、第一conv2d_bn模块、第一最大池化层(maxpooling2d)、第二conv2d_bn模块、第二最大池化层(maxpooling2d)、第一inception模块、第三最大池化层(maxpooling2d)、第二inception模块、第四最大池化层(maxpooling2d)、第三inception模块、平均池化层(averagepooling2d)、扁平化层(flatten)、第一丢弃层(dropout)、第一全连接层(dense)、第二丢弃层(dropout)、第二全连接层(dense)和第三全连接层(dense);所述第一conv2d_bn模块的输出通道数为64个,并包括有一个卷积核心为(7,7)且步长为(2,2)的所述conv2d_bn子模块;所述第一最大池化层的池化核心为(2,2),步长为(2,2),padding类型为same;所述第二conv2d_bn模块的输出通道数为128个,并包括有一个卷积核心为(3,3)且步长为(2,2)的所述conv2d_bn子模块;所述第二最大池化层的池化核心为(2,2),步长为(2,2),padding类型为same;所述第一inception模块包括有五个输出通道数分别为64个的且在通道的维度上按顺序拼接起来的并行分支:第一个并行分支包括有一个卷积核心为(1,1)的所述conv2d_bn子模块,第二个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(3,3)的所述conv2d_bn子模块,第三个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(5,5)的所述conv2d_bn子模块,第四个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(7,7)的所述conv2d_bn子模块,第五个并行分支沿正向传播方向依次包括有一个池化核心为(3,3)的最大池化层和一个卷积核心为(1,1)的所述conv2d_bn子模块;所述第三最大池化层的池化核心为(2,2),步长为(2,2),padding类型为same;所述第二inception模块也包括有五个输出通道数分别为96个的且在通道的维度上按顺序拼接起来的并行分支:第一个并行分支包括有一个卷积核心为(1,1)的所述conv2d_bn子模块,第二个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(3,3)的所述conv2d_bn子模块,第三个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(5,5)的所述conv2d_bn子模块,第四个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(7,7)的所述conv2d_bn子模块,第五个并行分支沿正向传播方向依次包括有一个池化核心为(3,3)的最大池化层和一个卷积核心为(1,1)的所述conv2d_bn子模块;所述第四最大池化层的池化核心为(2,2),步长为(2,2),padding类型为same;所述第三inception模块也包括有五个输出通道数分别为96个的且在通道的维度上按顺序拼接起来的并行分支:第一个并行分支包括有一个卷积核心为(1,1)的所述conv2d_bn子模块,第二个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(3,3)的所述conv2d_bn子模块,第三个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(5,5)的所述conv2d_bn子模块,第四个并行分支沿正向传播方向依次包括有一个卷积核心为(1,1)的所述conv2d_bn子模块和一个卷积核心为(7,7)的所述conv2d_bn子模块,第五个并行分支沿正向传播方向依次包括有一个池化核心为(3,3)的最大池化层和一个卷积核心为(1,1)的所述conv2d_bn子模块;所述平均池化层的池化核心为(3,3),步长为(2,2),padding类型为same;所述扁平化层用于将特征图展成一行,便于后面接第一丢弃层;所述第一丢弃层用于随机地使一半的神经元失活,避免过拟合,提高泛化能力;所述第一全连接层的神经元数量为512,激活函数为relu激活函数;所述第二丢弃层用于随机地使一半的神经元失活,避免过拟合,提高泛化能力;所述第二全连接层的神经元数量为512,激活函数为relu激活函数;所述第三全连接层的神经元数量为2,激活函数为softmax激活函数,其中,所述softmax激活函数用于识别影像为正例的概率和识别影像为负例的概率。前述所有的技术术语均为现有深度学习技术中的常见用语,于此不再赘述。此外,所述第一全连接层的神经元数量可举例为1024。

s104.按照如下公式计算本次训练的平均损失值loss:

式中,i为介于1~2n之间的自然数,yi为第i张标准样本医学影像的诊断标签编码值,为第i张标准样本医学影像的且识别影像为正例的概率,κ为常数。

在所述步骤s104中,所述诊断标签编码值即表示识别影像为正例的真实概率,若标记为1,表示识别影像为正例的真实概率为100%,若标记为0,表示识别影像为正例的真实概率为0%,因此前述公式的核心是将预测概率与真实概率做交叉熵。此外,κ可举例为2、e或10等常数。

s105.通过梯度下降法将所述平均损失值loss逐层反向传播,进行模型参数的更新。

在所述步骤s105中,所述通过梯度下降法来进行模型参数更新的方式为现有常规方式。

s106.返回执行步骤s103~s105,直到所述平均损失值loss降低到预设阈值,完成模型训练。

在所述步骤s106中且返回执行步骤s103时,可以从所述训练样本数据集中抽取2n张与前次相同、部分相同或完全不同的标准样本医学影像。此外,也可以在训练精度达到要求时,停止返回执行,完成模型训练。

s107.获取待诊医学影像。

在所述步骤s107中,所述待诊医学影像具体可从现有医疗摄影设备的输出接口(例如usb接口或hdmi接口)中导出得到。

s108.对所述待诊医学影像进行图像预处理,得到呈正方形、尺寸大小与所述标准样本医学影像一致和归一化方式与所述标准样本医学影像一致的标准待诊医学影像。

在所述步骤s108中,所述图像预处理方式可与所述步骤s102中的图像预处理方式完全一致。

s109.将所述标准待诊医学影像输入完成训练的googlenet网络模型中,得到对应所述标准待诊医学影像的第二预测概率,其中,所述第二预测概率包括识别影像为正例的概率和识别影像为负例的概率。

s110.根据所述第二预测概率得出诊断结果:若所述第二预测概率中识别影像为正例的概率不小于正例与负例的分割阈值,则判定诊断结果为正例和/或阳性,否则判定诊断结果为负例和/或阴性。

在所述步骤s110中,所述正例与负例的分割阈值一般为0.5。

遵照前述步骤s101~s110,本实施例还举例地利用mura数据库中的肱骨部(humerus)医学影像进行了测试:先对1017张医学影像(单通道影像)进行了如步骤s102所述的样本处理,得到尺寸为512*512像素的标准样本医学图像,并构成所述训练样本数据集;然后对288张医学影像(单通道影像)进行了如步骤s108所述的图像处理,得到尺寸为512*512像素的标准待诊医学图像,并构成一个验证集;再然后配置googlenet网络模型的参数如下:优化器——adam,随机梯度下降法,损失函数——binay_crossentropy,学习率——0.01;再然后通过迭代20余次即完成模型训练;最后应用完成训练的googlenet网络模型对所述验证集进行逐一验证,发现只有几次出现诊断错误,大大超过了人类专家的诊断准确率。另外,即使每次数据增强之后,所述训练样本数据集会有差别,并使最终结果也会略有不同,但也都超过了人类专家的诊断准确率。

综上,采用本实施例所提供的基于googlenet网络模型实现医学影像辅助诊断的方法,具有如下技术效果:

(1)本实施例提供了一种利用人工智能中最新深度学习技术实现医学影像辅助诊断的新方法,即在具有很好泛化能力的googlenet网络模型中,先用卷积神经网络提取各种复杂的特征,然后利用全连接网络对特征进行综合的判断,进而使模型可以涵盖各种复杂的情况,可利于在医学影像辅助诊断的全过程中,实现端到端的自动化诊断,效率和准确率都极高;

(2)还可以对新增部位或影像进行迁移学习,使模型升级变得简单易操作,而且能保证对原有知识的最大兼容性,便于实际应用和推广。

实施例二

如图6所示,本实施例提供了一种实现实施例一所述基于googlenet网络模型实现医学影像辅助诊断的方法的硬件装置,包括样本获取模块、样本预处理模块、模型迭代训练模块、影像获取模块、影像预处理模块、影像识别模块和影像诊断模块,其中,所述模型迭代训练模块包括模型训练子模块、损失计算子模块、参数更新子模块和迭代控制子模块;

所述样本获取模块,用于获取若干张样本医学影像和对应各张样本医学影像的诊断标签,其中,所述诊断标签为正例或负例;

所述样本预处理模块,通信连接所述样本获取模块,用于对各张样本医学影像进行图像预处理,得到相应的且呈正方形、尺寸大小一致和归一化的标准样本医学影像,同时对相应的诊断标签进行如下形式的数字编码;若为正例,标记为1,而若为负例,标记为0;

所述模型训练子模块,通信连接所述样本预处理模块,用于从训练样本数据集中抽取2n张标准样本医学影像,然后将这些标准样本医学影像导入至googlenet网络模型中进行正向传播,得到对应各张标准样本医学影像的第一预测概率,其中,所述训练样本数据集包括有不少于2n张的所述标准样本医学影像,n为介于4~8之间的自然数,所述第一预测概率包括识别影像为正例的概率和识别影像为负例的概率;

所述损失计算子模块,分别通信连接所述模型训练子模块和所述样本预处理模块,用于按照如下公式计算本次训练的平均损失值loss:

式中,i为介于1~2n之间的自然数,yi为第i张标准样本医学影像的诊断标签编码值,为第i张标准样本医学影像的且识别影像为正例的概率,κ为常数;

所述参数更新子模块,通信连接所述损失计算子模块,用于通过梯度下降法将所述平均损失值loss逐层反向传播,进行模型参数的更新;

所述迭代控制子模块,通信连接所述损失计算子模块,用于循环地依次启动所述模型训练子模块、所述损失计算子模块和所述参数更新子模块,直到所述平均损失值loss降低到预设阈值,完成模型训练;

所述影像获取模块,用于获取待诊医学影像;

所述影像预处理模块,通信连接所述影像获取模块,用于对所述待诊医学影像进行图像预处理,得到呈正方形、尺寸大小与所述标准样本医学影像一致和归一化方式与所述标准样本医学影像一致的标准待诊医学影像;

所述影像识别模块,分别通信连接所述模型迭代训练模块和所述影像预处理模块,用于将所述标准待诊医学影像输入完成训练的googlenet网络模型中,得到对应所述标准待诊医学影像的第二预测概率,其中,所述第二预测概率包括识别影像为正例的概率和识别影像为负例的概率;

所述影像诊断模块,通信连接所述影像识别模块,用于根据所述第二预测概率得出诊断结果:若所述第二预测概率中识别影像为正例的概率不小于正例与负例的分割阈值,则判定诊断结果为正例和/或阳性,否则判定诊断结果为负例和/或阴性。

本实施例提供的前述装置的工作过程、工作细节和技术效果,可以参见实施例一,于此不再赘述。

实施例三

如图7所示,本实施例提供了一种实现实施例一所述基于googlenet网络模型实现医学影像辅助诊断的方法的硬件设备,包括通信相连的存储器和处理器,其中,所述存储器用于存储计算机程序,所述处理器用于执行所述计算机程序实现如实施例一所述基于googlenet网络模型实现医学影像辅助诊断的方法步骤。

本实施例提供的前述设备的工作过程、工作细节和技术效果,可以参见实施例一,于此不再赘述。

实施例四

本实施例提供了一种存储包含实施例一所述基于googlenet网络模型实现医学影像辅助诊断的方法的计算机程序的存储介质,即在所述存储介质上存储有计算机程序,所述计算机程序被处理器执行时实现如实施例一所述基于googlenet网络模型实现医学影像辅助诊断的方法步骤。其中,计算机可以是通用计算机、专用计算机、计算机网络、或者其他可编程装置,也可以是移动智能设备(如智能手机、pad或ipad等)。

本实施例提供的前述存储介质的工作过程、工作细节和技术效果,可以参见实施例一,于此不再赘述。

以上所描述的多个实施例仅仅是示意性的,若涉及到作为分离部件说明的单元,其可以是或者也可以不是物理上分开的;若涉及到作为单元显示的部件,其可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。本领域普通技术人员在不付出创造性的劳动的情况下,即可以理解并实施。

以上实施例仅用以说明本发明的技术方案,而非对其限制;尽管参照前述实施例对本发明进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换。而这些修改或者替换,并不使相应技术方案的本质脱离本发明各实施例技术方案的精神和范围。

最后应说明的是,本发明不局限于上述可选的实施方式,任何人在本发明的启示下都可得出其他各种形式的产品。上述具体实施方式不应理解成对本发明的保护范围的限制,本发明的保护范围应当以权利要求书中界定的为准,并且说明书可以用于解释权利要求书。

当前第1页1 2 3 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1