Ensemble uncertainty as a criterion for dataset expansion in distinct
bone segmentation from upper-body CT images
- URL: http://arxiv.org/abs/2208.09216v1
- Date: Fri, 19 Aug 2022 08:39:23 GMT
- Title: Ensemble uncertainty as a criterion for dataset expansion in distinct
bone segmentation from upper-body CT images
- Authors: Eva Schnider, Antal Huck, Mireille Toranelli, Georg Rauter, Azhar Zam,
Magdalena M\"uller-Gerbl, Philippe Cattin
- Abstract summary: The localisation and segmentation of individual bones is an important preprocessing step in many planning and navigation applications.
We present an end-to-end learnt algorithm that is capable of segmenting 125 distinct bones in an upper-body CT.
We also provide an ensemble-based uncertainty measure that helps to single out scans to enlarge the training dataset with.
- Score: 0.7388859384645263
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Purpose: The localisation and segmentation of individual bones is an
important preprocessing step in many planning and navigation applications. It
is, however, a time-consuming and repetitive task if done manually. This is
true not only for clinical practice but also for the acquisition of training
data. We therefore not only present an end-to-end learnt algorithm that is
capable of segmenting 125 distinct bones in an upper-body CT, but also provide
an ensemble-based uncertainty measure that helps to single out scans to enlarge
the training dataset with. Methods We create fully automated end-to-end learnt
segmentations using a neural network architecture inspired by the 3D-Unet and
fully supervised training. The results are improved using ensembles and
inference-time augmentation. We examine the relationship of
ensemble-uncertainty to an unlabelled scan's prospective usefulness as part of
the training dataset. Results: Our methods are evaluated on an in-house dataset
of 16 upper-body CT scans with a resolution of \SI{2}{\milli\meter} per
dimension. Taking into account all 125 bones in our label set, our most
successful ensemble achieves a median dice score coefficient of 0.83. We find a
lack of correlation between a scan's ensemble uncertainty and its prospective
influence on the accuracies achieved within an enlarged training set. At the
same time, we show that the ensemble uncertainty correlates to the number of
voxels that need manual correction after an initial automated segmentation,
thus minimising the time required to finalise a new ground truth segmentation.
Conclusion: In combination, scans with low ensemble uncertainty need less
annotator time while yielding similar future DSC improvements. They are thus
ideal candidates to enlarge a training set for upper-body distinct bone
segmentation from CT scans. }
Related papers
- Towards Unifying Anatomy Segmentation: Automated Generation of a
Full-body CT Dataset via Knowledge Aggregation and Anatomical Guidelines [113.08940153125616]
We generate a dataset of whole-body CT scans with $142$ voxel-level labels for 533 volumes providing comprehensive anatomical coverage.
Our proposed procedure does not rely on manual annotation during the label aggregation stage.
We release our trained unified anatomical segmentation model capable of predicting $142$ anatomical structures on CT data.
arXiv Detail & Related papers (2023-07-25T09:48:13Z) - Extraction of volumetric indices from echocardiography: which deep
learning solution for clinical use? [6.144041824426555]
We show that the proposed 3D nnU-Net outperforms alternative 2D and recurrent segmentation methods.
Overall, the experimental results suggest that with sufficient training data, 3D nnU-Net could become the first automated tool to meet the standards of an everyday clinical device.
arXiv Detail & Related papers (2023-05-03T09:38:52Z) - Prompt Tuning for Parameter-efficient Medical Image Segmentation [79.09285179181225]
We propose and investigate several contributions to achieve a parameter-efficient but effective adaptation for semantic segmentation on two medical imaging datasets.
We pre-train this architecture with a dedicated dense self-supervision scheme based on assignments to online generated prototypes.
We demonstrate that the resulting neural network model is able to attenuate the gap between fully fine-tuned and parameter-efficiently adapted models.
arXiv Detail & Related papers (2022-11-16T21:55:05Z) - PCA: Semi-supervised Segmentation with Patch Confidence Adversarial
Training [52.895952593202054]
We propose a new semi-supervised adversarial method called Patch Confidence Adrial Training (PCA) for medical image segmentation.
PCA learns the pixel structure and context information in each patch to get enough gradient feedback, which aids the discriminator in convergent to an optimal state.
Our method outperforms the state-of-the-art semi-supervised methods, which demonstrates its effectiveness for medical image segmentation.
arXiv Detail & Related papers (2022-07-24T07:45:47Z) - Diminishing Uncertainty within the Training Pool: Active Learning for
Medical Image Segmentation [6.3858225352615285]
We explore active learning for the task of segmentation of medical imaging data sets.
We propose three new strategies for active learning: increasing frequency of uncertain data to bias the training data set, using mutual information among the input images as a regularizer and adaptation of Dice log-likelihood for Stein variational gradient descent (SVGD)
The results indicate an improvement in terms of data reduction by achieving full accuracy while only using 22.69 % and 48.85 % of the available data for each dataset, respectively.
arXiv Detail & Related papers (2021-01-07T01:55:48Z) - Deep Learning to Segment Pelvic Bones: Large-scale CT Datasets and
Baseline Models [20.061463073787234]
We aim to bridge the data gap by curating a large pelvic CT dataset pooled from multiple sources and different manufacturers.
We propose for the first time, to the best of our knowledge, to learn a deep multi-class network for segmenting lumbar spine, sacrum, left hip, and right hip.
Finally, we introduce a post-processing tool based on the signed distance function (SDF) to eliminate false predictions.
arXiv Detail & Related papers (2020-12-16T03:30:40Z) - 3D Segmentation Networks for Excessive Numbers of Classes: Distinct Bone
Segmentation in Upper Bodies [1.2023648183416153]
This paper discusses the intricacies of training a 3D segmentation network in a many-label setting.
We show necessary modifications in network architecture, loss function, and data augmentation.
As a result, we demonstrate the robustness of our method by automatically segmenting over one hundred distinct bones simultaneously in an end-to-end learnt fashion from a CT-scan.
arXiv Detail & Related papers (2020-10-14T12:54:15Z) - MetricUNet: Synergistic Image- and Voxel-Level Learning for Precise CT
Prostate Segmentation via Online Sampling [66.01558025094333]
We propose a two-stage framework, with the first stage to quickly localize the prostate region and the second stage to precisely segment the prostate.
We introduce a novel online metric learning module through voxel-wise sampling in the multi-task network.
Our method can effectively learn more representative voxel-level features compared with the conventional learning methods with cross-entropy or Dice loss.
arXiv Detail & Related papers (2020-05-15T10:37:02Z) - 3D medical image segmentation with labeled and unlabeled data using
autoencoders at the example of liver segmentation in CT images [58.720142291102135]
This work investigates the potential of autoencoder-extracted features to improve segmentation with a convolutional neural network.
A convolutional autoencoder was used to extract features from unlabeled data and a multi-scale, fully convolutional CNN was used to perform the target task of 3D liver segmentation in CT images.
arXiv Detail & Related papers (2020-03-17T20:20:43Z) - VerSe: A Vertebrae Labelling and Segmentation Benchmark for
Multi-detector CT Images [121.31355003451152]
Large Scale Vertebrae Challenge (VerSe) was organised in conjunction with the International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI) in 2019 and 2020.
We present the the results of this evaluation and further investigate the performance-variation at vertebra-level, scan-level, and at different fields-of-view.
arXiv Detail & Related papers (2020-01-24T21:09:18Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.