Synthetic Data Generation for Long-Tail Medical Image Classification: A Case Study in Skin Lesions
On the map: Diffusion models → Medical imaging and segmentation
Abstract (arXiv)
Long-tailed class distributions are pervasive in multi-class medical datasets and pose significant challenges for deep learning models which typically underperform on tail classes with limited samples. This limitation is particularly problematic in medical applications, where rare classes often correspond to severe or high-risk diseases and therefore require high diagnostic accuracy. Existing solutions-including specialized architectures, rebalanced loss functions, and handcrafted data augmentation-offer only marginal improvements and struggle to scale due to their limited and largely deterministic variability. To address these challenges, we introduce a diffusion-model-driven synthetic data augmentation pipeline tailored for medical long-tailed classification. Our approach features a novel inpainting diffusion model combined with an Out-of-Distribution (OOD) post-selection mechanism to ensure diverse, realistic, and clinically meaningful synthetic samples. Evaluated on the ISIC2019 skin lesion classification dataset, one of the largest and most imbalanced medical imaging benchmarks, our method yields substantial improvements in overall performance, with particularly pronounced gains on tail classes with more than 28% improvement on the class with the fewest samples. These results demonstrate the effectiveness of diffusion-based augmentation in mitigating long-tail imbalance and enhancing medical classification robustness.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Medical imaging and segmentation, one of 14 directions in Diffusion models. Builds on 28 of its 40 references that are in the map. First posted on arXiv 4 May 2026.
Builds on
28 of 40 references.
References · 28
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,415
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
- 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,466
- 2022A ConvNet for the 2020scited by 3,545
- 2022DreamBooth: Fine Tuning Text-to-Image Diffusion Models for Subject-Driven Generationcited by 3,286
- 2023Stable Video Diffusion: Scaling Latent Video Diffusion Models to Large Datasetscited by 2,526
16 more
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,404
- 2019Learning Imbalanced Datasets with Label-Distribution-Aware Margin Losscited by 987
- 2019Decoupling Representation and Classifier for Long-Tailed Recognitioncited by 806
- 2021Deep Long-Tailed Learning: A Surveycited by 416
- 2019BBN: Bilateral-Branch Network with Cumulative Learning for Long-Tailed Visual Recognitioncited by 403
- 2020Balanced Meta-Softmax for Long-Tailed Visual Recognitioncited by 397
- 2020Equalization Loss for Long-Tailed Object Recognitioncited by 285
- 2020Disentangling Label Distribution for Long-tailed Visual Recognitioncited by 172
- 2016Range Loss for Deep Face Recognition with Long-tailcited by 111
- 2021Label-Imbalanced and Group-Sensitive Classification under Overparameterizationcited by 74
- 2021Self Supervision to Distillation for Long-Tailed Visual Recognitioncited by 73
- 2021Learning Fast Sample Re-weighting Without Reward Datacited by 69
- 2023Class-Balancing Diffusion Modelscited by 47
- 2019Skin Lesion Classification Using Ensembles of Multi-Resolution EfficientNets with Meta Datacited by 43
- 2023CUDA: Curriculum of Data Augmentation for Long-Tailed Recognitioncited by 30
- 2024MONICA: Benchmarking on Long-tailed Medical Image Classificationcited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2023Augmenting medical image classifiers with synthetic data from latent diffusion models
- 2024Derm-T2IM: Harnessing Synthetic Skin Lesion Data via Stable Diffusion Models for Enhanced Skin Disease Classification using ViT and CNN
- 2023Diffusion-based Data Augmentation for Skin Disease Classification: Impact Across Original Medical Datasets to Fully Synthetic Images
- 2024LoGex: Improved tail detection of extremely rare histopathology classes via guided diffusion
- 2025Conditional Diffusion Models are Medical Image Classifiers that Provide Explainability and Uncertainty for Free
- 2026Few-Shot Synthetic Data Generation with Diffusion Models for Downstream Vision Tasks
- 2026DerMAE: Improving skin lesion classification through conditioned latent diffusion and MAE distillation
- 2025SkinDualGen: Prompt-Driven Diffusion for Simultaneous Image-Mask Generation in Skin Lesions