Longitudinal Targeted Minimum Loss-based Estimation with Temporal-Difference Heterogeneous Transformer
Fuente:
arXiv
Guardado en:
| Autores principales: | , , , , , , , |
|---|---|
| Formato: | Preprint |
| Publicado: |
2024
|
| Materias: | |
| Acceso en línea: | |
| Etiquetas: |
Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
|
| _version_ | 1866918047008161792 |
|---|---|
| author | Shirakawa, Toru Li, Yi Wu, Yulun Qiu, Sky Li, Yuxuan Zhao, Mingduo Iso, Hiroyasu van der Laan, Mark |
| author_facet | Shirakawa, Toru Li, Yi Wu, Yulun Qiu, Sky Li, Yuxuan Zhao, Mingduo Iso, Hiroyasu van der Laan, Mark |
| contents | We propose Deep Longitudinal Targeted Minimum Loss-based Estimation (Deep LTMLE), a novel approach to estimate the counterfactual mean of outcome under dynamic treatment policies in longitudinal problem settings. Our approach utilizes a transformer architecture with heterogeneous type embedding trained using temporal-difference learning. After obtaining an initial estimate using the transformer, following the targeted minimum loss-based likelihood estimation (TMLE) framework, we statistically corrected for the bias commonly associated with machine learning algorithms. Furthermore, our method also facilitates statistical inference by enabling the provision of 95% confidence intervals grounded in asymptotic statistical theory. Simulation results demonstrate our method's superior performance over existing approaches, particularly in complex, long time-horizon scenarios. It remains effective in small-sample, short-duration contexts, matching the performance of asymptotically efficient estimators. To demonstrate our method in practice, we applied our method to estimate counterfactual mean outcomes for standard versus intensive blood pressure management strategies in a real-world cardiovascular epidemiology cohort study. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2404_04399 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Longitudinal Targeted Minimum Loss-based Estimation with Temporal-Difference Heterogeneous Transformer Shirakawa, Toru Li, Yi Wu, Yulun Qiu, Sky Li, Yuxuan Zhao, Mingduo Iso, Hiroyasu van der Laan, Mark Machine Learning Artificial Intelligence Applications Methodology We propose Deep Longitudinal Targeted Minimum Loss-based Estimation (Deep LTMLE), a novel approach to estimate the counterfactual mean of outcome under dynamic treatment policies in longitudinal problem settings. Our approach utilizes a transformer architecture with heterogeneous type embedding trained using temporal-difference learning. After obtaining an initial estimate using the transformer, following the targeted minimum loss-based likelihood estimation (TMLE) framework, we statistically corrected for the bias commonly associated with machine learning algorithms. Furthermore, our method also facilitates statistical inference by enabling the provision of 95% confidence intervals grounded in asymptotic statistical theory. Simulation results demonstrate our method's superior performance over existing approaches, particularly in complex, long time-horizon scenarios. It remains effective in small-sample, short-duration contexts, matching the performance of asymptotically efficient estimators. To demonstrate our method in practice, we applied our method to estimate counterfactual mean outcomes for standard versus intensive blood pressure management strategies in a real-world cardiovascular epidemiology cohort study. |
| title | Longitudinal Targeted Minimum Loss-based Estimation with Temporal-Difference Heterogeneous Transformer |
| topic | Machine Learning Artificial Intelligence Applications Methodology |
| url | https://arxiv.org/abs/2404.04399 |