Dual-Task Mutual Learning for Semi-Supervised Medical Image Segmentation

by   Yichi Zhang, et al.

The success of deep learning methods in medical image segmentation tasks usually requires a large amount of labeled data. However, obtaining reliable annotations is expensive and time-consuming. Semi-supervised learning has attracted much attention in medical image segmentation by taking the advantage of unlabeled data which is much easier to acquire. In this paper, we propose a novel dual-task mutual learning framework for semi-supervised medical image segmentation. Our framework can be formulated as an integration of two individual segmentation networks based on two tasks: learning region-based shape constraint and learning boundary-based surface mismatch. Different from the one-way transfer between teacher and student networks, an ensemble of dual-task students can learn collaboratively and implicitly explore useful knowledge from each other during the training process. By jointly learning the segmentation probability maps and signed distance maps of targets, our framework can enforce the geometric shape constraint and learn more reliable information. Experimental results demonstrate that our method achieves performance gains by leveraging unlabeled data and outperforms the state-of-the-art semi-supervised segmentation methods.



There are no comments yet.


page 8


Uncertainty-Guided Mutual Consistency Learning for Semi-Supervised Medical Image Segmentation

Medical image segmentation is a fundamental and critical step in many cl...

Semi-supervised Medical Image Segmentation via Geometry-aware Consistency Training

The performance of supervised deep learning methods for medical image se...

Uncertainty-aware Self-ensembling Model for Semi-supervised 3D Left Atrium Segmentation

Training deep convolutional neural networks usually requires a large amo...

ATSO: Asynchronous Teacher-Student Optimizationfor Semi-Supervised Medical Image Segmentation

In medical image analysis, semi-supervised learning is an effective meth...

Mixed-Supervised Dual-Network for Medical Image Segmentation

Deep learning-based medical image segmentation models usually require la...

Multi-Task Attention-Based Semi-Supervised Learning for Medical Image Segmentation

We propose a novel semi-supervised image segmentation method that simult...

Semi-supervised Anatomical Landmark Detection via Shape-regulated Self-training

Well-annotated medical images are costly and sometimes even impossible t...

Code Repositories

This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.