Diffusion‑synthesized Chest X‑rays improve fairness and diagnostic performance

Publication date

2026-04

Authors

Rehman, Dawood
Chen, Huan Yu
Lee, Chi Chun
Vilor-Tejedor, NataliaORCID 0000-0003-4935-6721
Upadhyay, Shreya
Kuo, Po Chih

Editors

Advisors

Supervisors

Document Type

Article
Open Access logo

License

cc_by

Abstract

Deep learning models have been widely applied to chest X-ray (CXR) disease classification and diagnosis; however, challenges such as data scarcity and shortcut learning often lead to biased model behavior. This study addresses fairness-related concerns in conventional deep learning models trained on CXR data and proposes mitigating demographic disparities through image synthesis. We fine-tune a pre-trained stable diffusion model using Low-Rank Adaptation (LoRA) and a CLIP tokenizer, incorporating low-rank constraints into key attention layers while preserving the original architecture. This enables the generation of high-quality, realistic CXR images with reduced parameter complexity. Experimental results demonstrate that models trained with our synthetic data achieve improved classification performance and exhibit significantly reduced disparities across demographic groups. Furthermore, the proposed models show increased attention to disease-relevant regions and diminished reliance on spurious shortcuts. These findings highlight the potential of generative AI in enhancing fairness in medical imaging workflows, particularly when combined with efficient and adaptable fine-tuning strategies.

Keywords

Health Informatics

Citation

Rehman, D, Chen, H Y, Lee, C C, Vilor-Tejedor, N, Upadhyay, S & Kuo, P C 2026, 'Diffusion‑synthesized Chest X‑rays improve fairness and diagnostic performance', PLOS Digital Health, vol. 5, e0001277. https://doi.org/10.1371/journal.pdig.0001277