结合多任务学习的半监督病理图像分割方法

Semi-supervised Histopathological Image Segmentation Method Based on Multi-task Learning

  • 摘要: 病理图像自动分割是计算机辅助诊断技术的重要组成部分,可降低病理科医师工作负担,提高诊断效率和准确性。本文介绍一种结合多任务学习的半监督病理图像分割方法。该方法基于半监督的方式同时进行癌症区域图像分割与分类,即首先基于极少量像素级标注图像对分割网络进行训练,然后结合图像级标注图像同时完成图像分割及分类。在网络训练过程中,通过此2个任务的交替迭代以优化网络参数,降低了深度学习模型对图像标注的依赖性。在此基础上,模型引入了动态加权交叉熵损失函数,可利用分类预测概率值自动完成每个像素的权重分配,以提高分割网络对预测概率值较低目标区域的关注度。该策略可有效保持癌症区域的细节信息,经验证可在像素标注数据量不足的情况下对乳腺癌病理图像获得良好的癌症区域分割结果。

     

    Abstract: Automatic segmentation of histopathological image is an important step of computer-aided diagnosis, which can reduce the workload of pathologists and improve the efficiency and diagnosis accuracy. This paper introduces a semi-supervised histopathological image segmentation method combined with multi-task learning. This method could simultaneously segment cancer region and classify image using semi-supervised method. Firstly, we used a limited number of pixel-level labels to train a segmentation network, and then the segmentation network and a classification network were trained using some image-level labels. The network parameters were optimized by alternating iterative method in the process of the network training. This method could reduce the annotation cost compared with the standard supervised method for deep learning model. Furthermore, we introduced a dynamically weighted cross entropy loss function to train the network, which could automatically allocate the weight of each pixel by using the probability of classification prediction. This strategy could promote the segmentation network to pay attention to some target regions with the low probability of classification prediction. Therefore, the details of the cancer regions could be preserved. Experimental results on the breast cancer histopathological image verified that our method outperformed other state-of-the-arts on the cancer segmentation performance under the condition of insufficient pixel-level label data.

     

/

返回文章
返回