论文标题
混合监督的半监督医学图像细分的教师学生框架
A Teacher-Student Framework for Semi-supervised Medical Image Segmentation From Mixed Supervision
论文作者
论文摘要
基于全面监督卷积网络的医学图像的标准分割需要准确的致密注释。这种学习框架建立在艰苦的手动注释上,并限制了对专业知识的需求,从而导致高质量标签不足。为了克服这种局限性并利用大量弱标记的数据,我们放松了刚性的标签要求,并基于基于教师学生的器官和病变细分的方式制定了半监督的学习框架,并具有部分密集标记的监督和补充的松散边界框,并更容易获得。在大多数情况下,观察器官及其内部病变的几何关系,我们提出了教师分段中的分层器官到质量(O2L)注意模块,以产生伪标记。然后,对学生进行了分段,并通过手动标记和伪标记的注释进行了培训。我们进一步提出了一个本地化分支,该分支通过在深层解码器中的高级特征聚集来预测器官和病变的位置,从而丰富了学生进行精确的定位信息。我们通过消融研究在模型中验证了LITS挑战数据集的每个设计,并显示了与最新方法相比其最先进的性能。我们显示我们的模型对边界框的质量非常强大,并且与全面监督的学习方法相比,实现了可比的性能。
Standard segmentation of medical images based on full-supervised convolutional networks demands accurate dense annotations. Such learning framework is built on laborious manual annotation with restrict demands for expertise, leading to insufficient high-quality labels. To overcome such limitation and exploit massive weakly labeled data, we relaxed the rigid labeling requirement and developed a semi-supervised learning framework based on a teacher-student fashion for organ and lesion segmentation with partial dense-labeled supervision and supplementary loose bounding-box supervision which are easier to acquire. Observing the geometrical relation of an organ and its inner lesions in most cases, we propose a hierarchical organ-to-lesion (O2L) attention module in a teacher segmentor to produce pseudo-labels. Then a student segmentor is trained with combinations of manual-labeled and pseudo-labeled annotations. We further proposed a localization branch realized via an aggregation of high-level features in a deep decoder to predict locations of organ and lesion, which enriches student segmentor with precise localization information. We validated each design in our model on LiTS challenge datasets by ablation study and showed its state-of-the-art performance compared with recent methods. We show our model is robust to the quality of bounding box and achieves comparable performance compared with full-supervised learning methods.