Tutorials › Propensity Score Matching and Instrumental Variables › Choosing Between ATE, ATT, ATU, and Overlap Weights

Weighting and Stratification · Tutorial 208 of 699

Choosing Between ATE, ATT, ATU, and Overlap Weights

Learn how each weighting scheme changes the population an effect describes, and choose an estimand based on the scientific question and the available propensity-score overlap.

Intermediate 11 min read

What You'll Learn

  • Define ATE, ATT, ATU, and overlap effects by their target populations.
  • Derive treatment-arm weights from a common propensity-score tilting function.
  • Calculate target-population profile shares without applying the tilt twice.
  • Compare how the four estimands respond to treatment-effect heterogeneity.
  • Diagnose weight instability, poor overlap, and covariate imbalance.
  • Choose an estimand that matches the scientific question and available support.

One Treatment, Different Target Populations

A treatment-effect estimate is incomplete unless it states which population the effect describes. The average treatment effect (ATE) concerns the full population; the average treatment effect on the treated (ATT) concerns those who received treatment; the average treatment effect on the untreated (ATU) concerns those who did not; and the overlap effect concerns a population weighted toward people whose treatment status is comparatively uncertain. These are different causal questions, not merely alternative formulas for the same target.

Propensity-score weighting makes the distinction visible. Changing the weighting scheme changes the covariate distribution represented by the weighted sample, and therefore can change the effect when treatment effects vary across covariate profiles. The introductory material on inverse probability weighting and overlap weights develops the individual estimators; here the focus is how their targets and weights compare.

Definition: Let \(A\in\{0,1\}\) indicate treatment, let \(X\) denote measured pretreatment covariates, and let \(e(X)=P(A=1\mid X)\) be the propensity score. Let \(Y(1)\) and \(Y(0)\) be the potential outcomes under treatment and control, and define the individual causal effect as \(Y(1)-Y(0)\). The ATE is \(E[Y(1)-Y(0)]\); the ATT is \(E[Y(1)-Y(0)\mid A=1]\); and the ATU is \(E[Y(1)-Y(0)\mid A=0]\). The overlap effect is the average individual effect in the population tilted in proportion to \(e(X)\{1-e(X)\}\).

A Common Form for the Four Weighting Schemes

A useful way to compare these estimands is to define a nonnegative tilting function \(h(e)\). It determines how much each covariate profile contributes to the target population. The target distribution is the original covariate distribution reweighted in proportion to \(h\), and normalized to sum to one. If \(h\) depends on \(X\) only through \(e(X)\), the target share of a profile with original population share \(q\) is proportional to \(q h(e)\).

$$ f_h(x)=\frac{h\{e(x)\}f(x)}{E[h\{e(X)\}]} $$

The corresponding treatment-arm weights are \(h\{e(X)\}/e(X)\) for treated units and \(h\{e(X)\}/[1-e(X)]\) for control units. The reason is that, within a covariate profile, the treated fraction is \(e(X)\) and the control fraction is \(1-e(X)\). Multiplying each fraction by its arm-specific weight gives the same tilted contribution \(h\{e(X)\}\) in both arms.

$$ e(x)\frac{h\{e(x)\}}{e(x)} = \{1-e(x)\}\frac{h\{e(x)\}}{1-e(x)} = h\{e(x)\} $$

Thus, the two weighted arms represent the same target covariate distribution. Under the identification assumptions stated below, their weighted outcome means estimate \(E_h[Y(1)]\) and \(E_h[Y(0)]\), and their difference estimates the causal effect in that target population.

EstimandTilting function \(h(e)\)Treated weightControl weightTarget population
ATE11/e1/(1-e)Full population
ATTe1e/(1-e)Population with the covariate distribution of treated units
ATU1-e(1-e)/e1Population with the covariate distribution of untreated units
Overlap effecte(1-e)1-eePopulation emphasizing covariate profiles with treatment overlap

These formulas describe the target as well as the weights. For example, the ATT tilting function \(h(e)=e\) gives profiles more representation when they are more common among treated units. The ATU tilt \(1-e\) instead follows the untreated covariate distribution. The overlap tilt is largest at \(e=0.5\) and approaches zero as \(e\) approaches either boundary.

Assumptions for a Causal Interpretation

