论文标题
探索半监督医学图像细分的平滑度和类分离
Exploring Smoothness and Class-Separation for Semi-supervised Medical Image Segmentation
论文作者
论文摘要
半监督分割在医学成像中仍然具有挑战性,因为注释的医学数据量通常很少,并且在粘合剂边缘或低对比度区域附近或低对比度区域都有许多模糊的像素。为了解决这些问题,我们主张首先限制有和没有强大扰动的像素的一致性,以应用足够的平滑度约束,并进一步鼓励班级分离以利用低渗透拷贝的正则化来进行模型培训。特别是,在本文中,我们通过同时探索像素级的平滑度和类间的分离来提议半监督医学图像分割任务的SS-NET。像素级平滑度迫使模型在对抗扰动下产生不变结果。同时,阶层间的分离鼓励各个班级特征应接近其相应的高质量原型,以使每个类别的分布紧凑并分开不同的类。我们针对公共LA和ACDC数据集的五种最新方法评估了我们的SS-NET。在两个半监督的设置下进行的广泛实验结果证明了我们提出的SS-NET模型的优越性,在两个数据集中都实现了新的最先进(SOTA)性能。该代码可在https://github.com/ycwu1997/ss-net上找到。
Semi-supervised segmentation remains challenging in medical imaging since the amount of annotated medical data is often scarce and there are many blurred pixels near the adhesive edges or in the low-contrast regions. To address the issues, we advocate to firstly constrain the consistency of pixels with and without strong perturbations to apply a sufficient smoothness constraint and further encourage the class-level separation to exploit the low-entropy regularization for the model training. Particularly, in this paper, we propose the SS-Net for semi-supervised medical image segmentation tasks, via exploring the pixel-level smoothness and inter-class separation at the same time. The pixel-level smoothness forces the model to generate invariant results under adversarial perturbations. Meanwhile, the inter-class separation encourages individual class features should approach their corresponding high-quality prototypes, in order to make each class distribution compact and separate different classes. We evaluated our SS-Net against five recent methods on the public LA and ACDC datasets. Extensive experimental results under two semi-supervised settings demonstrate the superiority of our proposed SS-Net model, achieving new state-of-the-art (SOTA) performance on both datasets. The code is available at https://github.com/ycwu1997/SS-Net.