[发明专利]一种用于前列腺癌辅助诊断的深度学习方法

说明书

本发明公开了一种用于前列腺癌辅助诊断的深度学习方法，构建一个含有预训练模型和链式残差池化模块的分割网络模型，对输入的不同尺度的前列腺MR图像进行特征提取和融合，然后使用随机梯度下降算法对所构建的分割网络进行优化得到训练好的模型，再将待分割的前列腺MR图像输入进已训练的模型中，最后将该模型输出的结果通过条件随机场来输出最后的分割结果。本发明方法的分割结果可以辅助医生对前列腺癌进行临床诊断与治疗，有效提高医生的诊断准确率和工作效率。

技术领域

本发明涉及深度学习方法领域，具体是一种用于前列腺癌辅助诊断的深度学习方法。

背景技术

如今，前列腺癌已经成为男性尤其是老年男性的健康问题的一大威胁，是近几十年来最常见的癌症之一。在临床诊断前列腺癌时，需要医生对前列腺MR图像中的前列腺组织与周围的组织器官分离出来以便于诊断与治疗。这个分割的过程所提取的有意义的信息包括形状、器官的相对位置、体积和异常。尽管MR图像中软组织器官之间的对比比计算机断层扫描要好，由于在MR图像中属于前列腺组织的区域很小，可用的有效信息较少，而且每位患者的前列腺组织的大小，形状，位置都各不相同，对于人类观察者来说，对前列腺边界的精确描绘仍然是比较困难的。因此前列腺MR图像的自动分割成为一项具有挑战性的工作。

基于深度学习方法的前列腺组织分割技术，是指用前列腺MR图像训练一个分割网络模型，然后使用随机梯度下降算法对所构建的分割网络进行优化得到训练好的模型，用训练好的模型来对前列腺MR图像进行自动分割，并用条件随机场来提升分割效果，得到最终的分割结果。这种自动分割技术应用到临床诊断中，相对于人工分割，精度高，耗时少，可以大大提高临床工作的效率。

发明内容

本发明的目的是提供一种用于前列腺癌辅助诊断的深度学习方法，以解决现有技术对前列腺组织分割精度低、耗时高的问题。

为了达到上述目的，本发明所采用的技术方案为：

一种用于前列腺癌辅助诊断的深度学习方法，其特征在于：构建一个含有预训练模型和链式残差池化模块的分割网络模型，对输入的不同尺度的前列腺MR图像进行特征提取和融合，然后使用随机梯度下降算法对所构建的分割网络进行优化得到训练好的模型，再将待分割的前列腺MR图像输入进已训练的模型中，最后将该模型输出的结果通过条件随机场来输出最后的分割结果，其步骤如下：

(1)、选取多个病人的前列腺MR图像作为训练数据集；

(2)、对训练数据集中的图片采用水平和垂直翻转以及调节亮度、对比度、饱和度的数据增强方式来进行扩展，将扩展后的训练图片分别按原图的{1,0.75,0.5}的大小调整为3个尺度；

(3)、将步骤(2)得到的多尺度图片输入到分割网络模型进行训练，此分割网络主要由ResNet预训练模型和链式残差池化模块构成。三个尺度的图片分别输入一个ResNet预训练模型，通过微调ResNet预训练模型的参数来提取输入图像的多尺度特征；将得到的三种不同尺度的特征图上采样到与原始图像相同大小并进行融合，然后把融合后的特征图输入链式残差池化模块，该模块是一个包含多个池化块的链，每个池化块由一个最大池化层和一个卷积层组成，一个池化块将前一个池化块的输出作为输入，使得当前池化块能再次利用前面池化块的结果，将所有池化块的输出特征图与输入特征图通过残差连接的方式融合在一起形成一个链式残差池化模块来从大图像区域捕获背景上下文信息；

(4)、对(3)所构建的分割网络使用随机梯度下降(SGD)的方法进行优化，该算法是通过不断的求偏导求解出当前位置下最优化的数据。分割网络所采用的损失函数为BCE(二进制交叉熵)损失，该损失是通过惩罚输出的分割结果与对应的人工分割图之间的距离来定义的，所以最小化该损失即使分割网络产生的分割结果更接近于人工分割图；BCE损失的定义式如下：