Diminishing Uncertainty within the Training Pool: Active Learning for Medical Image Segmentation

01/07/2021
by   Vishwesh Nath, et al.
25

Active learning is a unique abstraction of machine learning techniques where the model/algorithm could guide users for annotation of a set of data points that would be beneficial to the model, unlike passive machine learning. The primary advantage being that active learning frameworks select data points that can accelerate the learning process of a model and can reduce the amount of data needed to achieve full accuracy as compared to a model trained on a randomly acquired data set. Multiple frameworks for active learning combined with deep learning have been proposed, and the majority of them are dedicated to classification tasks. Herein, we explore active learning for the task of segmentation of medical imaging data sets. We investigate our proposed framework using two datasets: 1.) MRI scans of the hippocampus, 2.) CT scans of pancreas and tumors. This work presents a query-by-committee approach for active learning where a joint optimizer is used for the committee. At the same time, we propose three new strategies for active learning: 1.) increasing frequency of uncertain data to bias the training data set; 2.) Using mutual information among the input images as a regularizer for acquisition to ensure diversity in the training dataset; 3.) adaptation of Dice log-likelihood for Stein variational gradient descent (SVGD). The results indicate an improvement in terms of data reduction by achieving full accuracy while only using 22.69 and 48.85

READ FULL TEXT

page 13

page 14

page 15

page 17

research
10/04/2022

CLINICAL: Targeted Active Learning for Imbalanced Medical Image Classification

Training deep learning models on medical datasets that perform well for ...
research
06/20/2022

Deep reinforced active learning for multi-class image classification

High accuracy medical image classification can be limited by the costs o...
research
10/04/2022

DIAGNOSE: Avoiding Out-of-distribution Data using Submodular Information Measures

Avoiding out-of-distribution (OOD) data is critical for training supervi...
research
11/08/2019

Char-RNN and Active Learning for Hashtag Segmentation

We explore the abilities of character recurrent neural network (char-RNN...
research
04/05/2022

An Exploration of Active Learning for Affective Digital Phenotyping

Some of the most severe bottlenecks preventing widespread development of...
research
07/31/2023

Active Learning in Genetic Programming: Guiding Efficient Data Collection for Symbolic Regression

This paper examines various methods of computing uncertainty and diversi...
research
07/18/2021

Fully Automated Machine Learning Pipeline for Echocardiogram Segmentation

Nowadays, cardiac diagnosis largely depends on left ventricular function...

Please sign up or login with your details

Forgot password? Click here to reset