Weights alone do not establish that a weighted contrast is causal. For the target effect to be identified from observed outcomes, the standard conditions are needed:

  • Consistency: the observed outcome is \(Y=Y(A)\), so a person’s observed outcome equals the potential outcome under the treatment actually received. The treatment and outcome must be defined clearly enough for this statement to be meaningful.
  • Conditional exchangeability: \(\{Y(1),Y(0)\}\) is independent of \(A\) conditional on \(X\). In context, measured covariates must suffice to control confounding for the comparison being made.
  • Positivity on the target support: where \(h\{e(X)\}>0\), both treatment probabilities must be positive for the target effect to be identified by these arm comparisons. For the ATE, this requires \(0<e(X)<1\) throughout the population of interest. The ATT and ATU have their own target support, but still require outcome information from both arms wherever that target assigns positive mass.

In practice, the propensity score is estimated, not known. The formulas then yield estimated weights, and their usefulness depends on propensity-model adequacy and the observed covariate balance. Weighting cannot remove unmeasured confounding, and it cannot create data in covariate regions where one treatment arm is absent.

Worked Example: Comparing the Four Target Populations

Suppose a hypothetical population has three covariate profiles. Profile A makes up 30% of the population and has propensity score 0.20; profile B makes up 40% and has propensity score 0.50; profile C makes up 30% and has propensity score 0.90. Calculate each profile’s share of the four target populations.

For each target, multiply the original population share by the relevant tilting function, then divide by the sum of those products. For the ATE, \(h(e)=1\), so the target shares remain \(0.30, 0.40, 0.30\).

For the ATT, the unnormalized shares are \(0.30(0.20)=0.06\), \(0.40(0.50)=0.20\), and \(0.30(0.90)=0.27\). Their sum is \(0.53\), so the normalized shares are \(0.06/0.53=0.1132\), \(0.20/0.53=0.3774\), and \(0.27/0.53=0.5094\). Profile C is common in the treated target because its original population share is tilted by its high treatment probability.

For the ATU, the unnormalized shares are \(0.30(0.80)=0.24\), \(0.40(0.50)=0.20\), and \(0.30(0.10)=0.03\), summing to \(0.47\). The normalized shares are \(0.5106, 0.4255,\) and \(0.06383\). For overlap weighting, the products are \(0.30(0.20)(0.80)=0.048\), \(0.40(0.50)(0.50)=0.100\), and \(0.30(0.90)(0.10)=0.027\). Their sum is \(0.175\), giving target shares \(0.2743, 0.5714,\) and \(0.1543\).

The results show that these are distinct targets. The ATE preserves the original profile mix; the ATT concentrates on profile C; the ATU concentrates on profile A; and the overlap target gives greatest emphasis to profile B, where treatment and control are equally probable.

How the Arm Weights Produce Those Shares

For the same hypothetical profiles, the weights are obtained by substituting each propensity score into the arm-specific formulas. The table below shows one unit’s weight in each arm for each profile. These individual weights need not equal the target-population shares; it is the weighted distribution of each arm, after accounting for how many units of each profile occur in that arm, that represents the target.

ProfileATE treated / controlATT treated / controlATU treated / controlOverlap treated / control
A, e=0.205.000 / 1.2501.000 / 0.2504.000 / 1.0000.800 / 0.200
B, e=0.502.000 / 2.0001.000 / 1.0001.000 / 1.0000.500 / 0.500
C, e=0.901.111 / 10.001.000 / 9.0000.1111 / 1.0000.100 / 0.900

Worked Example: Verifying the ATT Target Share

Use profile C to verify how ATT weights recover its target share. In the original population, profile C has share \(0.30\), and \(e=0.90\). Its treated population share before weighting is \(0.30(0.90)=0.27\). The ATT treated weight is 1, so its weighted treated mass remains \(0.27\). The corresponding weighted masses for profiles A and B are \(0.06\) and \(0.20\). Thus the treated arm’s normalized weighted share for C is \(0.27/(0.06+0.20+0.27)=0.5094\), precisely the ATT target share.

The control arm gives the same result. Profile C’s control population share is \(0.30(0.10)=0.03\), and its ATT control weight is \(0.90/0.10=9\). Its weighted control mass is therefore \(0.03(9)=0.27\). The other control masses are \(0.30(0.80)(0.20/0.80)=0.06\) for A and \(0.40(0.50)(0.50/0.50)=0.20\) for B. Normalizing again gives \(0.27/0.53=0.5094\).

This calculation also explains an important bookkeeping rule: the target share is formed by multiplying the original population share by \(h(e)\), then normalizing. Once a target share has been calculated, it already incorporates the tilt. Multiplying that target share by \(h(e)\) again would incorrectly apply the tilt twice.

Effect Heterogeneity Can Change the Answer

When effects differ across profiles, different target populations generally have different average effects. This is not an inconsistency: each estimand answers a different question. If the effect is constant for everyone, the four population averages coincide, assuming they are identified. With heterogeneity, they need not.

