Less Is More: A Comparison of Active Learning Strategies for 3D Medical Image Segmentation

Josafat-Mattias Burmeister, Marcel Fernandez Rosas, Johannes Hagemann, Jonas Kordt, Jasper Blum +3 more
7/2/2022
cs.CVcs.AIcs.LG

No official code found. Best implementation: HealthML/active-segmentation (20 stars, AGPL-3.0, PyTorch, last commit 2022-07-05).

Abstract

Since labeling medical image data is a costly and labor-intensive process, active learning has gained much popularity in the medical image segmentation domain in recent years. A variety of active learning strategies have been proposed in the literature, but their effectiveness is highly dependent on the dataset and training scenario. To facilitate the comparison of existing strategies and provide a baseline for evaluating novel strategies, we evaluate the performance of several well-known active learning strategies on three datasets from the Medical Segmentation Decathlon. Additionally, we consider a strided sampling strategy specifically tailored to 3D image data. We demonstrate that both random and strided sampling act as strong baselines and discuss the advantages and disadvantages of the studied methods. To allow other researchers to compare their work to our results, we provide an open-source framework for benchmarking active learning strategies on a variety of medical segmentation datasets.

View on arXivView PDF

Code Implementations(1)

PyTorchAGPL-3.0

ActiveSegmentation: A Simulation Framework for Benchmarking Active Learning Strategies for 3D Medical Image Segmentation

201Python, Jupyter NotebookOct 30, 20214 years agoAGPL-3.0
3d-image-segmentationactive-learningdeep-learningmedical-image-segmentationmedical-imaging+3 more

Similar Papers

Cite this paper

@article{burmeister2022less,
  title  = {Less Is More: A Comparison of Active Learning Strategies for 3D Medical Image Segmentation},
  author = {Josafat-Mattias Burmeister and Marcel Fernandez Rosas and Johannes Hagemann and Jonas Kordt and Jasper Blum and Simon Shabo and Benjamin Bergner and Christoph Lippert},
  year   = {2022},
  eprint = {2207.00845},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2207.00845v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2207.00845.svg)](https://codeofpaper.com/paper/2207.00845)

Discussion