HAL will be down for maintenance from Friday, June 10 at 4pm through Monday, June 13 at 9am. More information
Skip to Main content Skip to Navigation
Conference papers

Style Data Augmentation for Robust Segmentation of Multi-Modality Cardiac MRI

Abstract : We propose a data augmentation method to improve thesegmentation accuracy of the convolutional neural network on multi-modality cardiac magnetic resonance (CMR) dataset. The strategy aims to reduce over-fitting of the network toward any specific intensity or contrast of the training images by introducing diversity in these two aspects. The style data augmentation (SDA) strategy increases the size of the training dataset by using multiple image processing functions including adaptive histogram equalisation, Laplacian transformation, Sobel edge detection, intensity inversion and histogram matching. For the segmentation task, we developed the thresholded connection layer network (TCL-Net), a minimalist rendition of the U-Net architecture, which is designed to reduce convergence and computation time. We integrate the dual U-Net strategy to increase the resolution of the 3D segmentation target. Utilising these approaches on a multi-modality dataset, with SSFP and T2 weighted images as training and LGE as validation, we achieve 90% and 96% validation Dice coefficient for endocardium and epicardium segmentations. This result can be interpreted as a proof of concept for a generalised segmentation network that is robust to the quality or modality of the input images. When testing with our mono-centric LGE image dataset, the SDA method also improves the performance of the epicardium segmentation, with an increase from 87% to 90% for the single network segmentation.
Document type :
Conference papers
Complete list of metadata

https://hal.inria.fr/hal-02401643
Contributor : Buntheng Ly Connect in order to contact the contributor
Submitted on : Thursday, May 13, 2021 - 6:24:52 PM
Last modification on : Friday, February 4, 2022 - 3:24:03 AM

File

stacom_ms_seg_hal.pdf
Files produced by the author(s)

Identifiers

  • HAL Id : hal-02401643, version 3

Citation

Buntheng Ly, Hubert Cochet, Maxime Sermesant. Style Data Augmentation for Robust Segmentation of Multi-Modality Cardiac MRI. STACOM 2019 - 10th Workhop on Statistical Atlases and Computational Modelling of the Heart, Oct 2019, Shenzhen, China. ⟨hal-02401643v3⟩

Share

Metrics

Record views

263

Files downloads

400