# Conditional Generation of Medical Images via Disentangled Adversarial Inference

arXiv: https://arxiv.org/abs/2012.04764 · HTML: https://mlsanity.com/paper/2012.04764/

## Key facts

- Cited by 13 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 8 Dec 2020.

## Built on by (most cited first, 13 of 13)

- [Generative models improve fairness of medical classifiers under distribution shifts](https://mlsanity.com/paper/2304.09218/) (2023)
- [Learning Disentangled Representations in the Imaging Domain](https://mlsanity.com/paper/2108.12043/) (2021)
- [Controllable cardiac synthesis via disentangled anatomy arithmetic](https://mlsanity.com/paper/2107.01748/) (2021)
- [Review of Disentanglement Approaches for Medical Applications -- Towards Solving the Gordian Knot of Generative Models in Healthcare](https://mlsanity.com/paper/2203.11132/) (2022)
- [Disentangling representations of retinal images with generative models](https://mlsanity.com/paper/2402.19186/) (2024)
- [MouseGAN++: Unsupervised Disentanglement and Contrastive Representation for Multiple MRI Modalities Synthesis and Structural Segmentation of Mouse Brain](https://mlsanity.com/paper/2212.01825/) (2022)
- [Constrained Adversarial Learning for Automated Software Testing: a literature review](https://mlsanity.com/paper/2303.07546/) (2023)
- [Decomposing Normal and Abnormal Features of Medical Images into Discrete Latent Codes for Content-Based Image Retrieval](https://mlsanity.com/paper/2103.12328/) (2021)
- [GANs Conditioning Methods: A Survey](https://mlsanity.com/paper/2408.15640/) (2024)
- [Leapfrog Latent Consistency Model (LLCM) for Medical Images Generation](https://mlsanity.com/paper/2411.15084/) (2024)
- [RADiff: Controllable Diffusion Models for Radio Astronomical Maps Generation](https://mlsanity.com/paper/2307.02392/) (2023)
- [Minimum Data, Maximum Impact: 20 annotated samples for explainable lung nodule classification](https://mlsanity.com/paper/2508.00639/) (2025)
- [Representation Synthesis by Probabilistic Many-Valued Logic Operation in Self-Supervised Learning](https://mlsanity.com/paper/2309.04148/) (2023)

## Abstract (arXiv)

Synthetic medical image generation has a huge potential for improving healthcare through many applications, from data augmentation for training machine learning systems to preserving patient privacy. Conditional Adversarial Generative Networks (cGANs) use a conditioning factor to generate images and have shown great success in recent years. Intuitively, the information in an image can be divided into two parts: 1) content which is presented through the conditioning vector and 2) style which is the undiscovered information missing from the conditioning vector. Current practices in using cGANs for medical image generation, only use a single variable for image generation (i.e., content) and therefore, do not provide much flexibility nor control over the generated image. In this work we propose a methodology to learn from the image itself, disentangled representations of style and content, and use this information to impose control over the generation process. In this framework, style is learned in a fully unsupervised manner, while content is learned through both supervised learning (using the conditioning vector) and unsupervised learning (with the inference mechanism). We undergo two novel regularization steps to ensure content-style disentanglement. First, we minimize the shared information between content and style by introducing a novel application of the gradient reverse layer (GRL); second, we introduce a self-supervised regularization method to further separate information in the content and style variables. We show that in general, two latent variable models achieve better performance and give more control over the generated image. We also show that our proposed model (DRAI) achieves the best disentanglement score and has the best overall performance.
