arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2009.04448cs.CV
收藏

基于双任务一致性的半监督医学图像分割

Semi-supervised Medical Image Segmentation through Dual-task Consistency

University of Electronic Science and Technology of China(电子科技大学);SenseTime Research(商汤科技研究院);Tongji University(同济大学)

Xiangde Luo, Jieneng Chen, Tao Song, Yinan Chen, Guotai Wang, Shaoting Zhang

AI总结:

针对现有半监督医学图像分割方法多依赖网络/数据扰动正则化的问题,提出基于双任务一致性的半监督框架,通过分割与水平集表示双任务的一致性约束利用无标注数据,性能超越现有最优方法。

摘要:

基于深度学习的半监督学习(SSL)算法已在医学图像分割中取得了可观成效,可通过利用无标注数据减轻医生的高成本标注负担。然而,现有文献中大多数SSL算法往往通过扰动网络和/或数据来正则化模型训练。观察到多/双任务学习会关注不同层级的信息,这些信息本身就存在预测扰动,我们在本研究中提出问题:能否为SSL显式构建任务级正则化,而非隐式构造网络级和/或数据级的扰动与变换?为回答该问题,我们首次提出了一种新颖的双任务一致性半监督框架。具体而言,我们采用双任务深度网络,联合预测目标的像素级分割图和几何感知的水平集表示。水平集表示通过可微任务变换层转换为近似分割图。同时,我们针对有标注和无标注数据,在水平集生成的分割图与直接预测的分割图之间引入双任务一致性正则化。在两个公开数据集上的大量实验表明,我们的方法通过融入无标注数据可大幅提升性能,且该框架优于当前最先进的半监督医学图像分割方法。代码可在以下链接获取:https://github.com/Luoxd1996/DTC

英文摘要:

Deep learning-based semi-supervised learning (SSL) algorithms have led to promising results in medical images segmentation and can alleviate doctors' expensive annotations by leveraging unlabeled data. However, most of the existing SSL algorithms in literature tend to regularize the model training by perturbing networks and/or data. Observing that multi/dual-task learning attends to various levels of information which have inherent prediction perturbation, we ask the question in this work: can we explicitly build task-level regularization rather than implicitly constructing networks- and/or data-level perturbation-and-transformation for SSL? To answer this question, we propose a novel dual-task-consistency semi-supervised framework for the first time. Concretely, we use a dual-task deep network that jointly predicts a pixel-wise segmentation map and a geometry-aware level set representation of the target. The level set representation is converted to an approximated segmentation map through a differentiable task transform layer. Simultaneously, we introduce a dual-task consistency regularization between the level set-derived segmentation maps and directly predicted segmentation maps for both labeled and unlabeled data. Extensive experiments on two public datasets show that our method can largely improve the performance by incorporating the unlabeled data. Meanwhile, our framework outperforms the state-of-the-art semi-supervised medical image segmentation methods. Code is available at: https://github.com/Luoxd1996/DTC

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

↑