[发明专利]一种基于任务动态学习部分标记的医学图像分割方法

说明书

本发明公开了一种基于任务动态学习部分标记的医学图像分割方法，实现多器官和肿瘤的分割。该方法首先采用卷积神经网络搭建一个编码解码模块，以医学图像作为输入，提取图像的高级语义特征。接着通过一个任务编码模块，将不同任务对应的数据集进行编码，生成的one‑hot编码作为其任务先验。然后设计了一个控制器，以one‑hot编码和图像本身的特征为条件，为每一个图像生成特定任务的卷积核。最后，生成的卷积核在解码模块得到的特征图上进行卷积操作，得到对应任务的分割结果。本发明的分割模型可以在一个简单的分割网络下，高效地实现多个器官、多个肿瘤的同时分割并且能够巧妙地整合多个数据集的资源，能够实现更通用、泛化能力更强的多器官和肿瘤分割。

技术领域

本发明属于图像处理技术领域，具体涉及一种医学图像分割方法。

背景技术

医学图像分割是计算机视觉和医学影像分析领域共同关注的难题，其主要挑战在于标注代价高昂而导致的标注数据量不足、标注单一的问题。当前公开的医学图像数据集往往只提供一种类别器官或肿瘤的标注，即部分标注，而没有一个公开的大型全标注多器官数据集。比如LiTS肝脏及肿瘤分割数据集中，只提供了肝脏及其肿瘤的分割标注，其他的器官和肿瘤被简单当做背景处理。当前的主流医学图像分割模型均采用“一对一”的设计范式，即一个模型只能解决某一个数据集上提供标注的器官或肿瘤分割任务，粗暴地将其他的器官或肿瘤当做背景处理。当前亟需一种分割方法，不仅能够整合多个数据集，而且能够有效地解决其部分标记问题。

发明内容

为了克服现有技术的不足，本发明提供了一种基于任务动态学习部分标记的医学图像分割方法，实现多器官和肿瘤的分割。该方法首先采用卷积神经网络搭建一个编码解码模块，以医学图像作为输入，提取图像的高级语义特征。接着通过一个任务编码模块，将不同任务对应的数据集进行编码，生成的one-hot编码作为其任务先验。然后设计了一个控制器，以one-hot编码和图像本身的特征为条件，为每一个图像生成特定任务的卷积核。最后，生成的卷积核在解码模块得到的特征图上进行卷积操作，得到对应任务的分割结果。本发明的分割模型可以在一个简单的分割网络下，高效地实现多个器官、多个肿瘤的同时分割并且能够巧妙地整合多个数据集的资源，能够实现更通用、泛化能力更强的多器官和肿瘤分割。

本发明解决其技术问题所采用的技术方案包括以下步骤：

步骤1：采用编码解码器提取图像的特征；

采用卷积神经网络构建编码-解码器；

给定图像X_ij，i表示图像数据集的索引，j表示数据集i里图像的索引；将图像X_ij输入编码器生成图像X_ij的高级语义特征F_ij＝f_E(X_ij；θ_E)，f_E(.)表示编码器，θ_E表示编码器参数；再输入解码器通过上采样操作将图像X_ij恢复到原始分辨率，得到预分割特征M_ij＝f_D(X_ij；θ_D)，f_D(.)表示解码器，θ_D表示解码器参数；

步骤2：对图像的部分标注信息进行任务编码；

将图像X_ij的部分标注信息编码到一个m维度的one-hot向量T_ij∈{0,1}^m作为任务编码，1表示带有标注，0表示没有标注；

步骤3：以任务编码作为条件，设计控制器为每一幅图像生成相应任务的卷积核参数；

所述控制器为单层卷积层或多个卷积层堆叠而成；