arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2107.04886cs.CV
收藏

基于多域数据融合的分层自监督学习医学图像分割

Hierarchical Self-Supervised Learning for Medical Image Segmentation Based on Multi-Domain Data Aggregation

University of Notre Dame(圣母大学)

Hao Zheng, Jun Han, Hongxiao Wang, Lin Yang, Zhuo Zhao, Chaoli Wang, Danny Z. Chen

摘要:

大型标记数据集是监督式深度学习成功的关键,但对于医学图像分割,获得足够的注释图像用于模型训练具有高度挑战性。在许多情况下,未注释的图像是丰富的,易于获取。自监督学习在挖掘原始数据信息和表示学习方面显示出巨大的潜力。在本文中,我们提出了分层自监督学习(HSSL),一个新的自监督框架,提高医学图像分割,充分利用未注释的数据。与当前关于特定任务的自监督预训练以及随后的监督微调的文献不同,我们利用SSL从各种医学图像分割任务的异构数据中学习与任务无关的知识。具体来说,我们首先从几个医疗挑战中聚合一个数据集,然后以自我监督的方式对网络进行预训练,最后对标记数据进行微调。我们开发了一个新的损失函数,结合对比损失和分类损失和预训练的编码器-解码器架构的分割任务。我们的大量实验表明,多域联合预训练有利于下游分割任务,并显着优于单域预训练。与从头开始学习相比,我们的新方法在各种任务上都有更好的性能(例如,+0.69%至+18.60%(Dice评分,注释数据为5%)。在有限的训练数据的情况下,我们的方法可以实质性地弥合性能差距。更密集的注释(例如,10%与~ 100%的注释数据)。

英文摘要:

A large labeled dataset is a key to the success of supervised deep learning, but for medical image segmentation, it is highly challenging to obtain sufficient annotated images for model training. In many scenarios, unannotated images are abundant and easy to acquire. Self-supervised learning (SSL) has shown great potentials in exploiting raw data information and representation learning. In this paper, we propose Hierarchical Self-Supervised Learning (HSSL), a new self-supervised framework that boosts medical image segmentation by making good use of unannotated data. Unlike the current literature on task-specific self-supervised pretraining followed by supervised fine-tuning, we utilize SSL to learn task-agnostic knowledge from heterogeneous data for various medical image segmentation tasks. Specifically, we first aggregate a dataset from several medical challenges, then pre-train the network in a self-supervised manner, and finally fine-tune on labeled data. We develop a new loss function by combining contrastive loss and classification loss and pretrain an encoder-decoder architecture for segmentation tasks. Our extensive experiments show that multi-domain joint pre-training benefits downstream segmentation tasks and outperforms single-domain pre-training significantly. Compared to learning from scratch, our new method yields better performance on various tasks (e.g., +0.69% to +18.60% in Dice scores with 5% of annotated data). With limited amounts of training data, our method can substantially bridge the performance gap w.r.t. denser annotations (e.g., 10% vs.~100% of annotated data).

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