Scientific article
OA Policy
English

An active learning approach to train a deep learning algorithm for tumor segmentation from brain MR images

Published inInsights into imaging, vol. 14, no. 1, 141
Publication date2023-08-25
First online date2023-08-25
Abstract

Purpose: This study focuses on assessing the performance of active learning techniques to train a brain MRI glioma segmentation model.

Methods: The publicly available training dataset provided for the 2021 RSNA-ASNR-MICCAI Brain Tumor Segmentation (BraTS) Challenge was used in this study, consisting of 1251 multi-institutional, multi-parametric MR images. Post-contrast T1, T2, and T2 FLAIR images as well as ground truth manual segmentation were used as input for the model. The data were split into a training set of 1151 cases and testing set of 100 cases, with the testing set remaining constant throughout. Deep convolutional neural network segmentation models were trained using the NiftyNet platform. To test the viability of active learning in training a segmentation model, an initial reference model was trained using all 1151 training cases followed by two additional models using only 575 cases and 100 cases. The resulting predicted segmentations of these two additional models on the remaining training cases were then addended to the training dataset for additional training.

Results: It was demonstrated that an active learning approach for manual segmentation can lead to comparable model performance for segmentation of brain gliomas (0.906 reference Dice score vs 0.868 active learning Dice score) while only requiring manual annotation for 28.6% of the data.

Conclusion: The active learning approach when applied to model training can drastically reduce the time and labor spent on preparation of ground truth training data.

Critical relevance statement: Active learning concepts were applied to a deep learning-assisted segmentation of brain gliomas from MR images to assess their viability in reducing the required amount of manually annotated ground truth data in model training.

Key points:

• This study focuses on assessing the performance of active learning techniques to train a brain MRI glioma segmentation model.

• The active learning approach for manual segmentation can lead to comparable model performance for segmentation of brain gliomas.

• Active learning when applied to model training can drastically reduce the time and labor spent on preparation of ground truth training data.

Keywords
  • Active learning
  • Deep learning
  • Gliomas
  • MRI
  • Segmentation
Citation (ISO format)
BÖHRINGER, Andrew Stephen et al. An active learning approach to train a deep learning algorithm for tumor segmentation from brain MR images. In: Insights into imaging, 2023, vol. 14, n° 1, p. 141. doi: 10.1186/s13244-023-01487-6
Main files (1)
Article (Published version)
Secondary files (1)
Appendix - Supplemental material, table and figures
accessLevelPublic
Identifiers
Journal ISSN1869-4101
104views
51downloads

Technical informations

Creation25/08/2023 08:35:47
First validation13/11/2023 18:17:21
Update24/03/2026 14:58:12
Status update24/03/2026 14:58:12
Last indexation24/03/2026 15:00:20
All rights reserved by Archive ouverte UNIGE and the University of GenevaunigeBlack