Diversity-Promoting Ensemble for Medical Image Segmentation
- URL: http://arxiv.org/abs/2210.12388v1
- Date: Sat, 22 Oct 2022 08:47:25 GMT
- Title: Diversity-Promoting Ensemble for Medical Image Segmentation
- Authors: Mariana-Iuliana Georgescu, Radu Tudor Ionescu, Andreea-Iuliana Miron
- Abstract summary: We propose a strategy to generate ensembles of different architectures for medical image segmentation.
To promote diversity, we select models with low Dice scores among each other.
Our empirical results show that DiPE surpasses both individual models as well as the ensemble creation strategy based on selecting the top scoring models.
- Score: 25.089517950882527
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Medical image segmentation is an actively studied task in medical imaging,
where the precision of the annotations is of utter importance towards accurate
diagnosis and treatment. In recent years, the task has been approached with
various deep learning systems, among the most popular models being U-Net. In
this work, we propose a novel strategy to generate ensembles of different
architectures for medical image segmentation, by leveraging the diversity
(decorrelation) of the models forming the ensemble. More specifically, we
utilize the Dice score among model pairs to estimate the correlation between
the outputs of the two models forming each pair. To promote diversity, we
select models with low Dice scores among each other. We carry out
gastro-intestinal tract image segmentation experiments to compare our
diversity-promoting ensemble (DiPE) with another strategy to create ensembles
based on selecting the top scoring U-Net models. Our empirical results show
that DiPE surpasses both individual models as well as the ensemble creation
strategy based on selecting the top scoring models.
Related papers
- Explore In-Context Segmentation via Latent Diffusion Models [132.26274147026854]
latent diffusion model (LDM) is an effective minimalist for in-context segmentation.
We build a new and fair in-context segmentation benchmark that includes both image and video datasets.
arXiv Detail & Related papers (2024-03-14T17:52:31Z) - Diagonal Hierarchical Consistency Learning for Semi-supervised Medical Image Segmentation [0.0]
We propose a novel framework for robust semi-supervised medical image segmentation using diagonal hierarchical consistency learning (DiHC-Net)
It is composed of multiple sub-models with identical multi-scale architecture but with distinct sub-layers, such as up-sampling and normalisation layers.
A series of experiments verifies the efficacy of our simple framework, outperforming all previous approaches on public benchmark dataset covering organ and tumour.
arXiv Detail & Related papers (2023-11-10T12:38:16Z) - Evaluation of Multi-indicator And Multi-organ Medical Image Segmentation
Models [4.302265156822829]
"U-shaped" neural networks with encoder and decoder structures have gained popularity in the field of medical image segmentation.
We propose a comprehensive method for evaluating medical image segmentation models for multi-indicator and multi-organ.
MIMO offers novel insights into multi-indicator and multi-organ medical image evaluation.
arXiv Detail & Related papers (2023-06-01T08:35:51Z) - Ambiguous Medical Image Segmentation using Diffusion Models [60.378180265885945]
We introduce a single diffusion model-based approach that produces multiple plausible outputs by learning a distribution over group insights.
Our proposed model generates a distribution of segmentation masks by leveraging the inherent sampling process of diffusion.
Comprehensive results show that our proposed approach outperforms existing state-of-the-art ambiguous segmentation networks.
arXiv Detail & Related papers (2023-04-10T17:58:22Z) - MultiTalent: A Multi-Dataset Approach to Medical Image Segmentation [1.146419670457951]
Current practices limit model training and supervised pre-training to one or a few similar datasets.
We propose MultiTalent, a method that leverages multiple CT datasets with diverse and conflicting class definitions.
arXiv Detail & Related papers (2023-03-25T11:37:16Z) - MedSegDiff-V2: Diffusion based Medical Image Segmentation with
Transformer [53.575573940055335]
We propose a novel Transformer-based Diffusion framework, called MedSegDiff-V2.
We verify its effectiveness on 20 medical image segmentation tasks with different image modalities.
arXiv Detail & Related papers (2023-01-19T03:42:36Z) - Understanding the Tricks of Deep Learning in Medical Image Segmentation:
Challenges and Future Directions [66.40971096248946]
In this paper, we collect a series of MedISeg tricks for different model implementation phases.
We experimentally explore the effectiveness of these tricks on consistent baselines.
We also open-sourced a strong MedISeg repository, where each component has the advantage of plug-and-play.
arXiv Detail & Related papers (2022-09-21T12:30:05Z) - Modality Completion via Gaussian Process Prior Variational Autoencoders
for Multi-Modal Glioma Segmentation [75.58395328700821]
We propose a novel model, Multi-modal Gaussian Process Prior Variational Autoencoder (MGP-VAE), to impute one or more missing sub-modalities for a patient scan.
MGP-VAE can leverage the Gaussian Process (GP) prior on the Variational Autoencoder (VAE) to utilize the subjects/patients and sub-modalities correlations.
We show the applicability of MGP-VAE on brain tumor segmentation where either, two, or three of four sub-modalities may be missing.
arXiv Detail & Related papers (2021-07-07T19:06:34Z) - Two layer Ensemble of Deep Learning Models for Medical Image
Segmentation [0.2699900017799093]
We propose a two-layer ensemble of deep learning models for the segmentation of medical images.
The prediction for each training image pixel made by each model in the first layer is used as the augmented data of the training image.
The prediction of the second layer is then combined by using a weights-based scheme in which each model contributes differently to the combined result.
arXiv Detail & Related papers (2021-04-10T16:52:34Z) - Adversarial Sample Enhanced Domain Adaptation: A Case Study on
Predictive Modeling with Electronic Health Records [57.75125067744978]
We propose a data augmentation method to facilitate domain adaptation.
adversarially generated samples are used during domain adaptation.
Results confirm the effectiveness of our method and the generality on different tasks.
arXiv Detail & Related papers (2021-01-13T03:20:20Z)
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.