Estimating Individualized Treatment Effects in Acute Ischemic Stroke with Causal Transformation Models (TRAM-DAG): A Multi-Centre Observational Study with External RCT Validation
- URL: http://arxiv.org/abs/2606.12623v2
- Date: Mon, 15 Jun 2026 05:34:41 GMT
- Title: Estimating Individualized Treatment Effects in Acute Ischemic Stroke with Causal Transformation Models (TRAM-DAG): A Multi-Centre Observational Study with External RCT Validation
- Abstract summary: Mechanical thrombectomy has been shown to be more effective on average than lysis in randomized controlled trials (RCTs)<n>We aim to identify which individual patients benefit most from mechanical thrombectomy compared to lysis.<n>We demonstrate that causal transformation models on directed acyclic graphs (TRAM-DAG) can be used for ITE estimation after being fitted on observational stroke patient data.
- Score: 0.5359378066251386
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Personalized medicine in acute ischemic stroke requires moving beyond average treatment effects (ATE) to individualized treatment effect (ITE) estimates to support treatment decisions. In acute ischemic stroke, mechanical thrombectomy has been shown to be more effective on average than lysis in randomized controlled trials (RCTs), such as the MR CLEAN study. We aim to identify which individual patients benefit most from mechanical thrombectomy compared to lysis. The outcome of interest is the modified Rankin Scale (mRS) at three months, an ordinal measure of functional disability (0: no symptoms, 6: death). We demonstrate that causal transformation models on directed acyclic graphs (TRAM-DAG) can be used for ITE estimation after being fitted on observational MAGIC multi-center stroke patient data. To ensure comparability with the MR CLEAN population, which we use for validation, we train the TRAM-DAG on a MAGIC sub-population with NIHSS at admission >= 6, corresponding to one inclusion criterion of MR CLEAN. The fitted model is then used to estimate ITEs for stroke patients in the MR CLEAN population. While these ITE estimates cannot be confirmed experimentally, we show that their average is consistent with the trial's reported ATE. Furthermore, the ITE estimates correctly rank trial patients by their observed frequency of a good outcome (mRS at three months <= 2). These findings support the use of causal models like TRAM-DAG for personalized decision-making in stroke care and highlight their ability to bridge the gap between observational evidence and clinical trials.
Related papers
- Metamorphic Testing for Clinical ML Models: A Framework Proposal and Pilot Study [3.4905170154782343]
Machine learning models for clinical prediction tasks routinely achieve high AUROC scores.<n> AUROC measures ranking performance rather than clinical sensibility.<n>This paper proposes applying metamorphic testing to machine learning models to evaluate behavioral correctness.
arXiv Detail & Related papers (2026-07-25T01:42:45Z) - Machine learning models for estimating counterfactuals in a single-arm inflammatory bowel disease study [1.685177132716279]
Single-arm trials accelerate study timelines by reducing the number of patients that must be recruited for a concurrent control group.<n>One approach is to construct a virtual control arm using a machine learning (ML) model trained on external control data to predict the counterfactual outcomes of the treatment arm.<n>Our study supports virtual controls as a viable and effective substitute for expensive, lengthy or unethical patient recruitment in an inflammatory bowel disease (IBD) trial.
arXiv Detail & Related papers (2026-04-25T23:09:07Z) - TEA-Time: Transporting Effects Across Time [44.54570296032634]
We develop a framework for temporal transportation: extrapolating treatment effects to time periods where no experiment was conducted.<n>We provide two identification strategies -- one using replicated trials comparing the same treatments at different times, another using common treatment arms observed across time.
arXiv Detail & Related papers (2026-03-07T03:34:13Z) - SurgeryLSTM: A Time-Aware Neural Model for Accurate and Explainable Length of Stay Prediction After Spine Surgery [44.119171920037196]
We develop and evaluate machine learning (ML) models for predicting length of stay (LOS) in elective spine surgery.<n>We compare traditional ML models with our developed model, SurgeryLSTM, a masked bidirectional long short-term memory (BiLSTM) with an attention.<n>Performance was evaluated using the coefficient of determination (R2) and key predictors were identified using explainable AI.
arXiv Detail & Related papers (2025-07-15T01:18:28Z) - Beyond the ATE: Interpretable Modelling of Treatment Effects over Dose and Time [46.2482873419289]
We propose a framework for modelling treatment effect trajectories as smooth surfaces over dose and time.<n>Our approach decouples the estimation of trajectory shape from the specification of clinically relevant properties.<n>We show that our method yields accurate, interpretable, and editable models of treatment dynamics.
arXiv Detail & Related papers (2025-07-09T20:33:33Z) - Causal machine learning for heterogeneous treatment effects in the presence of missing outcome data [0.9087641068861047]
We discuss the impact that missing outcome data has on causal machine learning estimators for the conditional average treatment effect (CATE)<n>We propose two de-biased machine learning estimators for the CATE, the mDR-learner and mEP-learner, which address the issue of under-representation.<n>We show that under reasonable conditions, these estimators are oracle efficient, and illustrate their favorable performance through simulated data settings.
arXiv Detail & Related papers (2024-12-27T16:10:03Z) - Treatment-wise Glioblastoma Survival Inference with Multi-parametric
Preoperative MRI [34.830878479276286]
We propose a treatment-conditioned regression model for glioblastoma ST that incorporates treatment information in addition to MR scans.
Our approach allows us to effectively utilize the data from all of the treatments in a unified manner, rather than having to train separate models for each of the treatments.
arXiv Detail & Related papers (2024-02-10T16:13:09Z) - Towards Outcome-Driven Patient Subgroups: A Machine Learning Analysis
Across Six Depression Treatment Studies [41.34047608276278]
We analyzed data from six clinical trials of pharmacological treatment for depression using a neural network model.
A model classifying remission and outputting individual remission probabilities for five first-line monotherapies and three combination treatments was trained.
Post-hoc analyses yielded clusters (subgroups) based on patient prototypes learned during training.
arXiv Detail & Related papers (2023-03-24T14:34:09Z) - TranSOP: Transformer-based Multimodal Classification for Stroke
Treatment Outcome Prediction [2.358784542343728]
We propose a transformer-based multimodal network (TranSOP) for a classification approach that employs clinical metadata and imaging information.
This includes a fusion module to efficiently combine 3D non-contrast computed tomography (NCCT) features and clinical information.
In comparative experiments using unimodal and multimodal data, we achieve a state-of-the-art AUC score of 0.85.
arXiv Detail & Related papers (2023-01-25T21:05:10Z) - MIA-Prognosis: A Deep Learning Framework to Predict Therapy Response [58.0291320452122]
This paper aims at a unified deep learning approach to predict patient prognosis and therapy response.
We formalize the prognosis modeling as a multi-modal asynchronous time series classification task.
Our predictive model could further stratify low-risk and high-risk patients in terms of long-term survival.
arXiv Detail & Related papers (2020-10-08T15:30:17Z) - Prediction of Thrombectomy Functional Outcomes using Multimodal Data [2.358784542343728]
We propose a novel deep learning approach to directly exploit multimodal data to estimate the success of endovascular treatment.
We incorporate an attention mechanism in our architecture to model global feature inter-dependencies, both channel-wise and spatially.
arXiv Detail & Related papers (2020-05-26T21:51:58Z) - Estimating Counterfactual Treatment Outcomes over Time Through
Adversarially Balanced Representations [114.16762407465427]
We introduce the Counterfactual Recurrent Network (CRN) to estimate treatment effects over time.
CRN uses domain adversarial training to build balancing representations of the patient history.
We show how our model achieves lower error in estimating counterfactuals and in choosing the correct treatment and timing of treatment.
arXiv Detail & Related papers (2020-02-10T20:47:36Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.