[发明专利]一种基于转导推理的医学影像小样本分割方法

说明书

本发明公开了一种基于转导推理的医学影像小样本分割方法，包括以下步骤；(1)使用神经网络作为特征提取网络，提取二维医学影像或者三维影像切片的特征图；(2)基于转导推理对支撑集数据与查询集数据进行融合分析，并从中分别提取出用于对查询集进行分割的前景类别对应的原型特征向量与背景类别对应原型特征向量；(3)对查询集的特征图中的每一个特征，分别计算其与各个类别的原型特征相似度，然后通过比较各个类别相似度的大小，确定查询集中各个特征对应的类别，即其类别为相似度最大的原型特征所对应的类别。本发明能够提升分割效果，实现分割结果的精度最大化。

技术领域

本发明属于人工智能、计算机视觉与医学影像分析技术领域，具体涉及一种基于转导推理的医学影像小样本分割方法。

背景技术

随着医学影像成像技术的发展，尤其是CT、MRI等成像手段的出现，极大的促进了医疗诊断与治疗手段的进步，同时也极大地促进了医学影像自动化分析算法的发展，其中，基于深度学习的计算机视觉算法在医学影像相关的分类、检测、分割等任务中，表现亮眼，可以为医生提供辅助甚至于实现自动化诊断算法，大幅提成现代医疗水平。但常见的基于深度学习的医学影像分割方法《U-Net:ConvolutionalNetworksforBiomedicalImageSegmentation》，通常需要大量的数据进行学习，才能针对有限范围内的病灶或者器官进行分割，这导致了U-Net难以在实际场景中被运用。同时，影像数据本身受到隐私、伦理等方面的限制，且标注则需要专业的医师花费大量时间进行标注，这两个限制都使得传统的医学影像分割算法的研发与推广面领着较大的困难。

据发明人了解，针对目前医学影像分割算法通常需要使用大量数据进行训练的问题，尽管Ouyang等研究人员提出了诸如《Self-supervisionwithSuperpixels:TrainingFew-ShotMedicalImageSegmentationWithoutAnnotation》等基于采用小样本的医学影像分割方法作为解决方案，但是现有的方法仅从支撑集提取任务关键信息用于分割任务，而无法有效的利用查询集自身的信息，同时忽略了支撑集与查询集之间的分布差异，导致分割效果受限，且需要较多的支撑数据。

发明内容

为了克服上述现有技术存在的不足，本发明的目的在于提供一种基于转导推理的医学影像小样本分割方法，通过转导推理分析支撑集与查询集数据特征，提取更加有效的原型特征，对没有学习过的分割任务所涉及的区域进行分割，能够提升分割效果，实现分割结果的精度最大化。

为了实现上述目的，本发明采用的技术方案是：

一种基于转导推理的医学影像小样本分割方法，其中支撑集数据与查询集数据作为输入，支撑集为带有分割标注的影像数据，查询集为带分割的影像数据，分割方法包括以下步骤；

(1)使用神经网络作为特征提取网络，提取二维医学影像或者三维影像切片x的特征图z；

(2)基于转导推理对支撑集数据与查询集数据进行融合分析，并从中分别提取出用于对查询集进行分割的前景类别对应的原型特征向量与背景类别对应原型特征向量；

(2a)采用标注信息嵌入网络，将影像对应的分割标注y作为输入构建影像的标注信息v；

(2b)使用信息融合网络，将影像对应的特征图z与通过影像标注构建的标注信息v作为输入，对这两种信息进行融合，形成融合信息u；

(2c)通过转导推理模块，将支撑集与查询集的融合信息u^S与u^Q作为输入，并通过转导推理模块进行融合、学习，然后对输入的查询集融合信息u^Q进行补全，并生成新的查询集融合信息

(2d)采用解算网络，对转导推理模块的输出的融合信息进行处理，来解码其中的潜在的查询集标注信息v^Q；