Worked Example: Averaging Profile-Specific Effects

Suppose the true hypothetical treatment effects, in outcome units, are 2 for profile A, 6 for profile B, and 10 for profile C. Use the target shares calculated above to find the four target effects. This calculation describes population averages under the stated hypothetical profile effects; it is not a claim that a small observed sample would estimate them exactly.

The ATE is \(0.30(2)+0.40(6)+0.30(10)=6.000\) outcome units. The ATT is \(0.1132(2)+0.3774(6)+0.5094(10)\), which equals approximately \(7.585\) outcome units (using the unrounded target shares gives \(4.02/0.53=7.585\)). The treated target places more weight on profile C, where the effect is largest.

The ATU is \(0.5106(2)+0.4255(6)+0.06383(10)\), approximately \(4.213\) outcome units; using unrounded shares, this is \(1.98/0.47=4.213\). The overlap effect is \(0.2743(2)+0.5714(6)+0.1543(10)=5.520\) outcome units, equivalently \(\{0.048(2)+0.100(6)+0.027(10)\}/0.175=5.520\).

The estimates differ because the target profile shares differ, not because any one calculation is automatically preferable. A report should name the estimand and explain why its population is relevant to the scientific question.

Choosing and Diagnosing an Estimand

The choice should start with the question, then be checked against the data. If the aim is the effect for the full eligible population, ATE weighting is conceptually aligned, but can be unstable when some propensities are near zero or one. If the question concerns people who received treatment, ATT targets their covariate distribution; if it concerns those who remained untreated, ATU targets theirs. Overlap weighting instead describes a population with emphasis on profiles represented in both arms. It can reduce the influence of extreme propensities, but it does not estimate the full-population effect unless additional conditions make the targets coincide.

Useful diagnostics include examining propensity-score distributions by treatment arm, checking covariate balance after weighting, and assessing weight concentration. A common summary is the effective sample size, \(ESS=(\sum_i w_i)^2/\sum_i w_i^2\), calculated separately by arm where appropriate. A small effective sample size signals that a few observations carry much of the weighted information. For ATE weights, propensities near either boundary can create large weights; ATT control weights can be large when \(e\) is near one, and ATU treated weights can be large when \(e\) is near zero. Overlap weights are bounded between zero and one for both arms, but they deliberately downweight boundary profiles and change the target population.

A failure to balance measured covariates after weighting may indicate propensity-model misspecification, insufficient overlap, or an implementation error. Balance diagnostics do not prove exchangeability, which concerns potential outcomes and cannot be verified from observed data alone. Trimming or changing the target to improve stability may be reasonable in some applications, but it changes the population being described and must be made explicit.

Common Mistake

Pitfall: “I have calculated the ATT target share for profile C as 0.509, so I should multiply it by \(h(0.90)=0.90\) to find how much C matters.” This applies the tilt twice. The profile’s target share is already its share after tilting and normalization. To construct it, start with the original share \(0.30\), multiply by \(0.90\), and normalize: \(0.30(0.90)/0.53=0.5094\). Use the target share itself when averaging effects over the ATT target.

The main alternatives are not interchangeable shortcuts. ATE, ATT, ATU, and overlap weighting each define a target population; matching or outcome-model standardization may address related causal questions but also require explicit target definitions and their own assumptions. The practical comparison is therefore not “which weights are universally best?” but “which population is relevant, and can the available data support a credible comparison for it?”

Check Your Understanding

Answer each question, then compare your reasoning with the solutions below.

  1. For a profile with \(e=0.25\), what are its treated and control weights under ATT weighting?
  2. A profile has original population share 0.20 and propensity score 0.80. What is its unnormalized contribution to the ATU target?
  3. Why can the ATT and ATE effects differ even if both are estimated without bias for their stated targets?
  4. What does an effective sample size much smaller than the actual sample size suggest?

Solutions. (1) The ATT treated weight is 1, and the control weight is \(e/(1-e)=0.25/0.75=0.3333\). (2) For the ATU, \(h(e)=1-e\), so the unnormalized target contribution is \(0.20(1-0.80)=0.040\); it must be divided by the sum of such products over all profiles to obtain a normalized share. (3) The ATE averages effects over the full population, whereas the ATT averages over the treated population. If effects vary across covariate profiles and those populations have different profile shares, their true averages can differ. (4) A few observations may dominate the weighted estimate, reducing the amount of independent information represented by the weighted sample and making the result more sensitive to those observations.

Clinical Biostats