Pathology Synthesis of 3D-Consistent Cardiac MR Images using 2D VAEs and GANs

TL;DR

This paper introduces a novel method combining VAEs and GANs to synthesize 3D-consistent cardiac MR images with diverse pathologies, enhancing data augmentation for deep learning applications.

Contribution

It presents a new approach for pathology synthesis using latent space interpolation in VAEs and label-conditional GANs, enabling realistic and diverse cardiac MR image generation.

Findings

Generated images improve segmentation robustness across multiple vendors.

Latent space manipulation enables realistic interpolation of cardiac images.

3D consistency achieved through modeling slice relationships.

Abstract

We propose a method for synthesizing cardiac magnetic resonance (MR) images with plausible heart pathologies and realistic appearances for the purpose of generating labeled data for the application of supervised deep-learning (DL) training. The image synthesis consists of label deformation and label-to-image translation tasks. The former is achieved via latent space interpolation in a VAE model, while the latter is accomplished via a label-conditional GAN model. We devise three approaches for label manipulation in the latent space of the trained VAE model; i) \textbf{intra-subject synthesis} aiming to interpolate the intermediate slices of a subject to increase the through-plane resolution, ii) \textbf{inter-subject synthesis} aiming to interpolate the geometry and appearance of intermediate images between two dissimilar subjects acquired with different scanner vendors, and iii) \textbf{pathology synthesis} aiming to synthesize a series of pseudo-pathological synthetic subjects with characteristics of a desired heart disease. Furthermore, we propose to model the relationship between 2D slices in the latent space of the VAE prior to reconstruction for generating 3D-consistent subjects from stacking up 2D slice-by-slice generations. We demonstrate that such an approach could provide a solution to diversify and enrich an available database of cardiac MR images and to pave the way for the development of generalizable DL-based image analysis algorithms. We quantitatively evaluate the quality of the synthesized data in an augmentation scenario to achieve generalization and robustness to multi-vendor and multi-disease data for image segmentation. Our code is available at https://github.com/sinaamirrajab/CardiacPathologySynthesis.