一种用于前列腺癌辅助诊断的深度学习方法技术

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

A Deep Learning Method for Assisted Diagnosis of Prostate Cancer

【技术实现步骤摘要】
一种用于前列腺癌辅助诊断的深度学习方法
本专利技术涉及深度学习方法领域，具体是一种用于前列腺癌辅助诊断的深度学习方法。
技术介绍
如今，前列腺癌已经成为男性尤其是老年男性的健康问题的一大威胁，是近几十年来最常见的癌症之一。在临床诊断前列腺癌时，需要医生对前列腺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损失的定义式如下：其中，m为每次抽取作为训练样本的数量，tj为前列腺MR图像对应的人工分割图，pj为分割网络输出的分割结果，通过随机梯度下降算法来最小化(1)式。经过至少1000次的迭代更新，直至损失函数收敛，得到稳定的训练好的分割网络；(5)、将待分割的病人的前列腺MR图像输入进步骤(4)中得到的训练好的网络中，输出分割结果图；(6)、对(5)输出的分割结果图，进一步用条件随机场来平滑噪声，细化边缘细节，得到最终的分割结果，将此结果与正常的影像学形态进行对比，给出辅助诊断信息，从而应用于前列腺癌的诊断中。本专利技术通过从已知人工分割图的前列腺MR图像中进行训练的基础上，得到一个训练好的稳定的模型，来自动分割病人的前列腺MR图像，根据分割结果辅助医生诊断病人的情况，这种自动分割方法不仅提高了医生的工作效率，而且使诊断的准确率也有所提高。该技术可用于对前列腺癌的辅助诊断。本专利技术中，将输入图片大小调整为三个不同尺寸，通过预训练模型来提取图像的多尺度特征，并将多尺度特征进行融合，经过链式残差池化模块可以从大图像区域捕获背景上下文信息，使分割结果更准确。对构建的分割网络进行训练，训练结束可以得到一个稳定的优化的模型来自动分割前列腺MR图像，提高分割效率。本专利技术使用条件随机场来进行后处理，该方法对由分割网络输出的分割结果进行进一步的提升，可以平滑图像的噪声，细化分割的边缘细节等，使最终得到的分割图更加精细，准确率更高。本专利技术的有益效果是：本专利技术用前列腺MR图像训练一个分割网络模型，然后使用随机梯度下降算法对所构建的分割网络进行优化得到训练好的模型，用训练好的模型来对前列腺MR图像进行自动分割，并用条件随机场来进一步提升分割效果，辅助前列腺癌的临床诊断，可以有效提高分割精度，减轻医生的工作量，提高诊断准确率和诊断效率。附图说明图1是本专利技术的实现流程框架图。图2中的(a)为原始待分割的前列腺MR图像；(b)为(a)对应的人工分割图；(c)为用本专利技术的方法对(a)进行分割得到的结果图。图3为六例不同病人的前列腺分割的结果对比。图中灰色线表示人工分割结果，白色线表示本专利技术的分割结果。经过比较，本专利技术提出的分割方法十分接近人工分割结果，可以用来辅助医生进行诊断。具体实施方式如图1所示，一种用于前列腺癌辅助诊断的深度学习方法，过程如下：(1)、选取45个病人的686张前列腺MR图像以及对应的前列腺组织的人工分割图作为训练数据集；(2)、对数据集进行预处理，采用水平和垂直翻转以及调节亮度、对比度、饱和度的数据增强方式来扩展数据集，将扩展后的训练图片分别按原图的{1,0.75,0.5}的大小调整为3个尺度；(3)、将步骤(2)得到的多尺度图片输入到分割网络模型进行训练，此分割网络主要由ResNet预训练模型和链式残差池化模块构成。三个尺度的图片分别输入一个ResNet预训练模型，通过微调ResNet预训练模型的参数来提取输入图像的多尺度特征。将得到的三种不同尺度的特征图上采样到与原始图像相同大小并进行融合，然后把融合后的特征图输入链式残差池化模块，该模块是一个包含多个池化块的链，每个池化块由一个最大池化层和一个卷积层组成，一个池化块将前一个池化块的输出作为输入，使得当前池化块能再次利用前面池化块的结果，将所有池化块的输出特征图与输入特征图通过残差连接的方式融合在一起形成一个链式残差池化模块来从大图像区域捕获背景上下文信息；(4)、对(3)所构建的分割网络使用随机梯度下降(SGD)的方法进行优化，该算法是通过不断的求偏导求解出当前位置下最优化的数据。分割网络所采用的损失函数为BCE(二进制交叉熵)损失，该损失是通过惩罚输出的分割结果与对应的人工分割图之间的距离来定义的，所以最小化该损失即使分割网络产生的分割结果更接近于人工分割图。BCE损失的定义式如下：其中，m为每次抽取作为训练样本的数量，tj为前列腺MR图像对应的人工分割图，pj为分割网络输出的分割结果，通过随机梯度本文档来自技高网...

