Saved in:
Bibliographic Details
Main Authors: Liu, Richard, Williams, Nicholas T., Rudolph, Kara E., Díaz, Iván
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2408.14620
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916791257661440
author Liu, Richard
Williams, Nicholas T.
Rudolph, Kara E.
Díaz, Iván
author_facet Liu, Richard
Williams, Nicholas T.
Rudolph, Kara E.
Díaz, Iván
contents Causal mediation analyses investigate the mechanisms through which causes exert their effects, and are therefore central to scientific progress. The literature on the non-parametric definition and identification of mediational effects in rigourous causal models has grown significantly in recent years, and there has been important progress to address challenges in the interpretation and identification of such effects. Despite great progress in the causal inference front, statistical methodology for non-parametric estimation has lagged behind, with few or no methods available for tackling non-parametric estimation in the presence of multiple, continuous, or high-dimensional mediators. In this paper we show that the identification formulas for six popular non-parametric approaches to mediation analysis proposed in recent years can be recovered from just two statistical estimands. We leverage this finding to propose an all-purpose one-step estimation algorithm that can be coupled with machine learning in any mediation study that uses any of these six definitions of mediation. The estimators have desirable properties, such as $\sqrt{n}$-convergence and asymptotic normality. Estimating the first-order correction for the one-step estimator requires estimation of complex density ratios on the potentially high-dimensional mediators, a challenge that is solved using recent advancements in so-called Riesz learning. We illustrate the properties of our methods in a simulation study and illustrate its use on real data to estimate the extent to which pain management practices mediate the total effect of having a chronic pain disorder on opioid use disorder.
format Preprint
id arxiv_https___arxiv_org_abs_2408_14620
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle General targeted machine learning for modern causal mediation analysis
Liu, Richard
Williams, Nicholas T.
Rudolph, Kara E.
Díaz, Iván
Machine Learning
Causal mediation analyses investigate the mechanisms through which causes exert their effects, and are therefore central to scientific progress. The literature on the non-parametric definition and identification of mediational effects in rigourous causal models has grown significantly in recent years, and there has been important progress to address challenges in the interpretation and identification of such effects. Despite great progress in the causal inference front, statistical methodology for non-parametric estimation has lagged behind, with few or no methods available for tackling non-parametric estimation in the presence of multiple, continuous, or high-dimensional mediators. In this paper we show that the identification formulas for six popular non-parametric approaches to mediation analysis proposed in recent years can be recovered from just two statistical estimands. We leverage this finding to propose an all-purpose one-step estimation algorithm that can be coupled with machine learning in any mediation study that uses any of these six definitions of mediation. The estimators have desirable properties, such as $\sqrt{n}$-convergence and asymptotic normality. Estimating the first-order correction for the one-step estimator requires estimation of complex density ratios on the potentially high-dimensional mediators, a challenge that is solved using recent advancements in so-called Riesz learning. We illustrate the properties of our methods in a simulation study and illustrate its use on real data to estimate the extent to which pain management practices mediate the total effect of having a chronic pain disorder on opioid use disorder.
title General targeted machine learning for modern causal mediation analysis
topic Machine Learning
url https://arxiv.org/abs/2408.14620