一种基于点标签学习的OCT图像积液分割方法技术

本发明专利技术公开了一种基于点标签学习的OCT图像积液分割方法。通过超像素算法，对点标注数据生成带噪标签；将少量有全标注数据的OCT图像和大量有带噪标签的OCT图像输入师生架构网络，训练得到带噪网络；采用自信学习模块识别噪声和标签修复模块对带噪标签进行细化得到去噪标签；将少量有全标注数据的图像和大量有去噪标签的图像输入师生架构网络，训练得到最终去噪网络。最后通过最终去噪网络完成OCT图像积液分割。这种两阶段的去噪方法可以使得最终的模型尽量避免学习到带噪标签中的噪声。终的模型尽量避免学习到带噪标签中的噪声。终的模型尽量避免学习到带噪标签中的噪声。

【技术实现步骤摘要】
一种基于点标签学习的OCT图像积液分割方法


[0001]本专利技术涉及计算机辅助病理诊断
，尤其涉及一种基于深度学习的弱监督领域的OCT图像积液分割的方法。

技术介绍

[0002]视网膜积液可以根据积聚的位置分为三种类型：视网膜内积液，视网膜下积液和色素上皮脱离。这些液体是年龄相关性黄斑变性和视网膜静脉阻塞这两种严重眼病的重要生物标志物。眼科医生可以通过观察患者视网膜OCT图像中积液的种类和面积的大小来判断病变，但无法准确量化积液并据此制定治疗方案。因此，一种能够自动分割积液区域的算法对于临床诊断尤为重要。早期的OCT积液分割使用传统技术，它们依赖于手工设计的特征。然而，这些传统方法往往容易受到图像质量变化的影响，需要广泛的领域知识，并且缺乏泛化能力。相比于传统分割方法需要依赖于精心制作的手工特征，卷积神经网络具有自动学习和提取图像特征的能力。然而，基于卷积神经网络的方法需要大量数据，并且医学图像中手动标注数据是一个高度专业化且耗时的过程。在实践中，从医学专家那里获取大量标记清晰的数据往往是困难的，这使得训练准确、稳定的卷积神经网络模型用于医学图像分割具有挑战性。如果没有足够的清晰标记的像素级注释，基于卷积神经网络的数据匮乏的分割方法往往难以拟合，从而导致性能下降。一种可靠的解决办法是使用弱标签，而如何利用这些不完整的弱标记信息进行训练，并取得良好的预测结果对辅助诊断显得尤为重要。

技术实现思路

