MM-DADM: Multimodal Drug-Aware Diffusion Model for Virtual Clinical Trials

View PDF HTML (experimental)

Abstract:High failure rates in cardiac drug development necessitate virtual clinical trials via electrocardiogram (ECG) generation to reduce risks and costs. However, existing ECG generation models struggle to balance morphological realism with pathological flexibility, fail to disentangle demographics from genuine drug effects, and are severely bottlenecked by early-phase data scarcity. To overcome these hurdles, we propose the Multimodal Drug-Aware Diffusion Model (MM-DADM), the first generative framework for generating individualized drug-induced ECGs. Specifically, our proposed MM-DADM integrates a Dynamic Cross-Attention (DCA) module that adaptively fuses External Physical Knowledge (EPK) to preserve morphological realism while avoiding the suppression of complex pathological nuances. To resolve feature entanglement, a Causal Feature Encoder (CFE) actively filters out demographic noise to extract pure pharmacological representations. These representations subsequently guide a Causal-Disentangled ControlNet (CDC-Net), which leverages counterfactual data augmentation to explicitly learn intrinsic pharmacological mechanisms despite limited clinical data. Extensive experiments on $9,443$ ECGs across $8$ drug regimens demonstrate that MM-DADM outperforms $10$ state-of-the-art ECG generation models, improving simulation accuracy by at least $6.13%$ and recall by $5.89%$, while providing highly effective data augmentation for downstream classification tasks.

Comments: Under review

Subjects:

Machine Learning (cs.LG); Quantitative Methods (q-bio.QM)

Cite as: arXiv:2502.07297 [cs.LG]

(or arXiv:2502.07297v3 [cs.LG] for this version)

https://doi.org/10.48550/arXiv.2502.07297

arXiv-issued DOI via DataCite

Submission history

From: Qian Shao [view email] [v1] Tue, 11 Feb 2025 06:50:33 UTC (2,862 KB) [v2] Sun, 18 May 2025 08:05:51 UTC (2,940 KB) [v3] Mon, 30 Mar 2026 10:51:39 UTC (2,987 KB)