Domain-Adversarial Learning for Multi-Centre, Multi-Vendor, and
Multi-Disease Cardiac MR Image Segmentation
- URL: http://arxiv.org/abs/2008.11776v1
- Date: Wed, 26 Aug 2020 19:40:55 GMT
- Title: Domain-Adversarial Learning for Multi-Centre, Multi-Vendor, and
Multi-Disease Cardiac MR Image Segmentation
- Authors: Cian M. Scannell and Amedeo Chiribiri and Mitko Veta
- Abstract summary: Domain-adversarial learning is used to train a domain-invariant 2D U-Net using labelled and unlabelled data.
This approach is evaluated on both seen and unseen domains from the M&Ms challenge dataset.
- Score: 3.4551186283197883
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Cine cardiac magnetic resonance (CMR) has become the gold standard for the
non-invasive evaluation of cardiac function. In particular, it allows the
accurate quantification of functional parameters including the chamber volumes
and ejection fraction. Deep learning has shown the potential to automate the
requisite cardiac structure segmentation. However, the lack of robustness of
deep learning models has hindered their widespread clinical adoption. Due to
differences in the data characteristics, neural networks trained on data from a
specific scanner are not guaranteed to generalise well to data acquired at a
different centre or with a different scanner. In this work, we propose a
principled solution to the problem of this domain shift. Domain-adversarial
learning is used to train a domain-invariant 2D U-Net using labelled and
unlabelled data. This approach is evaluated on both seen and unseen domains
from the M\&Ms challenge dataset and the domain-adversarial approach shows
improved performance as compared to standard training. Additionally, we show
that the domain information cannot be recovered from the learned features.
Related papers
- Curriculum-Based Augmented Fourier Domain Adaptation for Robust Medical
Image Segmentation [18.830738606514736]
This work proposes the Curriculum-based Augmented Fourier Domain Adaptation (Curri-AFDA) for robust medical image segmentation.
In particular, our curriculum learning strategy is based on the causal relationship of a model under different levels of data shift.
Experiments on two segmentation tasks of Retina and Nuclei collected from multiple sites and scanners suggest that our proposed method yields superior adaptation and generalization performance.
arXiv Detail & Related papers (2023-06-06T08:56:58Z) - DA-VSR: Domain Adaptable Volumetric Super-Resolution For Medical Images [69.63915773870758]
We present a novel algorithm called domain adaptable super-resolution (DA-VSR) to better bridge the domain inconsistency gap.
DA-VSR uses a unified feature extraction backbone and a series of network heads to improve image quality over different planes.
We demonstrate that DA-VSR significantly improves super-resolution quality across numerous datasets of different domains.
arXiv Detail & Related papers (2022-10-11T03:16:35Z) - Preservation of High Frequency Content for Deep Learning-Based Medical
Image Classification [74.84221280249876]
An efficient analysis of large amounts of chest radiographs can aid physicians and radiologists.
We propose a novel Discrete Wavelet Transform (DWT)-based method for the efficient identification and encoding of visual information.
arXiv Detail & Related papers (2022-05-08T15:29:54Z) - Cross-Site Severity Assessment of COVID-19 from CT Images via Domain
Adaptation [64.59521853145368]
Early and accurate severity assessment of Coronavirus disease 2019 (COVID-19) based on computed tomography (CT) images offers a great help to the estimation of intensive care unit event.
To augment the labeled data and improve the generalization ability of the classification model, it is necessary to aggregate data from multiple sites.
This task faces several challenges including class imbalance between mild and severe infections, domain distribution discrepancy between sites, and presence of heterogeneous features.
arXiv Detail & Related papers (2021-09-08T07:56:51Z) - Segmentation with Multiple Acceptable Annotations: A Case Study of
Myocardial Segmentation in Contrast Echocardiography [12.594060034146125]
We propose a new extended Dice to evaluate segmentation performance when multiple accepted ground truth is available.
We then solve the second problem by further incorporating the new metric into a loss function that enables neural networks to learn general features of myocardium.
Experiment results on our clinical MCE data set demonstrate that the neural network trained with the proposed loss function outperforms those existing ones.
arXiv Detail & Related papers (2021-06-29T17:32:24Z) - Embracing the Disharmony in Heterogeneous Medical Data [12.739380441313022]
Heterogeneity in medical imaging data is often tackled, in the context of machine learning, using domain invariance.
This paper instead embraces the heterogeneity and treats it as a multi-task learning problem.
We show that this approach improves classification accuracy by 5-30 % across different datasets on the main classification tasks.
arXiv Detail & Related papers (2021-03-23T21:36:39Z) - An Uncertainty-Driven GCN Refinement Strategy for Organ Segmentation [53.425900196763756]
We propose a segmentation refinement method based on uncertainty analysis and graph convolutional networks.
We employ the uncertainty levels of the convolutional network in a particular input volume to formulate a semi-supervised graph learning problem.
We show that our method outperforms the state-of-the-art CRF refinement method by improving the dice score by 1% for the pancreas and 2% for spleen.
arXiv Detail & Related papers (2020-12-06T18:55:07Z) - Studying Robustness of Semantic Segmentation under Domain Shift in
cardiac MRI [0.8858288982748155]
We study challenges and opportunities of domain transfer across images from multiple clinical centres and scanner vendors.
In this work, we build upon a fixed U-Net architecture configured by the nnU-net framework to investigate various data augmentation techniques and batch normalization layers.
arXiv Detail & Related papers (2020-11-15T17:50:23Z) - Fader Networks for domain adaptation on fMRI: ABIDE-II study [68.5481471934606]
We use 3D convolutional autoencoders to build the domain irrelevant latent space image representation and demonstrate this method to outperform existing approaches on ABIDE data.
arXiv Detail & Related papers (2020-10-14T16:50:50Z) - Unsupervised Domain Adaptation via CycleGAN for White Matter
Hyperintensity Segmentation in Multicenter MR Images [2.627822659948232]
Quantification of white matter hyperintensities in magnetic resonance images serves as a predictor for risk of stroke, dementia and mortality.
During the last years, convolutional neural networks (CNN) specifically tailored for biomedical image segmentation have outperformed all previous techniques in this task.
In this work, we explore the use of cycle-consistent adversarial networks (CycleGAN) to perform unsupervised domain adaptation on multicenter MR images with brain lesions.
arXiv Detail & Related papers (2020-09-10T16:48:19Z) - 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.