PyTMLE: A Flexible Python Library for Targeted Estimation of Survival and Competing Risks using Causal Machine Learning
Guski, J.; Aborageh, M.; Fröhlich, H.
Show abstract
BackgroundTargeted estimation offers a robust and unbiased approach for causal inference of the average treatment effect (ATE) from observational data, even with confounding, dependent censoring, and competing risks. Its advantages include double robustness, statistical rigor, and flexible data-adaptive modeling, potentially leveraging machine/deep learning. However, existing implementations lack model selection flexibility and are R-based, hindering adoption by the Python-focused machine learning community. ResultsWe propose PyTMLE, a flexible Python package for causal machine learning-based targeted estimation with survival outcomes and competing risks. PyTMLE supports scikit-survival and pycox, and inbuilt robustness checks based on E-values. PyTMLE is easy to use with initial estimation of nuisance parameters that are obtained via super learning by default. We showcase its basic usage on the established Hodgkins disease dataset, where our package reveals the protective effect of chemotherapy on relapse risk. ConclusionsThis package promotes targeted estimation in time-to-event analysis for applied machine learning, enabling fully data-adaptive nuisance parameter estimation, potentially with deep learning. Future enhancements may include time-dependent confounders and dynamic treatment regimes.
Matching journals
The top 1 journal accounts for 50% of the predicted probability mass.
Similar papers in this journal
- External control arm analysis: an evaluation of propensity score approaches, G-computation, and doubly debiased machine learning 96%
- Prediction-powered Inference for Clinical Trials 95%
- Comparison of Bayesian networks, G-estimation and linear models to estimate causal treatment effects in aggregated N-of-1 trials 94%
Similar papers in this journal
- A scalable approach for continuous time Markov models with covariates 93%
- Tree-informed Bayesian multi-source domain adaptation: cross-population probabilistic cause-of-death assignment using verbal autopsy 93%
- Adaptive Transfer Learning for Time-to-Event Modeling with Applications in Disease Risk Assessment 93%
Similar papers in this journal
- Causal Analysis for Multivariate Integrated Clinical and Environmental Exposures Data 93%
- Combining symbolic regression with the Cox proportional hazards model improves prediction of heart failure deaths 93%
- Addressing Label Noise for Electronic Health Records: Insights from Computer Vision for Tabular Data 92%
Similar papers in this journal
- A Stability-Enhanced Lasso Approach for Covariate Selection in Non-Linear Mixed Effect Model 95%
- A Double Machine Learning Approach for the Evaluation of COVID-19 Vaccine Effectiveness under the Test-Negative Design: Analysis of Québec Administrative Data 95%
- Penalized reduced rank regression for multi-outcome survival data supports a common metabolic risk score for age-related diseases 94%
Similar papers in this journal
- An Explainable Multi-Modal Neural Network Architecture for Predicting Epilepsy Comorbidities Based on Administrative Claims Data 91%
- Data-driven Discovery of Mathematical and Physical Relations in Oncology Data using Human-understandable Machine Learning 90%
- Fitting drift-diffusion decision models to trial-by-trial data 89%
"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.