Hypergraph-Based Doubly Robust Estimation for Causal Inference of Drug Combination Effects in Heart Failure Treatment
Xiao, T.; Yeh, Y.-Y.; Lee, Y. A.; Guo, Y.; Guo, J.; Jiang, Z.
Show abstract
AO_SCPLOWBSTRACTC_SCPLOWDisease management for heart failure with preserved ejection fraction (HFpEF) requires understanding the comparative effectiveness of real-world drug combinations rather than single agents. Standard randomized controlled trials (RCTs) for multi-drug regimens are prohibitively expensive, slow, and often infeasible at scale, motivating the use of causal machine learning methods on large-scale electronic health records (EHRs). However, reliable estimation of treatment effects is challenging due to the high-order drug-drug interactions, strong confounding factors, and patient heterogeneity across sex, age, and other statuses. Existing causal machine learning methods mostly focus on comparing treatments with pairwise drug combinations. Techniques for multi-drug regimens are less studied. To fill this gap, we propose a Hypergraph-based Doubly Robust framework for HF-pEF (HyperDR), which represents six HF drug classes as nodes and observed multi-drug regimens as hyperedges, and uses a hypergraph neural network to learn shared representations for both drugs and combinations from cross-sectional EHR data. On top of these representations, we jointly train a propensity-score model and an outcome model with a doubly robust objective that combines cross-entropy losses with an augmented inverse-probability-weighted regularizer. This enables consistent treatment-effect estimation when either component is correctly specified while stabilizing learning under rare regimens. Experiments on a real-world HFpEF cohort show that HyperDR improves outcome prediction (hospitalization risk) compared with baseline methods. We also did case studies to interpret model results in treatment rankings across different patient subgroups.
Matching journals
The top 6 journals account for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
Similar papers in this journal
Similar papers in this journal
- High-dimensional Biomarker Identification for Scalable and Interpretable Disease Prediction via Machine Learning Models 93%
- TUGDA: Task uncertainty guided domain adaptation for robust generalization of cancer drug response prediction from in vitro to in vivo settings 93%
- ARTEMIS integrates autoencoders and schrodinger bridges to predict continuous dynamics of gene expression, cell population and perturbation from time-series single-cell data 92%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.