Deep ensemble model for segmenting microscopy images in the presence of limited labeled dataDownload PDF

06 Apr 2021, 11:01 (modified: 22 Jul 2022, 20:09)MIDL Short 2021 PosterReaders: Everyone
Keywords: ensemble, segmentation, microscopy
TL;DR: We propose deep ensemble model to perform microscopy image segmentation with limited labeled data.
Abstract: Obtaining large amounts of high quality labeled microscopy data is expensive and time-consuming. To overcome this issue, we propose a deep ensemble model which aims to utilise limited labeled training data. We train multiple identical Convolutional Neural Network (CNN) segmentation models on training data that is partitioned into folds in two steps. First, the data is split based on sample diversity or expert knowledge reflecting the possible {\em modes} of the underlying data distribution. In the second step, these partitions are split into random folds like in a cross-validation setting. Segmentation models based on the U-net architecture are trained on each of these resulting folds yielding the candidate models for our deep ensemble model which are aggregated to obtain the final prediction. The proposed deep ensemble model is compared to relevant baselines, in their ability to segment interneurons in microscopic images of mice spinal cord, showing improved performance on an independent test set.
Paper Type: methodological development
Primary Subject Area: Learning with Noisy Labels and Limited Data
Secondary Subject Area: Segmentation
Paper Status: original work, not submitted yet
Source Code Url: https://github.com/JanKaminski95/EnsembleNet.git
Data Set Url: Data set is internal and can be accessed upon request.
Registration: I acknowledge that publication of this at MIDL and in the proceedings requires at least one of the authors to register and present the work during the conference.
Authorship: I confirm that I am the author of this work and that it has not been submitted to another publication before.
4 Replies

Loading