基于深度学习的视网膜层和积液区域的层分割方法及系统与流程

文档序号:21995612发布日期:2020-08-25 19:37阅读:来源:国知局

技术特征:

1.一种基于深度学习的视网膜层和积液区域的层分割方法,其特征在于,所述方法包括如下步骤:

获取医疗系统中各节点区域的视网膜oct数据集,将所述视网膜oct数据集划分为预训练数据集及测试数据集,并对所述预训练数据集中的数据进行随机平移,以得到训练数据集;

根据构建的改进型unet分割网络及该分割网络对应的损失函数,对分批次送入该分割网络的训练数据集中的数据进行前向传播,以得到分割预测图;

根据联合损失函数公式计算分割预测图与专家像素级标记图像进行one-hot编码后的标准概率图之间的联合损失值,将所述联合损失值进行反向传播,并通过预设周期长度的迭代训练得到分割网络模型;

通过所述测试数据集对分割网络模型进行测试,以验证所述分割网络模型的可靠性。

2.根据权利要求1所述的基于深度学习的视网膜层和积液区域的层分割方法,其特征在于,将所述视网膜oct数据集划分为预训练数据集及测试数据集,并对所述预训练数据集中的数据进行随机平移,以得到训练数据集的方法包括:

分别将第一数量患者及第二数量患者的视网膜oct图像及其对应的专家像素级标记图像裁剪填充为预设大小尺寸后,作为预训练数据集及测试数据集,其中,所述视网膜oct图像与专家像素级标记图像的尺寸一致;

将所述预训练数据集中的视网膜oct图像及其对应的专家像素级标记图像进行随机平移操作,以得到训练数据集。

3.根据权利要求1所述的基于深度学习的视网膜层和积液区域的层分割方法,其特征在于,根据构建的改进型unet分割网络及该分割网络对应的损失函数,对分批次送入该分割网络的训练数据集中的数据进行前向传播,以得到分割预测图的方法包括:

在unet网络中的每个编码模块和解码模块的3*3卷积块之前增加一个1*1卷积层,在该1*1卷积层之后分别建立一个se模块和空分支,se模块的分支和3*3卷积块并联,空分支和3*3卷积块的输出端连接,以得到改进型unet分割网络,其中所述3*3卷积块为两个3*3卷积层的串联,se模块用于获取每个特征通道的重要程度;

将所述训练数据集中的数据输入改进型unet分割网络后,先通过第一个改进后的编码块,以得到经1*1卷积层输出的第一特征图、经3*3卷积块输出的第二特征图及经se模块输出的第三特征图;

将所述第一特征图、第二特征图及第三特征图以残差的方式进行相加融合得到的特征图作为第一个下采样层即2*2最大池化层的输入,再通过分割网络剩余池化层和卷积层等的前向传播以得到分割预测图。

4.根据权利要求1所述的基于深度学习的视网膜层和积液区域的层分割方法,其特征在于,所述分割网络对应的损失函数为:

lseg=λmcelmce+λdiceldice

其中,λmce、λdice为权重系数,lmce、ldice分别为加权多类交叉熵损失函数、dice损失函数。

5.根据权利要求1所述的基于深度学习的视网膜层和积液区域的层分割方法,其特征在于,所述分割预测图的大小为n*c*h*w,其中,n为图像批次尺寸,c为分割类别数,h为图像高度,w为图像宽度。

6.根据权利要求1所述的基于深度学习的视网膜层和积液区域的层分割方法,其特征在于,将所述联合损失值进行反向传播,并通过预设周期长度的迭代训练得到分割网络模型的方法包括:

通过所述损失函数对网络权重参数进行逐层求偏导,并利用优化方法和学习率策略更新分割网络权重参数,以得到更新分割网络;

将下一批视网膜oct图像送入所述更新分割网络中进行前向传播,以得到分割预测概率图;

计算所述分割概率图与相对应的标准概率图之间的联合损失值,并利用集成了sgd的一阶动量和rmsprop的二阶动量的adam优化器进行优化,在预设周期内不断迭代,直至最小化损失函数值时,将当前的更新分割网络作为分割网络模型。

7.根据权利要求1所述的基于深度学习的视网膜层和积液区域的层分割方法,其特征在于,所述联合损失值用来表征分割预测图与其相对应的专家像素级标记图像经过one-hot编码后的标准概率图之间的误差;所述one-hot编码为将专家像素级标记图像中各个类别的像素用一个通道的特征图表示,在每个通道中,除了该类别的像素的对应位置为1,其余均为0。

8.一种基于深度学习的视网膜层和积液区域的层分割系统,其特征在于,所述系统包括:

获取模块,用于获取医疗系统中各节点区域的视网膜oct数据集,将所述视网膜oct数据集划分为预训练数据集及测试数据集,并对所述预训练数据集中的数据进行随机平移,以得到训练数据集;

分割模块,用于根据构建的改进型unet分割网络及该分割网络对应的损失函数,对分批次送入该分割网络的训练数据集中的数据进行前向传播,以得到分割预测图;

训练模块,用于根据联合损失函数公式计算分割预测图与专家像素级标记图像进行one-hot编码后的标准概率图之间的联合损失值,将所述联合损失值进行反向传播,并通过预设周期长度的迭代训练得到分割网络模型;

测试模块,用于通过所述测试数据集对分割网络模型进行测试,以验证所述分割网络模型的可靠性。

9.一种存储介质,其上存储有计算机程序,其特征在于,该程序被处理器执行时实现如权利要求1至7任意一项所述的方法。

10.一种智能设备,包括存储器、处理器以及存储在存储器上并可在处理器上运行的计算机程序,其特征在于,所述处理器执行所述程序时实现如权利要求1至7任意一项所述的方法。


技术总结
本发明公开了一种基于深度学习的视网膜层和积液区域的层分割方法及系统,所述方法包括:获取医疗系统中各节点区域的视网膜OCT数据集,将该数据集划分为预训练数据集及测试数据集,对预训练数据集中的数据进行随机平移,得到训练数据集;根据构建的分割网络及对应的损失函数对分批次送入该分割网络的训练数据集中的数据进行前向传播,得到分割预测图;根据联合损失函数公式计算分割预测图与专家像素级标记图像进行one‑hot编码后的标准概率图之间的联合损失值,将联合损失值反向传播,通过预设周期长度的迭代训练得到分割网络模型;通过测试数据集对分割网络模型进行测试,以验证分割网络模型的可靠性。本发明能够提高分割网络泛化能力与类别分割准确率。

技术研发人员:梁姗姗;岳孟挺;李新宇;张军
受保护的技术使用者:中山大学
技术研发日:2020.04.20
技术公布日:2020.08.25
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1