Atzeni, A;
Peter, L;
Robinson, E;
Blackburn, E;
Althonayan, J;
Alexander, DC;
Iglesias, JE;
(2022)
Deep active learning for suggestive segmentation of biomedical image stacks via optimisation of Dice scores and traced boundary length.
Medical Image Analysis
, 81
, Article 102549. 10.1016/j.media.2022.102549.
Preview |
Text
1-s2.0-S1361841522001967-main.pdf - Published Version Download (12MB) | Preview |
Abstract
Manual segmentation of stacks of 2D biomedical images (e.g., histology) is a time-consuming task which can be sped up with semi-automated techniques. In this article, we present a suggestive deep active learning framework that seeks to minimise the annotation effort required to achieve a certain level of accuracy when labelling such a stack. The framework suggests, at every iteration, a specific region of interest (ROI) in one of the images for manual delineation. Using a deep segmentation neural network and a mixed cross-entropy loss function, we propose a principled strategy to estimate class probabilities for the whole stack, conditioned on heterogeneous partial segmentations of the 2D images, as well as on weak supervision in the form of image indices that bound each ROI. Using the estimated probabilities, we propose a novel active learning criterion based on predictions for the estimated segmentation performance and delineation effort, measured with average Dice scores and total delineated boundary length, respectively, rather than common surrogates such as entropy. The query strategy suggests the ROI that is expected to maximise the ratio between performance and effort, while considering the adjacency of structures that may have already been labelled – which decrease the length of the boundary to trace. We provide quantitative results on synthetically deformed MRI scans and real histological data, showing that our framework can reduce labelling effort by up to 60–70% without compromising accuracy.
Type: | Article |
---|---|
Title: | Deep active learning for suggestive segmentation of biomedical image stacks via optimisation of Dice scores and traced boundary length |
Location: | Netherlands |
Open access status: | An open access version is available from UCL Discovery |
DOI: | 10.1016/j.media.2022.102549 |
Publisher version: | https://doi.org/10.1016/j.media.2022.102549 |
Language: | English |
Additional information: | © 2022 The Authors. Published by Elsevier B.V. This is an open access article under the CC BY license (http://creativecommons.org/licenses/by/4.0/). |
Keywords: | Active learning, Deep learning, Histology, Partial annotation, Segmentation |
UCL classification: | UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science > Dept of Computer Science UCL > Provost and Vice Provost Offices > UCL BEAMS UCL UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science > Dept of Med Phys and Biomedical Eng |
URI: | https://discovery.ucl.ac.uk/id/eprint/10156388 |
Archive Staff Only
![]() |
View Item |