en
Scientific article
Open access
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.

eng
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)
Identifiers
ISSN of the journal1869-4101
26views
4downloads

Technical informations

Creation08/25/2023 8:35:47 AM
First validation11/13/2023 6:17:21 PM
Update time11/13/2023 6:17:21 PM
Status update11/13/2023 6:17:21 PM
Last indexation05/06/2024 5:20:20 PM
All rights reserved by Archive ouverte UNIGE and the University of GenevaunigeBlack