Deep reinforced active learning for multi-class image classification

06/20/2022
by   Emma Slade, et al.
0

High accuracy medical image classification can be limited by the costs of acquiring more data as well as the time and expertise needed to label existing images. In this paper, we apply active learning to medical image classification, a method which aims to maximise model performance on a minimal subset from a larger pool of data. We present a new active learning framework, based on deep reinforcement learning, to learn an active learning query strategy to label images based on predictions from a convolutional neural network. Our framework modifies the deep-Q network formulation, allowing us to pick data based additionally on geometric arguments in the latent space of the classifier, allowing for high accuracy multi-class classification in a batch-based active learning setting, enabling the agent to label datapoints that are both diverse and about which it is most uncertain. We apply our framework to two medical imaging datasets and compare with standard query strategies as well as the most recent reinforcement learning based active learning approach for image classification.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
08/01/2022

Information Gain Sampling for Active Learning in Medical Image Classification

Large, annotated datasets are not widely available in medical image anal...
research
08/12/2021

Reinforcement Learning Approach to Active Learning for Image Classification

Machine Learning requires large amounts of labeled data to fit a model. ...
research
01/07/2021

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

Active learning is a unique abstraction of machine learning techniques w...
research
12/03/2022

Active learning using adaptable task-based prioritisation

Supervised machine learning-based medical image computing applications n...
research
03/26/2021

MedSelect: Selective Labeling for Medical Image Classification Combining Meta-Learning with Deep Reinforcement Learning

We propose a selective learning method using meta-learning and deep rein...
research
08/04/2022

Deep Surrogate of Modular Multi Pump using Active Learning

Due to the high cost and reliability of sensors, the designers of a pump...
research
01/13/2017

Cost-Effective Active Learning for Deep Image Classification

Recent successes in learning-based image classification, however, heavil...

Please sign up or login with your details

Forgot password? Click here to reset