---
title: Conditional Treatment-Averaged Causal Effect (C-TACE)
url: https://www.emergentmind.com/topics/conditional-treatment-averaged-causal-effect-c-tace
type: topic
---

# Conditional Treatment-Averaged Causal Effect (C-TACE)

Searching arXiv for the specified papers and closely related work on conditional treatment effects.
arXiv search query: 2509.26265 OR "Staged Event Trees for Transparent Treatment Effect Estimation"
Conditional Treatment-Averaged Causal Effect (C-TACE) denotes a covariate-conditional causal contrast: the effect of treatment evaluated within strata of observed characteristics rather than averaged over the full population. In the literature represented here, the standard term for this object is the conditional average treatment effect (CATE). For binary treatment and outcome with categorical covariates, it is the contrast between \(P(Y \mid \operatorname{do}(R=1), \mathbf Z)\) and \(P(Y \mid \operatorname{do}(R=0), \mathbf Z)\); in potential-outcomes notation it appears as \(f_P(x)=\mathbb{E}_P[Y^1-Y^0\mid X=x]\); and in nonparametric notation it is written as \(\tau(P)(w)=\mathbb{E}_P[Y\mid A=1,W=w]-\mathbb{E}_P[Y\mid A=0,W=w)\) [2509.26265] [2310.16819] [2605.15373]. Across these formulations, the defining role of C-TACE is to represent treatment-effect heterogeneity that can be obscured by the average treatment effect alone [2605.15373].

## 1. Terminological status and mathematical formulations

Within the cited work, “C-TACE” is not the standard label. The established term is CATE, and the relevant estimand is consistently the treatment effect conditional on covariates. The staged event tree paper states that it considers “the task of estimating the ATE or its conditional version (CATE) for a binary treatment variable (\(R\)) and a binary outcome (\(Y\)) when a set of additional categorical covariates (\(\bm Z\)) is available” [2509.26265]. The high-dimensional linear-regression paper likewise studies the conditional average treatment effect under the target
\[
f_P(x)=\mathbb{E}_P[Y^1-Y^0\mid X=x],
\]
and explicitly notes that it does not use the term “C-TACE” [2310.16819]. The nonparametric heterogeneity paper defines the same object as
\[
\tau(P)(w)=\mu(P)(1,w)-\mu(P)(0,w),
\]
with \(\mu(P)(a,w)=\mathbb{E}_P[Y\mid A=a,W=w]\) [2605.15373].

| Source | Formulation | Setting |
|---|---|---|
| [2509.26265] | \(\mathbb{E}[Y \mid \operatorname{do}(R=1), \mathbf Z=\mathbf z]-\mathbb{E}[Y \mid \operatorname{do}(R=0), \mathbf Z=\mathbf z]\) | Binary \(R\), binary \(Y\), categorical \(\mathbf Z\), staged event trees |
| [2310.16819] | \(f_P(x)=\mathbb{E}_P[Y^1-Y^0\mid X=x]\) | Two treatments, potential outcomes, high-dimensional linear regression |
| [2605.15373] | \(\tau(P)(w)=\mathbb{E}_P[Y\mid A=1,W=w]-\mathbb{E}_P[Y\mid A=0,W=w]\) | Nonparametric CATE function and heterogeneity summaries |

These formulations differ in notation and modeling assumptions, but they coincide in treating the conditional effect as a function of observed covariates. In the binary-outcome case considered in staged event trees, the contrast reduces to a difference in conditional outcome probabilities within a covariate-defined subtree [2509.26265]. In potential-outcome and outcome-regression formulations, the same object is expressed as a conditional mean difference [2310.16819] [2605.15373].

## 2. Identification and causal assumptions

