Semi-Supervised Multi-Organ Segmentation through Quality Assurance Supervision

by   Ho Hin Lee, et al.

Human in-the-loop quality assurance (QA) is typically performed after medical image segmentation to ensure that the systems are performing as intended, as well as identifying and excluding outliers. By performing QA on large-scale, previously unlabeled testing data, categorical QA scores can be generatedIn this paper, we propose a semi-supervised multi-organ segmentation deep neural network consisting of a traditional segmentation model generator and a QA involved discriminator. A large-scale dataset of 2027 volumes are used to train the generator, whose 2-D montage images and segmentation mask with QA scores are used to train the discriminator. To generate the QA scores, the 2-D montage images were reviewed manually and coded 0 (success), 1 (errors consistent with published performance), and 2 (gross failure). Then, the ResNet-18 network was trained with 1623 montage images in equal distribution of all three code labels and achieved an accuracy 94 images withheld for the test cohort. To assess the performance of using the QA supervision, the discriminator was used as a loss function in a multi-organ segmentation pipeline. The inclusion of QA-loss function boosted performance on the unlabeled test dataset from 714 patients to 951 patients over the baseline model. Additionally, the number of failures decreased from 606 (29.90 (19.83 (1) the QA scores can be used as a loss function to perform semi-supervised learning for unlabeled data, (2) the well trained discriminator is learnt by QA score rather than traditional true/false, and (3) the performance of multi-organ segmentation on unlabeled datasets can be fine-tuned with more robust and higher accuracy than the original baseline method.



page 3

page 6


Feature-enhanced Adversarial Semi-supervised Semantic Segmentation Network for Pulmonary Embolism Annotation

This study established a feature-enhanced adversarial semi-supervised se...

Semi-Supervised Medical Image Segmentation via Learning Consistency under Transformations

The scarcity of labeled data often limits the application of supervised ...

Adversarial Learning for Semi-Supervised Semantic Segmentation

We propose a method for semi-supervised semantic segmentation using the ...

Uncertainty-aware multi-view co-training for semi-supervised medical image segmentation and domain adaptation

Although having achieved great success in medical image segmentation, de...

Mars Terrain Segmentation with Less Labels

Planetary rover systems need to perform terrain segmentation to identify...

Outlier Guided Optimization of Abdominal Segmentation

Abdominal multi-organ segmentation of computed tomography (CT) images ha...

cGANs with Multi-Hinge Loss

We propose a new algorithm to incorporate class conditional information ...
This week in AI

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