Learning heterogeneous treatment effects under principal stratification

• URL: http://arxiv.org/abs/2606.29076v1
• Date: Sat, 27 Jun 2026 20:27:17 GMT
• Title: Learning heterogeneous treatment effects under principal stratification
• Abstract summary: We study the identification and estimation of the conditional principal causal effects under principal ignorability.<n>We propose a novel doubly cross-fit doubly robust machine that resolves the nested nuisance structure inherent to principal stratification.
• Score: 4.965155763091777
• License: http://creativecommons.org/licenses/by-nc-sa/4.0/
• Abstract: Principal stratification provides a foundational framework for causal inference with intermediate outcomes by defining causal effects within subpopulations, yet existing work has largely focused on average effects across strata rather than treatment effect heterogeneity within strata. Such within-stratum heterogeneity informs individualized treatment decisions but the associated methods are sparse. We address this gap by studying the identification and estimation of the conditional principal causal effects under principal ignorability combined with an odds ratio sensitivity parameterization, which relaxes the monotonicity assumption. To efficiently learn these estimands, we propose a novel doubly cross-fit doubly robust machine learner that resolves the nested nuisance structure inherent to principal stratification. Leveraging sequential orthogonal learning with regularized least-squares sieves, we derive $\mathcal{L}^2$ and uniform limit theory, establish oracle efficiency, and construct uniform confidence bands for the proposed estimator. We use simulations to demonstrate the finite-sample performance of our estimator, and provide an empirical analysis of a randomized trial in acute lung injury, revealing informative patterns of treatment effect heterogeneity within the always-survivor subpopulation.

Related papers

• Disentangled Double Machine Learning for Accurate Causal Effect Estimation [80.07136847623498]
  Confounding bias is a key challenge in causal effect estimation from observational data.<n>Double Machine Learning (DML) addresses this issue by estimating treatment and outcome nuisance functions.<n>DML often produces biased and unstable estimates in highdimensional or finite-sample scenarios.<n>We propose Disentangled Double Machine Learning (DDML), a novel algorithm that integrates two key strategies.
  arXiv  Detail & Related papers  (2026-05-24T01:40:43Z)
• Local Covariate Selection for Average Causal Effect Estimation without Pretreatment and Causal Sufficiency Assumptions [18.19166664646723]
  Existing approaches typically rely on global causal structure learning over all variables, or on strong assumptions such as causal sufficiency.<n>We propose a novel local learning method for covariate selection in nonparametric causal effect estimation that avoids both the pretreatment and causal sufficiency assumptions.<n> Experiments on multiple synthetic datasets and two real-world datasets show that our approach achieves accurate causal effect estimation while substantially improving computational efficiency.
  arXiv  Detail & Related papers  (2026-05-20T11:08:09Z)
• Estimation of heterogeneous principal effects under principal ignorability [5.858849874821984]
  We study estimation and inference for heterogeneous principal causal effects with binary treatments and binary intermediate variables.<n>We propose a framework for estimating and forming pointwise confidence intervals for heterogeneous principal causal effects.
  arXiv  Detail & Related papers  (2026-03-09T21:52:41Z)
• Causal Clustering for Conditional Average Treatment Effects Estimation and Subgroup Discovery [5.669361767058639]
  Estimating heterogeneous treatment effects is critical in domains such as personalized medicine, resource allocation, and policy evaluation.<n>We propose a novel framework that clusters individuals based on estimated treatment effects using a learned kernel derived from causal forests.
  arXiv  Detail & Related papers  (2025-09-06T17:01:23Z)
• Deep Disentangled Representation Network for Treatment Effect Estimation [17.787544758117377]
  Estimating individual-level treatment effect from observational data is a fundamental problem in causal inference.<n>We propose a novel treatment effect estimation algorithm that incorporates a mixture of experts with multi-head attention.<n>We conduct extensive experiments on both public semi-synthetic and real-world production datasets.
  arXiv  Detail & Related papers  (2025-07-09T08:29:37Z)
• Flexible Nonparametric Inference for Causal Effects under the Front-Door Model [2.6900047294457683]
  We develop novel one-step and targeted minimum loss-based estimators for both the average treatment effect and the average treatment effect on the treated under front-door assumptions.<n>Our estimators are built on multiple parameterizations of the observed data distribution, including approaches that avoid mediator density entirely.<n>We show how these constraints can be leveraged to improve the efficiency of causal effect estimators.
  arXiv  Detail & Related papers  (2023-12-15T22:04:53Z)
• Benchmarking Bayesian Causal Discovery Methods for Downstream Treatment Effect Estimation [137.3520153445413]
  A notable gap exists in the evaluation of causal discovery methods, where insufficient emphasis is placed on downstream inference. We evaluate seven established baseline causal discovery methods including a newly proposed method based on GFlowNets. The results of our study demonstrate that some of the algorithms studied are able to effectively capture a wide range of useful and diverse ATE modes.
  arXiv  Detail & Related papers  (2023-07-11T02:58:10Z)
• Nonparametric Identifiability of Causal Representations from Unknown Interventions [63.1354734978244]
  We study causal representation learning, the task of inferring latent causal variables and their causal relations from mixtures of the variables. Our goal is to identify both the ground truth latents and their causal graph up to a set of ambiguities which we show to be irresolvable from interventional data.
  arXiv  Detail & Related papers  (2023-06-01T10:51:58Z)
• SurvITE: Learning Heterogeneous Treatment Effects from Time-to-Event Data [83.50281440043241]
  We study the problem of inferring heterogeneous treatment effects from time-to-event data. We propose a novel deep learning method for treatment-specific hazard estimation based on balancing representations.
  arXiv  Detail & Related papers  (2021-10-26T20:13:17Z)
• Efficient Causal Inference from Combined Observational and Interventional Data through Causal Reductions [68.6505592770171]
  Unobserved confounding is one of the main challenges when estimating causal effects. We propose a novel causal reduction method that replaces an arbitrary number of possibly high-dimensional latent confounders. We propose a learning algorithm to estimate the parameterized reduced model jointly from observational and interventional data.
  arXiv  Detail & Related papers  (2021-03-08T14:29:07Z)
• Almost-Matching-Exactly for Treatment Effect Estimation under Network Interference [73.23326654892963]
  We propose a matching method that recovers direct treatment effects from randomized experiments where units are connected in an observed network. Our method matches units almost exactly on counts of unique subgraphs within their neighborhood graphs.
  arXiv  Detail & Related papers  (2020-03-02T15:21:20Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.