The Learnability Gap in Medical Latent Diffusion
On the map: Diffusion models → Medical imaging and segmentation
Abstract (arXiv)
Generative data augmentation with latent diffusion models is a promising strategy for addressing class imbalance in medical imaging, yet current approaches focus on perceptual fidelity and domain-specific autoencoder fine-tuning while neglecting a more fundamental bottleneck. We identify and formalize the learnability gap: large-scale pretrained autoencoders faithfully encode discriminative features for medical classification, as evidenced by near-lossless performance in reconstruction space, yet their latent representations are structured in ways that are difficult for classifiers to learn from. Across five autoencoder families and four medical benchmarks spanning chest radiography, dermatoscopy, computed tomography, and echocardiography, we show that this gap persists regardless of architecture, initialization strategy, or hyperparameter tuning, and that medical-domain fine-tuning of the autoencoder does not close it. To probe and partially narrow the gap, we develop noise-conditioned latent classifiers with FiLM layers and image-space distillation that offer 64x throughput and 120x memory gains over image-space models while serving as diagnostic tools for latent space quality. Our analysis provides a new framework for evaluating autoencoder latent spaces and identifies their structure, rather than their fidelity or domain specificity, as the primary obstacle to closing the performance gap between real and synthetic medical training data.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Medical imaging and segmentation, one of 14 directions in Diffusion models. Builds on 34 of its 42 references that are in the map. First posted on arXiv 16 May 2026.
Builds on
34 of 42 references.
References · 34
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2022Classifier-Free Diffusion Guidancecited by 5,847
- 2022A ConvNet for the 2020scited by 3,579
- 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,333
- 2019Learning Imbalanced Datasets with Label-Distribution-Aware Margin Losscited by 989
- 2019Decoupling Representation and Classifier for Long-Tailed Recognitioncited by 809
- 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkcited by 703
22 more
- 2020The Future of Digital Health with Federated Learningcited by 633
- 2017Skin Lesion Analysis Toward Melanoma Detection: A Challenge at the 2017 International Symposium on Biomedical Imaging (ISBI), Hosted by the International Skin Imaging Collaboration (ISIC)cited by 549
- 2023Analyzing and Improving the Training Dynamics of Diffusion Modelscited by 402
- 2025Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Modelscited by 393
- 2024Guiding a Diffusion Model with a Bad Version of Itselfcited by 299
- 2025REPA-E: Unlocking VAE for End-to-End Tuning with Latent Diffusion Transformerscited by 225
- 2024Generalist Foundation Models from a Multimodal Dataset for 3D Computed Tomographycited by 199
- 2025EQ-VAE: Equivariance Regularized Latent Space for Improved Generative Image Modelingcited by 89
- 2025Improving the Diffusability of Autoencoderscited by 80
- 2024Isambard-AI: a leadership class supercomputer optimised specifically for Artificial Intelligencecited by 62
- 2025A Fourier Space Perspective on Diffusion Modelscited by 52
- 2023Class-Balancing Diffusion Modelscited by 48
- 2022Long-Tailed Classification of Thorax Diseases on Chest X-Ray: A New Benchmark Studycited by 45
- 2023Towards long-tailed, multi-label disease classification from chest X-ray: Overview of the CXR-LT challengecited by 22
- 2024Image Generation Diversity Issues and How to Tame Themcited by 22
- 2025MedVAE: Efficient Automated Interpretation of Medical Images with Large-Scale Generalizable Autoencoderscited by 22
- 2025Alias-Free Latent Diffusion Models: Improving Fractional Shift Equivariance of Diffusion Latent Spacecited by 14
- 2025Adapting Self-Supervised Representations as a Latent Space for Efficient Generationcited by 13
- 2025CXR-LT 2024: A MICCAI challenge on long-tailed, multi-label, and zero-shot disease classification from chest X-raycited by 10
- 2025LCMem: A Universal Model for Robust Image Memorization Detectioncited by 4
- 2025CARDIUM: Congenital Anomaly Recognition with Diagnostic Images and Unified Medical recordscited by 2
- 2025GRASP: Guided Residual Adapters with Sample-wise Partitioningcited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026CONFLUX: A Latent Diffusion Model for 3D Chest-CT Synthesis with RL Post-Training
- 2023Investigating Data Memorization in 3D Latent Diffusion Models for Medical Image Synthesis
- 2026Domain-Specific Latent Representations Improve the Fidelity of Diffusion-Based Medical Image Super-Resolution
- 2025Conditional Diffusion Models are Medical Image Classifiers that Provide Explainability and Uncertainty for Free
- 2024Unconditional Latent Diffusion Models Memorize Patient Imaging Data: Implications for Openly Sharing Synthetic Data
- 2025Latent Diffusion Autoencoders: Toward Efficient and Meaningful Unsupervised Representation Learning in Medical Imaging
- 2025Anatomy-Grounded Weakly Supervised Prompt Tuning for Chest X-ray Latent Diffusion Models
- 2022Adapting Pretrained Vision-Language Foundational Models to Medical Imaging Domains