The identification conditions attached to C-TACE in these papers are standard but explicit. The staged event tree framework assumes **consistency**, **positivity**, and **conditional exchangeability** [2509.26265]. It further states that, from a DAG perspective, conditional exchangeability is equivalent to assuming that the covariates \(\mathbf Z\) are sufficient to control all paths from \(R\) to \(Y\) except the direct link \(R \rightarrow Y\) [2509.26265]. The high-dimensional linear framework uses **unconfoundedness**,
\[
\{Y^1,Y^0\}\perp D \mid X,
\]
and **overlap**,
\[
\varphi < \mathbb{P}(D=1\mid X) < 1-\varphi \quad \text{a.s.},
\]
for some \(0<\varphi<1/2\) [2310.16819]. The nonparametric heterogeneity paper states that under conditional exchangeability and positivity, the CATE is identifiable from observed data [2605.15373].

A central technical implication is that C-TACE is a conditional causal parameter, not merely a conditional associational contrast. The identifying role of the conditioning covariates is therefore substantive: they must be rich enough to justify exchangeability, and their support must be broad enough to justify positivity. This becomes especially important in stratified or tree-based representations where some strata can be sparse or structurally deterministic.

The staged event tree paper emphasizes positivity in a particularly concrete way: “The staged event tree clearly illustrates a violation of the positivity assumption in the group of high-risk, referred patients, as all of them receive treatment” [2509.26265]. This makes visible a common misconception: conditional effect estimation is not guaranteed merely because covariates are observed. If a stratum assigns treatment deterministically, the conditional causal contrast is not identified there under the stated assumptions.

## 3. Staged event trees and localized conditional effects

