Active Learning for Segmentation Based on Bayesian Sample Queries

by   Firat Ozdemir, et al.

Segmentation of anatomical structures is a fundamental image analysis task for many applications in the medical field. Deep learning methods have been shown to perform well, but for this purpose large numbers of manual annotations are needed in the first place, which necessitate prohibitive levels of resources that are often unavailable. In an active learning framework of selecting informed samples for manual labeling, expert clinician time for manual annotation can be optimally utilized, enabling the establishment of large labeled datasets for machine learning. In this paper, we propose a novel method that combines representativeness with uncertainty in order to estimate ideal samples to be annotated, iteratively from a given dataset. Our novel representativeness metric is based on Bayesian sampling, by using information-maximizing autoencoders. We conduct experiments on a shoulder magnetic resonance imaging (MRI) dataset for the segmentation of four musculoskeletal tissue classes. Quantitative results show that the annotation of representative samples selected by our proposed querying method yields an improved segmentation performance at each active learning iteration, compared to a baseline method that also employs uncertainty and representativeness metrics. For instance, with only 10 reaches within 5 the dataset given as annotated (an impractical scenario due to resource constraints), and this gap drops down to a mere 2 the dataset samples are annotated. Such active learning approach to selecting samples to annotate enables an optimal use of the expert clinician time, being often the bottleneck in realizing machine learning solutions in medicine.


Confident Coreset for Active Learning in Medical Image Analysis

Recent advances in deep learning have resulted in great successes in var...

Active Learning for Segmentation by Optimizing Content Information for Maximal Entropy

Segmentation is essential for medical image analysis tasks such as inter...

Hybrid Representation-Enhanced Sampling for Bayesian Active Learning in Musculoskeletal Segmentation of Lower Extremities

Purpose: Obtaining manual annotations to train deep learning (DL) models...

M-VAAL: Multimodal Variational Adversarial Active Learning for Downstream Medical Image Analysis Tasks

Acquiring properly annotated data is expensive in the medical field as i...

Deeply Supervised Active Learning for Finger Bones Segmentation

Segmentation is a prerequisite yet challenging task for medical image an...

Task-Aware Active Learning for Endoscopic Image Analysis

Semantic segmentation of polyps and depth estimation are two important r...

CS4ML: A general framework for active learning with arbitrary data based on Christoffel functions

We introduce a general framework for active learning in regression probl...

Please sign up or login with your details

Forgot password? Click here to reset