基于粒子群算法优化的卷积神经网络对肺结节识别的方法与流程

本发明属于人工智能与医学影像分析相结合的领域，涉及基于粒子群算法优化的卷积神经网络对肺结节识别的方法。

背景技术：

目前，计算机断层扫描(ct)是普遍使用的一种有效筛查早期肺癌的技术，ct技术的发展使得早期肺癌的发现率有了显著的提高。相比起常规的x线摄影，在ct下我们可以看到更高分辨率的肺部解剖结构图；但对于放射科医生来说，观察和解释这些图像是一个复杂又繁琐的工作。这是因为在复杂的ct图像中，肺结节的表现形式与一些肺部结构类似，肺结节密度低、体积小，有可能在离血管或肺边界很近的地方，或与血管或肺边界相连。其次，另一个因素是放射科医生需要分析大量的ct图像。每一次扫描都会产生50-250张图像，这使得放射科医生需要花大量的精力和心思去对每组ct数据分析，繁琐的工作可能会导致分析的不足、分心而出错。

技术实现要素：

针对现有技术的不足，本发明提出基于粒子群算法优化的卷积神经网络对肺结节识别的方法，通过粒子群算法对卷积神经网络进行优化，解决了卷积神经网络最优超参数的人工选取难的问题，对于肺结节有很好的识别效果。

本发明通过以下技术方案实现：基于粒子群算法优化的卷积神经网络对肺结节识别的方法，包括以下步骤：

s1、获取病人肺部的ct切片图像；

s2、对ct切片图像进行预处理，将获取的ct切片图像进行标注，形成数据集；

s3、构造卷积神经网络；

s4、使用粒子群算法优化卷积神经网络的超参数；

s5、利用标注好的数据集对优化后的卷积神经网络进行训练；

s6、使用训练完成的卷积神经网络提取肺结节特征。

优选地，所构造的卷积神经网络包括输入层、隐藏层及输出层，输入层、隐藏层、输出层为完全连接层，输出层采用softmax作为输出激活函数，隐藏层包括依次连接的卷积层c1、relu激活函数、池化层p1、卷积层c2、relu激活函数和池化层p2。

与现有技术相比，本发明具有以下有益效果：

(1)通过粒子群算法对卷积神经网络进行优化，解决了卷积神经网络最优超参数的人工选取难的问题。

(2)优化后的卷积神经网络，加快了网络的学习速度与学习效果，对于肺结节有很好的识别效果。

附图说明

图1为基于粒子群算法优化的卷积神经网络对肺结节识别的方法流程图；

图2为基于粒子群算法优化cnn的流程图。

具体实施方式

下面结合附图和实施例详细叙述本发明的技术方案，但本发明的实施方式不限于此。

本实施例中，基于粒子群算法优化的卷积神经网络对肺结节识别的方法，如图1、2所示，该方法包括以下步骤：

s1、获取病人肺部的ct切片图像；

s2、对ct切片图像进行预处理，将获取的ct切片图像进行标注，形成数据集；

s3、构造卷积神经网络cnn；

本实施例中，卷积神经网络cnn包括输入层、隐藏层及输出层，输入层、隐藏层、输出层为完全连接层，输出层采用softmax作为输出激活函数，隐藏层包括依次连接的卷积层c1、relu激活函数、池化层p1(大小：2x2)、卷积层c2、relu激活函数和池化层p2(大小：2x2)。

本实施例中，cnn采用修正线性单元relu作为激活函数，数学表达式为：

其中：y表示激活函数的输入，relu(y)表示激活函数的输出。

本实施例中，cnn采用归一化指数函数softmax作为输出激活函数，数学表达式为：

其中，j＝1，2，...，n，softmax输出激活函数能将一个含任意实数的k维向量z压缩到另一个k维实向量σ(z)中，使得每一个元素的范围都在(0，1)之间，并且所有元素的和为1。

s4、使用粒子群算法优化卷积神经网络的超参数；

具体为：利用粒子群算法优化卷积层c1和卷积层c2中的滤波器数目，并优化隐层的神经元数目、滤波器大小、池化类型、正则化概率。通过设置8个粒子坐标进行卷积神经网络的优化，设置粒子的初始速度、位置以及它们的范围、最大迭代次数。

8个粒子坐标分别对应：卷积层c1中滤波器数目、卷积层c2中滤波器数目、隐层的神经元数目、隐层的滤波器大小、训练中批次大小、池化层p1及池化层p2的池化类型、卷积层dropout概率、完全连接层dropout概率。

本实施例中，基于粒子群算法优化卷积神经网络cnn超参数的流程，如图2所示，包括：

s41、设置粒子群算法参数：包括粒子数目、粒子的初始速度与位置以及它们的范围、最大迭代次数；

s42、更新粒子群算法，计算更新后粒子的适应度：

粒子群算法的d维空间中，有n个粒子，其位置和速度分别如下：

第i个粒子的位置：

xi＝(xi1，xi2，...，xid)，i＝1，2，...，n

第i个粒子的速度：

vi＝(vi1，vi2，...，vid)，i＝1，2，...，n

将目前为止在空间中搜寻到的第i个粒子的最优位置记为：

pibest＝(pi1，pi2，...，pid)i＝1，2，...，ｎ

所有粒子搜索到的全局最优位置记为：

pgbest＝(pi1，pi2，...，pid)i＝1，2，...，ｎ

通过下面两条公式来调整自身的速度和位置来达到寻找最优解的目的：

对于上述公式，其中：c1、c2为认知和社会参数，w为惯性权重，r1、r2为二个随机函数，取值范围[0，1]，以增加搜索随机性。

本实施例中，认知和社会参数c1、c2设为2.5，惯性权重w设为0.6，惯性权重的有效值在0.2到0.9之间会达到好的效果。

s43、比较当前粒子适应度和全局历史最优适应度，若当前适应度更优，则全局历史最优的位置为当前粒子位置。

当达到最大迭代次数或者适应度误差达到设定的误差时结束迭代，输出全局最优的粒子为最佳的超参数，适用于肺结节分割。

s5、利用标注好的数据集对优化后的卷积神经网络进行训练；

s6、使用训练完成的卷积神经网络提取肺结节特征，从而实现对肺结节的识别。

上面对本发明的具体实施方式作出了详细的说明，但是本发明不限于上述实施方式，在本领域普通技术人员的知识范围内，还可以在不脱离本发明宗旨的基础上进行各种变化，其他的任何未背离本发明的精神实质与原理下所作的改变、修饰、替代、组合、简化，均应为等效的置换方式，都包含在本发明的保护范围之内。