论文标题
通过少量器官数据集的共同训练重量平均模型进行多器官分割
Multi-organ Segmentation via Co-training Weight-averaged Models from Few-organ Datasets
论文作者
论文摘要
多器官细分在许多临床应用中都有广泛的应用。要分割多个感兴趣的器官,通常很难在同一图像上收集所有器官的完整注释,因为某些医疗中心可能仅由于其自己的临床实践而只会注释一部分器官。在大多数情况下,可能会从一个训练集中获得单个或几个器官的注释,并从另一组训练图像中获取其他器官的注释。现有方法主要是为每个器官的每个子集训练和部署单个模型,这些器官的内存密集型且效率低下。在本文中,我们建议共同培训权重平均模型,以从少数器官数据集学习统一的多器官分割网络。我们协作培训两个网络,并让耦合的网络在未经通知的器官上相互教导。为了减轻网络之间嘈杂的教学监督,采用了加权平均模型来产生更可靠的软标签。此外,还利用一种新型的区域面具来选择性地对需要协作教学的未注销器官区域进行一致的约束,从而进一步提高了性能。来自MOBA的三个公共单器数据集的三个公共单器数据集进行了广泛的实验表明,我们的方法可以更好地利用少数器官数据集,并以较少的推理计算成本来实现卓越的性能。
Multi-organ segmentation has extensive applications in many clinical applications. To segment multiple organs of interest, it is generally quite difficult to collect full annotations of all the organs on the same images, as some medical centers might only annotate a portion of the organs due to their own clinical practice. In most scenarios, one might obtain annotations of a single or a few organs from one training set, and obtain annotations of the the other organs from another set of training images. Existing approaches mostly train and deploy a single model for each subset of organs, which are memory intensive and also time inefficient. In this paper, we propose to co-train weight-averaged models for learning a unified multi-organ segmentation network from few-organ datasets. We collaboratively train two networks and let the coupled networks teach each other on un-annotated organs. To alleviate the noisy teaching supervisions between the networks, the weighted-averaged models are adopted to produce more reliable soft labels. In addition, a novel region mask is utilized to selectively apply the consistent constraint on the un-annotated organ regions that require collaborative teaching, which further boosts the performance. Extensive experiments on three public available single-organ datasets LiTS, KiTS, Pancreas and manually-constructed single-organ datasets from MOBA show that our method can better utilize the few-organ datasets and achieves superior performance with less inference computational cost.