[0003]本专利技术针对现有方法的不足，提供一种基于点标签学习的OCT图像积液分割方法。本专利技术主要由以下四部分构成分别是：通过超像素(Achanta R,Shaji A,Smith K,et al.SLIC superpixels compared to state
‑
of
‑
the
‑
art superpixel methods[J].IEEE transactions on pattern analysis and machine intelligence,2012,34(11):2274
‑
2282.)算法，对点标注数据生成带噪标签；将少量有全标注数据的OCT图像和大量有带噪标签的OCT图像输入师生架构网络，训练得到带噪网络；采用自信学习(Northcutt C,Jiang L,Chuang I.Confident learning:Estimating uncertainty in dataset labels[J].Journal of Artificial Intelligence Research,2021,70:1373
‑
1411.)模块识别噪声和标签修复模块对带噪标签进行细化得到去噪标签；将少量有全标注数据的图像和大量有去噪标签的图像输入师生架构网络，训练得到去噪网络。这种两阶段的去噪方法可以使得最终的模型尽量避免学习到带噪标签中的噪声。本专利技术的一种基于点标签学习的OCT图像分割方法通过以下技术特征实现的：
[0004]本专利技术的一种基于点标签学习的OCT图像积液分割方法，具体包括以下步骤：
[0005]步骤1，构建OCT图像积液分割数据集，并进行预处理；
[0006]从医院获取不同时期不同患者的数次OCT检查采集的数据，构成OCT图像积液分割
数据集的原始OCT图像。对所有的OCT图像做全标注和点标注构建OCT图像积液分割数据集。
[0007]步骤2，基于点标注生成带噪标签；
[0008]步骤3，构建OCT图像积液分割模型，使用少量全标签数据和带噪标签数据训练师生架构网络，得到初步训练后的师生架构网络；
[0009]所述的OCT图像积液分割模型包括师生架构网络、自信学习模块和标签修复模块。所述的师生架构网络包括学生网络、教师网络，学生网络通过梯度下降训练，教师网络则是通过学生网络参数迭代。学生网络和教师网络均使用Unet作为骨干网络。
[0010]所述的自信学习模块通过对带噪标签进行置信度估计从而检测标签错误。标签修复模块将错误标签替换为预测的伪标签从而细化带噪标签。
[0011]步骤4，通过自信学习模块找到带噪标签的错误，通过标签修复模块对找到带噪标签的错误进行修复得到去噪标签；
[0012]步骤5，通过全标签数据和去噪标签数据对初步训练后的师生架构网络进行进一步训练，得到最终去噪网络。最后通过最终去噪网络完成OCT图像积液分割。
[0013]进一步的，步骤1具体方法如下：
[0014]从医院获取包括不同时期不同患者的数次OCT检查的OCT图像。从每次OCT检查获得的OCT图像中选择包含丰富积液的OCT图像构成OCT图像积液分割数据集的原始OCT图像。将所有图像中心裁剪，分辨率保存为600x250，输入网络时，所有的图像大小调整为256x128。同一患者的数据只能用于训练或测试。
[0015]对所有的OCT图像做全标注和点标注构建OCT图像积液分割数据集的掩膜。所述的全标注就是对积液区域做逐像素标记。点标注是指对于积液部分，用点标记积液(视网膜内积液和视网膜下积液)的中心，用线(多个点)标记色素上皮脱离的底部。
[0016]进一步的，所述的从点标注生成带噪标签，首先将原始OCT图像超像素化得到超像素图。通过计算每个超像素块是否包含标注的点标签生成超像素标签。遍历每一个超像素标签中不为背景类别的超像素块作为初始超像素块定义为计算初始超像素块与其一个邻域超像素块定义为计算这两者包含的每个像素值对应像素点的个数：
[0017][0018][0019]其中和表示初始超像素块和邻域超像素块包含的每个像素值对应像素点的个数，符号表示正确返回1错误则返回0，j:S
j
＝ms和j:S
j
＝ns表示初始超像素块和邻域超像素块在原始图像中对应的像素点。随后，通过和计算两者余弦相似度：
[0020][0021]倘若余弦相似度大于阈值，将初始超像素块的标签赋值给邻域超像素块，并将该邻域超像素块的邻域超像素块也视为初始超像素块的邻域超像素块，直到所有的邻域超像素块与初始超像素块的余弦相似度都小于阈值。遍历每一个超像素标签中不为背景类别的超像素块作为初始超像素块。所有不为背景的初始超像素块处理完毕后，将超像素标签转化为带噪标签。并且生成额外像素级标签权重图，初始的标签权重图的值设置为0.5，随后把值1.0分配给初始超像素块对应的像素点，当余弦相似度大于阈值时更新信任值，每相差两个超像素块的实际距离就在1的基础上减少0.1分配给邻域超像素块对应的像素点。
[0022]进一步的，使用少量全标签数据和带噪标签数据训练师生架构网络，具体方法如下：
[0023]所述的师生架构的学生网络和教师网络结构基于Unet网络。网络的输入数据包含两部分：10％全标签数据和90％的带噪标签数据；全标签数据输入师生架构的学生网络，带噪标签数据输入学生网络和教师网络；
[0024]全标注数据的交叉熵损本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于点标签学习的OCT图像积液分割方法，其特征在于，包括以下步骤：步骤1，构建OCT图像积液分割数据集，并进行预处理；从医院获取不同时期不同患者的数次OCT检查采集的数据，构成OCT图像积液分割数据集的原始OCT图像；对所有的OCT图像做全标注和点标注构建OCT图像积液分割数据集；步骤2，基于点标注生成带噪标签；步骤3，构建OCT图像积液分割模型，使用少量全标签数据和带噪标签数据训练师生架构网络，得到初步训练后的师生架构网络；所述的OCT图像积液分割模型包括师生架构网络、自信学习模块和标签修复模块；所述的师生架构网络包括学生网络、教师网络，学生网络通过梯度下降训练，教师网络则是通过学生网络参数迭代；学生网络和教师网络均使用Unet作为骨干网络；所述的自信学习模块通过对带噪标签进行置信度估计从而检测标签错误；标签修复模块将错误标签替换为预测的伪标签从而细化带噪标签；步骤4，通过自信学习模块找到带噪标签的错误，通过标签修复模块对找到带噪标签的错误进行修复得到去噪标签；步骤5，通过全标签数据和去噪标签数据对初步训练后的师生架构网络进行进一步训练，得到最终去噪网络；最后通过最终去噪网络完成OCT图像积液分割。2.根据权利要求1所述的一种基于点标签学习的OCT图像积液分割方法，其特征在于，步骤1具体方法如下：从医院获取包括不同时期不同患者的数次OCT检查的OCT图像；从每次OCT检查获得的OCT图像中选择包含丰富积液的OCT图像构成OCT图像积液分割数据集的原始OCT图像；将所有图像中心裁剪，分辨率保存为600x250，输入网络时，所有的图像大小调整为256x128；同一患者的数据只能用于训练或测试；对所有的OCT图像做全标注和点标注构建OCT图像积液分割数据集的掩膜；所述的全标注就是对积液区域做逐像素标记；点标注是指对于积液部分，用点标记积液的中心，用线标记色素上皮脱离的底部。3.根据权利要求2所述的一种基于点标签学习的OCT图像积液分割方法，其特征在于，所述的从点标注生成带噪标签，首先将原始OCT图像超像素化得到超像素图；通过计算每个超像素块是否包含标注的点标签生成超像素标签；遍历每一个超像素标签中不为背景类别的超像素块做为初始超像素块定义为计算初始超像素块与其一个邻域超像素块定义为计算这两者包含的每个像素值对应像素点的个数：计算这两者包含的每个像素值对应像素点的个数：其中和表示初始超像素块和邻域超像素块包含的每个像素值对应像素点的个数，符号表示正确返回1错误则返回0，j:S
j
＝ms和j:S
j
＝ns...

【专利技术属性】
技术研发人员：翁腾缙，王帅，颜成钢，黄星儒，高宇涵，王健博，何帆，于姗姗，周振，沈洋，
申请(专利权)人：杭州电子科技大学，
类型：发明
国别省市：