In staged event trees, C-TACE is represented structurally rather than only as an abstract functional. The causal ordering is
\[
\mathbf X=(\mathbf Z,R,Y),
\]
with covariates first, then treatment, then outcome [2509.26265]. A staged event tree is an \(\mathbf X\)-compatible rooted tree in which stages encode equality constraints on conditional probabilities:
\[
P(X_i\mid \bm{X}_{[i-1]}=\bm{x}_{[i-1]})
=
P(X_i\mid \bm{X}_{[i-1]}=\bm{x}'_{[i-1]}),
\]
if the corresponding nodes are in the same stage [2509.26265]. Nodes in the same stage are colored the same, and same-colored nodes have equal conditional probabilities [2509.26265].

This representation gives C-TACE a localized interpretation. The paper states that “Conditional treatment effects are directly observable from the transition probabilities for the outcome variable” [2509.26265]. In other words, once a path through the covariates \(\mathbf Z\) is fixed, the relevant contrast is read from the treated and untreated outcome branches in the corresponding subtree. Context-specific independencies become visible because different covariate contexts can share stages or split into distinct stages, allowing the effect of \(R\) on \(Y\) to vary across \(\mathbf Z\)-contexts [2509.26265].

The interventional semantics are encoded directly in the tree. For intervened variables \(\mathbf X_I\) set to values \(\mathbf z_I\), the staged-tree causal model defines
\[
P\left(\bm{X} = \bm{x} \mid \operatorname{do}( \bm{X}_I = \bm{z}_I) \right) =
\left\{
\begin{matrix}
\frac{ P( \bm{X} = \bm{x} ) }{\prod_{i \in I} P(X_i = x_i \mid \bm{X}_{[i-1]} = \bm{x}_{[i-1]}) }  & \text{if } \bm{x}_I = \bm{z}_I \\
0   & \text{otherwise}
\end{matrix}
\right.
\]
[2509.26265]. For treatment-effect estimation, the paper introduces a **randomized staged event tree** \(T_r=(V,E,\eta_r)\), where treatment labels are replaced by a “naive” randomized assignment while all other conditional structure is retained. It states that this “is equivalent to standardization with respect to the levels of \(\mathbf Z\) ... also known as adjusting through subclassification” [2509.26265].

A second construction, the **ps-stratified staged event tree** \(T_{ps}\), uses the treatment stages themselves to induce strata that are then propagated to the outcome subtree [2509.26265]. The paper describes this as a propensity-score stratification analogue naturally encoded in the stages’ structure of the treatment variable. In both constructions, C-TACE remains a stratum-specific contrast in outcome probabilities, but the strata are obtained either from the covariate paths directly or from treatment-assignment similarity.

The same paper compares these staged-tree estimators with classical estimators including **AIPW**, **q.model**, **BN_tabu**, **BN_pc**, **full**, and **oracle** [2509.26265]. Its stated implementation for ATE is: learn stages from data, randomize \(R\) in the fitted tree, and compute \(P(Y \mid \operatorname{do}(R \sim \text{Bernulli}(0.5)))\). For conditional effects, it states that confidence intervals for the outcome probabilities given treatment values across different strata can be obtained using standard inferential methods for categorical data, and that bootstrap is a possible solution to post-selection inference problems [2509.26265]. This suggests a view of C-TACE in which interpretation and inferential uncertainty are both attached to visually identifiable local contexts.

## 4. High-dimensional linear models and implicit sparsity

A distinct line of work studies C-TACE in high-dimensional linear regression, again under the standard CATE formulation. The model has two treatments \(D\in\{1,0\}\), two potential outcomes \(Y^1,Y^0\in\mathbb{R}\), covariates \(X\in\mathbb{R}^p\), and observed outcome
\[
Y = D Y^1 + (1-D)Y^0
\]
[2310.16819]. The estimand is
\[
f_P(x)=\mathbb{E}_P[Y^1-Y^0\mid X=x].
\]
Under the paper’s linear potential-outcome model,
\[
Y^d = X^\top \beta^d + \varepsilon^d, \qquad \mathbb{E}[\varepsilon^d\mid X]=0,
\]
the conditional effect becomes
\[
f_0(x)=x^\top(\beta_0^1-\beta_0^0)
\]
[2310.16819].

The central structural assumption is **implicit sparsity**. The coefficients are decomposed as
\[
\beta_0^d=
\begin{pmatrix}
\alpha_0^d \\
\gamma_0
\end{pmatrix},
\qquad d\in\{1,0\},
\]
so that a common component \(\gamma_0\) is shared across treatment arms while treatment-specific components differ [2310.16819]. Equivalently, if \(X=(W^\top,Z^\top)^\top\),
\[
Y^d = W^\top \alpha_0^d + Z^\top \gamma_0 + \varepsilon^d,
\]
and therefore
\[
Y^1-Y^0 = W^\top(\alpha_0^1-\alpha_0^0) + (\varepsilon^1-\varepsilon^0).
\]
The common terms cancel in the contrast, leaving only the treatment-specific differences [2310.16819]. The paper emphasizes that sparsity is thus imposed on the CATE contrast rather than on either potential-outcome regression separately.

The proposed estimator penalizes the difference between the two treatment-specific coefficient vectors:
\[
(\widehat{\beta}^1,\widehat{\beta}^0) \in \arg\min_{\beta^1,\beta^0\in\mathbb{R}^p}
\left\{
\delta\frac{\|\widetilde{\mathbb{Y}^1}-\widetilde{\mathbf{X}^1}\beta^1\|_2^2}{n}
+
(1-\delta)\frac{\|\widetilde{\mathbb{Y}^0}-\widetilde{\mathbf{X}^0}\beta^0\|_2^2}{n}
+
2\lambda\|\beta^1-\beta^0\|_1
\right\}
\]
[2310.16819]. Rewriting with \(\beta=\beta^1-\beta^0\) yields a two-stage procedure: estimate \(\beta_0^0\) by an interpolating or minimum-norm least-squares estimator, then estimate the contrast by Lasso, and finally set
\[
\widehat{f}(x)=x^\top \widehat{\beta}
\]
[2310.16819].

The theoretical results are oracle inequalities and consistency for the contrast parameter rather than for \(\widehat{\beta}^1\) and \(\widehat{\beta}^0\) individually. In the fixed-design sub-Gaussian setting, the paper proves high-probability bounds of the form
\[
\|\widehat{\beta}-\beta_0\|_1 \le \frac{C_1\lambda s_0}{\phi_0^2},
\qquad
\frac{\|\widetilde{\mathbf{X}^1}(\widehat{\beta}-\beta_0)\|_2^2}{n} \le \frac{C_2\lambda^2 s_0}{\phi_0^2}
\]
[2310.16819]. Under random design with sub-Gaussian covariates and \(\lambda \asymp \sqrt{\frac{\log p}{n}}\), it further gives
\[
\|\widehat{\beta}-\beta_0\|_1 = O_P\!\left(s_0\sqrt{\frac{\log p}{n}}\right),
\qquad
\frac{\|\widetilde{\mathbf{X}^1}(\widehat{\beta}-\beta_0)\|_2^2}{n}
=
O_P\!\left(s_0\frac{\log p}{n}\right),
\]
implying \(\widehat{\beta}\xrightarrow{p}\beta_0\) [2310.16819]. A plausible implication is that this framework is especially tailored to settings in which effect modification is sparse even when the arm-specific outcome surfaces are not.

## 5. Potential-outcome prediction by tree boosting

C-TACE can also be estimated by directly predicting both potential outcomes and differencing them. C-XGBoost is a tree-boosting model built on XGBoost and adapted to causal inference by learning the pair of conditional outcomes
\[
Q(0,\mathbf{x}_i;\theta), \qquad Q(1,\mathbf{x}_i;\theta)
\]
for each instance \(\mathbf{x}_i\) [2404.00751]. The conditional effect estimate is then
\[
Q(1,\mathbf{x}_i;\theta)-Q(0,\mathbf{x}_i;\theta),
\]
which the paper describes as the standard form of a conditional average treatment effect estimate at \(\mathbf{x}_i\), often called CATE, ITE, or heterogeneous treatment effect depending on context [2404.00751].

The model uses a treatment-masked squared-error loss:
\[
\mathcal{L}(\mathbf{X};\theta) = \frac{1}{n} \sum_i \mathcal{L}_1(\mathbf{x}_i;\theta) + \mathcal{L}_2(\mathbf{x}_i;\theta),
\]
with
\[
\mathcal{L}_1(\mathbf{x}_i;\theta) = (1 - t_i)(Q(0,\mathbf{x}_i;\theta)-y_i)^2,
\qquad
\mathcal{L}_2(\mathbf{x}_i;\theta) = t_i(Q(1,\mathbf{x}_i;\theta)-y_i)^2
\]
[2404.00751]. Only the factual branch is penalized for each sample, while the model maintains both potential-outcome predictors. The paper states that “at each iteration the C-XGBoost builds a multi-output tree with the size of leaf equals to the number of targets” [2404.00751]. Reported implementation settings include \(\textsf{n\_estimator}=100\), \(\textsf{max\_depth}=5\), \(\textsf{reg\_lambda}=1.0\), \(\textsf{tree\_method}="hist"\), and \(\textsf{multi\_strategy}="multi\_output\_tree"\) [2404.00751].

Evaluation distinguishes average and heterogeneous performance. The ATE metric is
\[
\left\vert \epsilon_{ATE}\right\vert
=
\left\vert
\frac{1}{n}\sum_{i=1}^{n}(y(\mathbf{x}_i,1) - y(\mathbf{x}_i,0))
-
\frac{1}{n}\sum_{i=1}^{n}(\hat{y}(\mathbf{x}_i,1) - \hat{y}(\mathbf{x}_i,0))
\right\vert,
\]
and heterogeneous-effect accuracy is assessed by
\[
\textnormal{PEHE}
=
\frac{1}{n} \sum_{i=1}^{n}
\bigl(
(y(\mathbf{x}_i,1) - y(\mathbf{x}_i,0)) -  (\hat{y}(\mathbf{x}_i,1) - \hat{y}(\mathbf{x}_i,0))
\bigr)^{2}
\]
[2404.00751]. Since PEHE measures error in effect differences, it is directly tied to C-TACE quality.

The empirical analysis uses a **Synthetic** semi-synthetic collection from Louizos et al. with 5000 samples and 1000 covariates, and **ACIC** data from the 2018 Atlantic Causal Inference Conference benchmark [2404.00751]. Baselines are **R-Forest**, **C-Forest**, **Dragonnet**, and **\(k\)NN-Dragonnet**. Statistical validation uses **Dolan and Moré performance profiles**, the **Friedman Aligned-Ranks (FAR)** test, and the **Finner post-hoc** test [2404.00751]. On Synthetic datasets, C-XGBoost is reported as best on both \(|\epsilon_{ATE}|\) and PEHE, achieving the best score on **76%** of datasets for \(|\epsilon_{ATE}|\) and on **58%** for PEHE; on ACIC datasets it is best or tied-best on PEHE and achieves **62.5%** of datasets with the best PEHE score [2404.00751]. These findings support the interpretation of C-TACE as a prediction target recoverable from paired potential-outcome models, not only from explicit causal-function estimators.

## 6. Distributional summaries, uncertainty, and nonregular inference

A recurring difficulty with C-TACE is that a multivariate conditional effect surface can be hard to communicate when it depends on many covariates. One response is to summarize the distribution of the conditional effect in the population rather than to inspect the full function directly. The nonparametric heterogeneity paper defines the **sublevel-set probability curve**
\[
\gamma(P)(\alpha)=P(\tau(P)(W)\le \alpha), \qquad \alpha\in[-1,1],
\]
and describes it as the cumulative distribution function of the random variable \(\tau(P)(W)\) [2605.15373]. Each value \(\gamma(\alpha)\) is the proportion of individuals whose expected treatment effect does not exceed threshold \(\alpha\); in particular, \(\gamma(0)\) is the fraction of the population for which treatment is not expected to be beneficial [2605.15373].

This construction does not replace C-TACE; it summarizes its heterogeneity. If treatment effects are nearly constant, \(\gamma\) is almost a step function concentrated around the ATE; if effects vary widely, the curve spreads more smoothly [2605.15373]. The paper’s technical contribution is that pointwise evaluation \(\gamma(\alpha)\) is not pathwise differentiable under a nonparametric model, so standard regular asymptotically linear inference is not directly available [2605.15373].

To address this, the paper proposes three estimators. The first is a cross-fitted plug-in estimator,
\[
\hat{\gamma}_n^{\text{pl}}(\alpha)
=
\frac{1}{K}\sum_{k=1}^{K}\mathbb{P}_n^k[\hat{\eta}_{\alpha,n}^{-k}],
\qquad
\hat{\eta}_{\alpha,n}^{-k}(w)=1\{\hat{\tau}_n^{-k}(w)\le \alpha\},
\]
[2605.15373]. The second is a **Grenander-type estimator**, constructed by first working with the antiderivative
\[
\Gamma(P)(\alpha)=\int_{-1}^{\alpha}\gamma(P)(u)\,du,
\]
estimating it by a cross-fitted one-step procedure, and then differentiating the greatest convex minorant:
\[
\hat{\gamma}_n^{\text{Gr}}(\alpha)
=
\frac{\partial}{\partial \alpha}\mathrm{GCM}(\hat{\Gamma}_n^{\bullet})(\alpha)
\]
[2605.15373]. The theory yields cube-root asymptotics involving Chernoff’s distribution. The third is a debiased machine-learning estimator for a best piecewise linear approximation
\[
\gamma^{\#}(\alpha)=\sum_{l=0}^{L+1} H_l(\alpha)\gamma_l^{\#},
\]
which is pathwise differentiable under continuity at knot points and therefore amenable to standard debiased machine-learning inference [2605.15373].

Uncertainty quantification is also a major issue in staged event trees. The same literature states that “appropriate ATE and CATE estimators should be accompanied by standard error estimates” and notes that post-selection inference problems are inevitable in the proposed approach; it identifies bootstrap as a possible solution and also notes that standard inferential methods for categorical data can produce confidence intervals for outcome probabilities within strata, thereby enabling formal inference on CATE [2509.26265]. Taken together, these results show that C-TACE is not only an estimand of heterogeneity but also a locus of nontrivial inferential challenges: conditioning sharpens substantive interpretation while simultaneously complicating identification, support conditions, and asymptotic theory.

Source: https://www.emergentmind.com/topics/conditional-treatment-averaged-causal-effect-c-tace