Interpretability-guided Data Augmentation for Robust Segmentation in
Multi-centre Colonoscopy Data
- URL: http://arxiv.org/abs/2308.15881v1
- Date: Wed, 30 Aug 2023 09:03:28 GMT
- Title: Interpretability-guided Data Augmentation for Robust Segmentation in
Multi-centre Colonoscopy Data
- Authors: Valentina Corbetta, Regina Beets-Tan, and Wilson Silva
- Abstract summary: We introduce an innovative data augmentation approach centred on interpretability saliency maps.
The proposed augmentation technique demonstrates increased robustness across different segmentation models and domains.
- Score: 0.4915744683251151
- License: http://creativecommons.org/licenses/by-nc-sa/4.0/
- Abstract: Multi-centre colonoscopy images from various medical centres exhibit distinct
complicating factors and overlays that impact the image content, contingent on
the specific acquisition centre. Existing Deep Segmentation networks struggle
to achieve adequate generalizability in such data sets, and the currently
available data augmentation methods do not effectively address these sources of
data variability. As a solution, we introduce an innovative data augmentation
approach centred on interpretability saliency maps, aimed at enhancing the
generalizability of Deep Learning models within the realm of multi-centre
colonoscopy image segmentation. The proposed augmentation technique
demonstrates increased robustness across different segmentation models and
domains. Thorough testing on a publicly available multi-centre dataset for
polyp detection demonstrates the effectiveness and versatility of our approach,
which is observed both in quantitative and qualitative results. The code is
publicly available at:
https://github.com/nki-radiology/interpretability_augmentation
Related papers
- Cross-Dataset Generalization For Retinal Lesions Segmentation [2.1160877779256126]
This study characterizes several known datasets and compares different techniques that have been proposed to enhance the generalisation performance of a model.
Our results provide insights into how to combine coarsely labelled data with a finely-grained dataset in order to improve the lesions segmentation.
arXiv Detail & Related papers (2024-05-14T05:52:01Z) - Harmonized Spatial and Spectral Learning for Robust and Generalized
Medical Image Segmentation [5.678550062099796]
We introduce the innovative Spectral Correlation Coefficient objective to improve the model's capacity to capture middle-order features and contextual long-range dependencies.
Experiments reveal that optimizing this objective with existing architectures like UNet and TransUNet significantly enhances generalization, interpretability, and noise robustness.
arXiv Detail & Related papers (2024-01-18T20:43:43Z) - Two Approaches to Supervised Image Segmentation [55.616364225463066]
The present work develops comparison experiments between deep learning and multiset neurons approaches.
The deep learning approach confirmed its potential for performing image segmentation.
The alternative multiset methodology allowed for enhanced accuracy while requiring little computational resources.
arXiv Detail & Related papers (2023-07-19T16:42:52Z) - Domain Generalization for Mammographic Image Analysis with Contrastive
Learning [62.25104935889111]
The training of an efficacious deep learning model requires large data with diverse styles and qualities.
A novel contrastive learning is developed to equip the deep learning models with better style generalization capability.
The proposed method has been evaluated extensively and rigorously with mammograms from various vendor style domains and several public datasets.
arXiv Detail & Related papers (2023-04-20T11:40:21Z) - 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) - Domain Generalization with Adversarial Intensity Attack for Medical
Image Segmentation [27.49427483473792]
In real-world scenarios, it is common for models to encounter data from new and different domains to which they were not exposed to during training.
domain generalization (DG) is a promising direction as it enables models to handle data from previously unseen domains.
We introduce a novel DG method called Adversarial Intensity Attack (AdverIN), which leverages adversarial training to generate training data with an infinite number of styles.
arXiv Detail & Related papers (2023-04-05T19:40:51Z) - AADG: Automatic Augmentation for Domain Generalization on Retinal Image
Segmentation [1.0452185327816181]
We propose a data manipulation based domain generalization method, called Automated Augmentation for Domain Generalization (AADG)
Our AADG framework can effectively sample data augmentation policies that generate novel domains.
Our proposed AADG exhibits state-of-the-art generalization performance and outperforms existing approaches.
arXiv Detail & Related papers (2022-07-27T02:26:01Z) - Few-shot Medical Image Segmentation using a Global Correlation Network
with Discriminative Embedding [60.89561661441736]
We propose a novel method for few-shot medical image segmentation.
We construct our few-shot image segmentor using a deep convolutional network trained episodically.
We enhance discriminability of deep embedding to encourage clustering of the feature domains of the same class.
arXiv Detail & Related papers (2020-12-10T04:01:07Z) - Disentangled Representations for Domain-generalized Cardiac Segmentation [19.108784219423377]
"Resolution Augmentation" method generates more diverse data by rescaling images to different resolutions within a range spanning different scanner protocols.
"Factor-based Augmentation" method generates more diverse data by projecting the original samples onto disentangled latent spaces.
Our experiments demonstrate the importance of efficient adaptation between seen and unseen domains, as well as model generalization ability.
arXiv Detail & Related papers (2020-08-26T12:20:09Z) - Pathological Retinal Region Segmentation From OCT Images Using Geometric
Relation Based Augmentation [84.7571086566595]
We propose improvements over previous GAN-based medical image synthesis methods by jointly encoding the intrinsic relationship of geometry and shape.
The proposed method outperforms state-of-the-art segmentation methods on the public RETOUCH dataset having images captured from different acquisition procedures.
arXiv Detail & Related papers (2020-03-31T11:50:43Z) - MS-Net: Multi-Site Network for Improving Prostate Segmentation with
Heterogeneous MRI Data [75.73881040581767]
We propose a novel multi-site network (MS-Net) for improving prostate segmentation by learning robust representations.
Our MS-Net improves the performance across all datasets consistently, and outperforms state-of-the-art methods for multi-site learning.
arXiv Detail & Related papers (2020-02-09T14:11:50Z)
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.