【技术保护点】
1.一种用于前列腺癌辅助诊断的深度学习方法，其特征在于：构建一个含有预训练模型和链式残差池化模块的分割网络模型，对输入的不同尺度的前列腺MR图像进行特征提取和融合，然后使用随机梯度下降算法对所构建的分割网络进行优化得到训练好的模型，再将待分割的前列腺MR图像输入进已训练的模型中，最后将该模型输出的结果通过条件随机场来输出最后的分割结果，其步骤如下：(1)、选取多个病人的前列腺MR图像作为训练数据集；(2)、对训练数据集中的图片采用水平和垂直翻转以及调节亮度、对比度、饱和度的数据增强方式来进行扩展，将扩展后的训练图片分别按原图的{1,0.75,0.5}的大小调整为3个尺度；(3)、将步骤(2)得到的多尺度图片输入到分割网络模型进行训练，此分割网络主要由ResNet预训练模型和链式残差池化模块构成；三个尺度的图片分别输入一个ResNet预训练模型，通过微调ResNet预训练模型的参数来提取输入图像的多尺度特征；将得到的三种不同尺度的特征图上采样到与原始图像相同大小并进行融合，然后把融合后的特征图输入链式残差池化模块；所述链式残差池化模块是一个包含多个池化块的链，每个池化块由一个最大池化层和一个卷积层组成，一个池化块将前一个池化块的输出作为输入，使得当前池化块能再次利用前面池化块的结果，将所有池化块的输出特征图与输入特征图通过残差连接的方式融合在一起形成一个链式残差池化模块来从大图像区域捕获背景上下文信息；(4)、对(3)所构建的分割网络使用随机梯度下降的方法进行优化，该算法是通过不断的求偏导求解出当前位置下最优化的数据；分割网络所采用的损失函数为BCE损失，该损失是通过惩罚输出的分割结果与对应的人工分割图之间的距离来定义的，所以最小化该损失即使分割网络产生的分割结果更接近于人工分割图；BCE损失的定义式如下：...

【技术特征摘要】
1.一种用于前列腺癌辅助诊断的深度学习方法，其特征在于：构建一个含有预训练模型和链式残差池化模块的分割网络模型，对输入的不同尺度的前列腺MR图像进行特征提取和融合，然后使用随机梯度下降算法对所构建的分割网络进行优化得到训练好的模型，再将待分割的前列腺MR图像输入进已训练的模型中，最后将该模型输出的结果通过条件随机场来输出最后的分割结果，其步骤如下：(1)、选取多个病人的前列腺MR图像作为训练数据集；(2)、对训练数据集中的图片采用水平和垂直翻转以及调节亮度、对比度、饱和度的数据增强方式来进行扩展，将扩展后的训练图片分别按原图的{1,0.75,0.5}的大小调整为3个尺度；(3)、将步骤(2)得到的多尺度图片输入到分割网络模型进行训练，此分割网络主要由ResNet预训练模型和链式残差池化模块构成；三个尺度的图片分别输入一个ResNet预训练模型，通过微调ResNet预训练模型的参数来提取输入图像的多尺度特征；将得到的三种不同尺度的特征图上采样到与原始图像相同大小并进行融合，然后把融合后的特征图输入链式残差池化模块；所述链式残差池化模块是一个包含多个池化块的链，每个池化块由一个最大池化层和一个卷积层组...

【专利技术属性】
技术研发人员：詹曙，陈爱莲，臧怀娟，
申请(专利权)人：合肥工业大学，
类型：发明
国别省市：安徽,34