CODE - XAI: Construing and Deciphering Treatment Effects via Explainable AI using Real-world Data.
Lu, M.; Covert, I.; White, N. J.; Lee, S.-I.
Show abstract
Clinicians rely on evidence from randomized controlled trials (RCTs) to decide on medical treatments for patients. However, RCTs often lack the granularity needed to inform decisions for individual patients or specific clinical scenarios. Recent advances in machine learning, particularly conditional average treatment effect (CATE) modeling, offer a promising approach for estimating patient-specific treatment effects. Yet, their adoption in clinical practice remains limited because these models often function as "black boxes", making it difficult to understand which features drive treatment effect heterogeneity among individuals. To overcome these barriers, we introduce LIFT-XAI, a robust and principled framework that interprets ensemble CATE models with Shapley values, thereby accurately identifying unique features and patient subgroups driving the treatment effects. We validate LIFT-XAI on real-world clinical data from four RCTs comprising over 42,000 patients. We demonstrate that LIFT-XAI uncovered fasting glucose as the primary factor explaining conflicting outcomes between two major blood pressure trials (SPRINT and ACCORD) and discovered age as a critical treatment modifier in a new clinical setting-a finding later validated by a subsequent RCT. By enabling such powerful cross-cohort analysis and the discovery of novel patient subgroups, LIFT-XAI advances precision medicine across diverse settings, from chronic disease management to acute care.
Matching journals
The top 2 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Automated Interpretable Discovery of Heterogeneous Treatment Effectiveness: A Covid-19 Case Study 96%
- Creating a computer assisted ICD coding system: performance metric choice and use of the ICD hierarchy 96%
- ARCH: Large-scale Knowledge Graph via Aggregated Narrative Codified Health Records Analysis 95%
Similar papers in this journal
Similar papers in this journal
Similar papers in this journal
- Bi-level Graph Learning Unveils Prognosis-Relevant Tumor Microenvironment Patterns in Breast Multiplexed Digital Pathology 94%
- Generating hard-to-obtain information from easy-to-obtain information: applications in drug discovery and clinical inference 94%
- scELMo: Embeddings from Language Models are Good Learners for Single-cell Data Analysis 94%
Similar papers in this journal
- Modular Clinical Decision Support Networks (MoDN)—Updatable, Interpretable, and Portable Predictions for Evolving Clinical Environments 95%
- Clinical Time-to-Event Prediction Enhanced by Incorporating Compatible Related Outcomes 94%
- Explainable deep learning for disease activity prediction in chronic inflammatory joint diseases 93%
"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.