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

Apr 06, 2021 (edited Apr 20, 2021)MIDL 2021 Conference Short SubmissionReaders: 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