Wang Duo, Li Ming, Ben-Shlomo Nir, Corrales C Eduardo, Cheng Yu, Zhang Tao, Jayender Jagadeesan
Department of Automation, Tsinghua University, Beijing, China.
Department of Radiology, Brigham and Women's Hospital, Boston, USA.
Med Image Comput Comput Assist Interv. 2019 Oct;11765:192-200. doi: 10.1007/978-3-030-32245-8_22. Epub 2019 Oct 10.
Deep learning based medical image segmentation models usually require large datasets with high-quality dense segmentations to train, which are very time-consuming and expensive to prepare. One way to tackle this difficulty is using the mixed-supervised learning framework, where only a part of data is densely annotated with segmentation label and the rest is weakly labeled with bounding boxes. The model is trained jointly in a multi-task learning setting. In this paper, we propose Mixed-Supervised Dual-Network (MSDN), a novel architecture which consists of two separate networks for the detection and segmentation tasks respectively, and a series of connection modules between the layers of the two networks. These connection modules are used to transfer useful information from the auxiliary detection task to help the segmentation task. We propose to use a recent technique called 'Squeeze and Excitation' in the connection module to boost the transfer. We conduct experiments on two medical image segmentation datasets. The proposed MSDN model outperforms multiple baselines.
基于深度学习的医学图像分割模型通常需要大量带有高质量密集分割标注的数据集来进行训练,而准备这些数据集非常耗时且成本高昂。解决这一难题的一种方法是使用混合监督学习框架,其中只有一部分数据带有密集的分割标签标注,其余数据则用边界框进行弱标注。该模型在多任务学习设置中进行联合训练。在本文中,我们提出了混合监督双网络(MSDN),这是一种新颖的架构,它分别由两个用于检测和分割任务的独立网络以及两个网络层之间的一系列连接模块组成。这些连接模块用于从辅助检测任务中传递有用信息,以帮助分割任务。我们建议在连接模块中使用一种名为“挤压与激励”的最新技术来增强这种传递。我们在两个医学图像分割数据集上进行了实验。所提出的MSDN模型优于多个基线模型。