MedDiTPro: A Prompt-Guided Diffusion Transformer for Multimodal Longitudinal Medical Data Synthesis
Yuan Zhong, Xiaochen Wang, Jiaqi Wang, Xiaokun Zhang, Fenglong Ma
Abstract
Diffusion models have recently emerged as a state-of-the-art approach for synthetic Electronic Health Record (EHR) generation, offering superior fidelity and diversity over traditional generative models. However, existing diffusion-based methods struggle with unique challenges: limited representation learning and modality utilization, where they fail to explicitly capture inter-modality dependencies and fine-grained code-level interactions, and constrained adaptability due to reliance on U-Net-based architectures, which are not well-suited for handling the heterogeneous and evolving nature of EHR data. Furthermore, current evaluation paradigms rely on either perplexity-based sequence modeling or global distributional measures, lacking robustness in assessing both intra-visit code relationships and inter-visit temporal patterns. To address these limitations, we propose MedDiTPro, a diffusion transformer-based framework that enhances multimodal EHR generation by integrating structured modality-aware guidance. Through a unified transformer for intra-visit representation learning, a modality-specific and datawise prompt learner, and a diffusion transformer with structured guidance, MedDiTPro achieves state-of-the-art performance in generating diverse and clinically meaningful synthetic records. Extensive experiments on publicly available datasets demonstrate that MedDiTPro achieves state-of-the-art fidelity, privacy preservation, and utility.
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Your agent calls
Lunesearch_papers
Free to start. No credit card required.
Terminal
Install the CLIlune papers get 963a3e0e-e488-4c81-b8ca-06a9bdec0593Related papers
- A Flexible Generative Model for Heterogeneous Tabular EHR with Missing ModalityHuan He, William Hao, Yuanzhe Xi, Yong Chen et al.ICLR 2024 · 16 citations
- Synthesizing Multimodal Electronic Health Records via Predictive Diffusion ModelsYuan Zhong, Xiaochen Wang, Jiaqi Wang, Xiaokun Zhang et al.KDD 2024 · 4 citations
- Generating Multi-Table Time Series EHR from Latent Space with Minimal PreprocessingEunbyeol Cho, Jiyoun Kim, Minjae Lee, Sungjin Park et al.NeurIPS 2025
- PromptEHR: Conditional Electronic Healthcare Records Generation with Prompt LearningZifeng Wang, Jimeng SunEMNLP 2022 · 19 citations
- TarDiff: Target-Oriented Diffusion Guidance for Synthetic Electronic Health Record Time Series GenerationBowen Deng, Chang Xu, Hao Li, Yu-Hao Huang et al.KDD 2025
