Transductive image segmentation: Self-training and effect of uncertainty estimation

07/19/2021
by   Konstantinos Kamnitsas, et al.
11

Semi-supervised learning (SSL) uses unlabeled data during training to learn better models. Previous studies on SSL for medical image segmentation focused mostly on improving model generalization to unseen data. In some applications, however, our primary interest is not generalization but to obtain optimal predictions on a specific unlabeled database that is fully available during model development. Examples include population studies for extracting imaging phenotypes. This work investigates an often overlooked aspect of SSL, transduction. It focuses on the quality of predictions made on the unlabeled data of interest when they are included for optimization during training, rather than improving generalization. We focus on the self-training framework and explore its potential for transduction. We analyze it through the lens of Information Gain and reveal that learning benefits from the use of calibrated or under-confident models. Our extensive experiments on a large MRI database for multi-class segmentation of traumatic brain lesions shows promising results when comparing transductive with inductive predictions. We believe this study will inspire further research on transductive learning, a well-suited paradigm for medical image analysis.

READ FULL TEXT
research
06/24/2020

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

In medical image analysis, semi-supervised learning is an effective meth...
research
12/05/2021

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

Medical image segmentation is a fundamental and critical step in many cl...
research
06/28/2020

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...
research
05/25/2023

Cross-supervised Dual Classifiers for Semi-supervised Medical Image Segmentation

Semi-supervised medical image segmentation offers a promising solution f...
research
06/25/2019

Learning a sparse database for patch-based medical image segmentation

We introduce a functional for the learning of an optimal database for pa...
research
08/12/2022

Triple-View Feature Learning for Medical Image Segmentation

Deep learning models, e.g. supervised Encoder-Decoder style networks, ex...
research
07/29/2019

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

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

Please sign up or login with your details

Forgot password? Click here to reset