论文标题
超级像素引导的标签软化,用于医疗图像分割
Superpixel-Guided Label Softening for Medical Image Segmentation
论文作者
论文摘要
感兴趣对象的分割是医学图像分析中的核心任务之一,这对于定量分析是必不可少的。在开发基于机器学习的自动分割方法时,通常将手动注释用作模型学会模仿的基础真理。尽管分割目标的庞大部分相对易于标记,但由于歧义的边界和部分体积效应等,外围区域通常很难处理,并且很可能被不确定性标记。标签中的这种不确定性可能会导致训练有素的模型的性能不令人满意。在本文中,我们提出了基于超像素的标签软化,以解决上述问题。通过无监督的过度分割产生,每个超像素预计将代表局部均匀的区域。如果超像素与注释边界相交,我们会考虑在该区域内标记不确定的概率。在这种直觉的驱动下,我们根据签名的距离对注释边界的签名距离软化了标签,并将[0,1]内的概率值分配给它们,与原始的“硬”标签0或1的二进制标签相比。然后,软化标签可用于将分割模型与硬标签一起训练分割模型。大脑MRI数据集和光学相干断层扫描数据集的实验结果表明,这种在概念上简单且实现的简单方法可以实现3D和2D医学图像的基线和比较方法的总体上级分段性能。
Segmentation of objects of interest is one of the central tasks in medical image analysis, which is indispensable for quantitative analysis. When developing machine-learning based methods for automated segmentation, manual annotations are usually used as the ground truth toward which the models learn to mimic. While the bulky parts of the segmentation targets are relatively easy to label, the peripheral areas are often difficult to handle due to ambiguous boundaries and the partial volume effect, etc., and are likely to be labeled with uncertainty. This uncertainty in labeling may, in turn, result in unsatisfactory performance of the trained models. In this paper, we propose superpixel-based label softening to tackle the above issue. Generated by unsupervised over-segmentation, each superpixel is expected to represent a locally homogeneous area. If a superpixel intersects with the annotation boundary, we consider a high probability of uncertain labeling within this area. Driven by this intuition, we soften labels in this area based on signed distances to the annotation boundary and assign probability values within [0, 1] to them, in comparison with the original "hard", binary labels of either 0 or 1. The softened labels are then used to train the segmentation models together with the hard labels. Experimental results on a brain MRI dataset and an optical coherence tomography dataset demonstrate that this conceptually simple and implementation-wise easy method achieves overall superior segmentation performances to baseline and comparison methods for both 3D and 2D medical images.