Estimating Treatment Effect Heterogeneity in Psychiatry: A Review and Tutorial with Causal Forests

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Sverdrup, Erik, Petukhova, Maria, Wager, Stefan
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917868138921984
author Sverdrup, Erik
Petukhova, Maria
Wager, Stefan
author_facet Sverdrup, Erik
Petukhova, Maria
Wager, Stefan
contents Flexible machine learning tools are increasingly used to estimate heterogeneous treatment effects. This paper gives an accessible tutorial demonstrating the use of the causal forest algorithm, available in the R package grf. We start with a brief non-technical overview of treatment effect estimation methods, focusing on estimation in observational studies; the same techniques can also be applied in experimental studies. We then discuss the logic of estimating heterogeneous effects using the extension of the random forest algorithm implemented in grf. Finally, we illustrate causal forest by conducting a secondary analysis on the extent to which individual differences in resilience to high combat stress can be measured among US Army soldiers deploying to Afghanistan based on information about these soldiers available prior to deployment. We illustrate simple and interpretable exercises for model selection and evaluation, including targeting operator characteristics curves, Qini curves, area-under-the-curve summaries, and best linear projections. A replication script with simulated data is available at https://github.com/grf-labs/grf/tree/master/experiments/ijmpr
format Preprint
id arxiv_https___arxiv_org_abs_2409_01578
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Estimating Treatment Effect Heterogeneity in Psychiatry: A Review and Tutorial with Causal Forests
Sverdrup, Erik
Petukhova, Maria
Wager, Stefan
Applications
Flexible machine learning tools are increasingly used to estimate heterogeneous treatment effects. This paper gives an accessible tutorial demonstrating the use of the causal forest algorithm, available in the R package grf. We start with a brief non-technical overview of treatment effect estimation methods, focusing on estimation in observational studies; the same techniques can also be applied in experimental studies. We then discuss the logic of estimating heterogeneous effects using the extension of the random forest algorithm implemented in grf. Finally, we illustrate causal forest by conducting a secondary analysis on the extent to which individual differences in resilience to high combat stress can be measured among US Army soldiers deploying to Afghanistan based on information about these soldiers available prior to deployment. We illustrate simple and interpretable exercises for model selection and evaluation, including targeting operator characteristics curves, Qini curves, area-under-the-curve summaries, and best linear projections. A replication script with simulated data is available at https://github.com/grf-labs/grf/tree/master/experiments/ijmpr
title Estimating Treatment Effect Heterogeneity in Psychiatry: A Review and Tutorial with Causal Forests
topic Applications
url https://arxiv.org/abs/2409.01578