Dynamic Survival Transformers for Causal Inference with Electronic Health Records
Abstract
In medicine, researchers often seek to infer the effects of a given treatment on patients' outcomes, such as the expected time until infection. However, the standard methods for causal survival analysis make simplistic assumptions about the data-generating process and cannot capture complex interactions among patient covariates. We introduce the Dynamic Survival Transformer (DynST), a deep survival model that trains on electronic health records (EHRs). Unlike previous transformers used in survival analysis, DynST can make use of time-varying information to predict evolving survival probabilities. We derive a semi-synthetic EHR dataset from MIMIC-III to show that DynST can accurately estimate the causal effect of a treatment intervention on restricted mean survival time (RMST). We demonstrate that DynST achieves better predictive and causal estimation than two alternative models.
Cite
Text
Chatha et al. "Dynamic Survival Transformers for Causal Inference with Electronic Health Records." NeurIPS 2022 Workshops: TS4H, 2022.Markdown
[Chatha et al. "Dynamic Survival Transformers for Causal Inference with Electronic Health Records." NeurIPS 2022 Workshops: TS4H, 2022.](https://mlanthology.org/neuripsw/2022/chatha2022neuripsw-dynamic/)BibTeX
@inproceedings{chatha2022neuripsw-dynamic,
title = {{Dynamic Survival Transformers for Causal Inference with Electronic Health Records}},
author = {Chatha, Prayag and Wang, Yixin and Wu, Zhenke and Regier, Jeffrey},
booktitle = {NeurIPS 2022 Workshops: TS4H},
year = {2022},
url = {https://mlanthology.org/neuripsw/2022/chatha2022neuripsw-dynamic/}
}