CausalSmith · seminar slides
Treatment Effects when Assignment Scores Lose Their Links
Labels attached to trial outcomes let us measure lost information, choose how much to retain, and account for the remaining uncertainty.
Research Question
A randomized trial releases treatment A, outcome Y, and a label R=g(E) of each person’s assignment probability E. We know the score distribution H and label rule g.
- We want the population average treatment effect (ATE).
- Inverse-probability weighting needs each outcome’s own assignment probability: Horvitz and Thompson (1952).
- Knowing the distribution of weights does not recover their association with outcomes.
- Fan et al. (2014) gives sharp treatment-effect bounds for fixed combined-data distributions.
How much uncertainty remains, and how should we design the labels?
Key Idea
Within each treatment–label cell, we know the outcome distribution and the reciprocal-weight distribution. Only their association is missing.
Pairing their ranks gives the exact ATE interval. Maximizing its width over outcome distributions gives a score-only design criterion, DH(g).
informal · Theorem T-2 With outcomes in [0,1], fixed overlap, any score law H, and any measurable finite-label rule g, maximum sharp ATE width has an exact score-only formula, attained by half-zero, half-one outcomes in every positive-mass arm-label cell.
A positive score-density floor makes the best ambiguity with K labels of order K−1. For continuous positive densities, we also obtain the exact leading constant and an attaining allocation.
Example
Half the population has E=1/4; half has E=3/4. Everyone receives the same label. Observed outcomes are half zeros and half ones in each arm.
- Higher scores are more common among treated rows; control assignment reverses this tilt.
- Within either arm, ones can belong to larger or smaller reciprocal assignment weights.
- The attainable mean of either potential outcome ranges from 1/3 to 2/3.
Compatible populations therefore attain ATEs 1/3 and −1/3 under the same release.
The sharp ATE interval is [−1/3,1/3], with width 2/3, even with unlimited data.
Giving the two scores separate labels eliminates this ambiguity.
Model
Let Y0,Y1∈[0,1] be control and treatment potential outcomes. For population law P, the target is θ(P)=EP[Y1−Y0].
Randomization: conditional on the score, assignment does not select on potential outcomes.
P(A=1∣E,Y0,Y1)=Ealmost surely.
Consistency: the observed outcome comes from the assigned arm.
Y=AY1+(1−A)Y0almost surely.
Release: the known measurable rule assigns each score a finite label.
R=g(E)almost surely.
Fixed overlap: E∈[ε,1−ε], with 0<ε<1/2, bounds both reciprocal weights. The example uses ε=1/4.
The score law H is known for identification and label design.
The Weighting Bridge
For arm a∈{0,1} and label r, define:
p1(e)=e,p0(e)=1−e,Cr={e:g(e)=r},qar=∫Crpa(e)H(de).
Here pa(e) is the arm’s assignment probability, Cr is the label’s score set, and qar is the arm–label probability.
Randomization and consistency give EP[Ya]=EP[1{A=a}Y/pa(E)]=∑rqarEP[Y/pa(E)∣A=a,R=r].
The release supplies the cell outcome law Far. Known H supplies the assignment-weighted score law and reciprocal weight:
Gar(de)=qar1{e∈Cr}pa(e)H(de),Ear∼Gar,War=pa(Ear)1.
The missing object is an outcome–weight product expectation with both marginal distributions known.
Why Sorting Is Extremal
Consider two outcome values y2≥y1 and two reciprocal weights w2≥w1.
Matching larger outcomes with larger weights, rather than crossing the pairs, changes their product sum by (y1w1+y2w2)−(y1w2+y2w1)=(y2−y1)(w2−w1)≥0.
- To maximize the weighted mean, swap any crossed pairs into matching order.
- To minimize it, swap matching pairs into opposite order.
- Repeating these swaps sorts the pairings. Quantile integrals express the same argument for distributions, including distributions with atoms.
In the example, the upper mean pairs ones with the larger reciprocal weights; the lower mean pairs ones with the smaller weights.
Sharp Identified Interval
Let QF(u) denote distribution F’s value at percentile u.
Opposite ranks give the lower arm mean; matching ranks give the upper arm mean:
μa=r:qar>0∑qar∫01QFar(u)QWar(1−u)du,μa=r:qar>0∑qar∫01QFar(u)QWar(u)du.
Each integral is a cell’s weighted outcome mean. Multiplying by qar and summing implements the population weighting identity.
For a compatible released law Prel, the exact attainable ATE range is:
I(Prel;H,g)=[μ1−μ0,μ1−μ0].
The arm endpoints can coexist: choose the conditional distributions of Y1 and Y0 given each score separately, join them independently, and then randomize treatment with probability E.
This reproduces both observed arm distributions. In the example, means 2/3 and 1/3 coexist; reversing their pairings gives the other ATE endpoint.
Why Half Zeros and Ones
A cell’s contribution to the arm-mean range is qar∫01QFar(u)[QWar(u)−QWar(1−u)]du.
- Below the halfway rank, the bracket is nonpositive.
- Above halfway, it is nonnegative.
- Outcomes in [0,1] maximize this gap by taking zero below halfway and one above.
This is a valid outcome distribution: half zeros and half ones, exactly as in the example.
The resulting upper-half minus lower-half weight integral equals absolute deviation from a median. The maximal cell contribution is qarinfcE∣War−c∣=infc∫Cr∣1−cpa(e)∣H(de).
The assignment factor pa(e) cancels the reciprocal weight. This turns uncertainty about outcomes into a criterion involving only scores and labels.
Exact Release Uncertainty
For fixed H and g, DH(g) is the largest sharp ATE width across compatible outcome laws:
DH(g)=r∈RJ∑[c∈[1/(1−ε),1/ε]inf∫Cr∣1−ce∣H(de)+c∈[1/(1−ε),1/ε]inf∫Cr∣1−c(1−e)∣H(de)].
The sum runs over the J labels in RJ. Each arm chooses its own constant reciprocal weight c; the remaining absolute error measures information lost within that label.
The half-zero, half-one outcome law attains this width, including for atomic score laws and disconnected score sets.
informal · Theorem T-3 Under fixed overlap, any score law H, and a measurable finite-label rule g, every compatible release identifies the ATE exactly if and only if the label determines the score H-almost surely.
Equivalently, DH(g)=0 iff E=h(R) almost surely for a recovery map h. Separate labels for the example’s two scores make every term vanish.
Label Design
Let ΔK(H) be the smallest DH(g) achievable with at most K labels. Write ℓ=1−2ε for the score interval’s length.
A density floor f(e)≥mf>0 puts baseline score mass in every subinterval. Under this condition:
2(1−ε)Kmfℓ2≤ΔK(H).
For every score law, equal-width labels g satisfy:
ΔK(H)≤2ε(1−ε)Kℓ,DH(g)≤2ε(1−ε)Kℓ.
These bounds establish order K−1.
Each absolute-deviation term is distance from a score representative: ∣1−ce∣=c∣e−1/c∣, with control representative 1−1/c.
Even assigning every score to its nearest of K representatives leaves integrated distance at least ℓ2/(4K). The density floor and two arms give the lower bound; equal-width cells control distance to label midpoints.
Can disconnected cells improve the exact leading constant?
Arbitrary Cells Cannot Improve
Assume now that f is continuous and strictly positive. Each label has a treatment and a control score representative.
The lower-bound argument allows arbitrary measurable cells:
- Localization: at the K−1 cost scale, the score length far from either representative vanishes. Positivity makes distant scores costly.
- Local weights: on the retained pieces, continuity makes the two distance coefficients nearly constant. Their sum approaches w(e)=f(e)/[e(1−e)].
- Shape bound: for any measurable score set B of length ∣B∣ and any representative z, ∫B∣e−z∣de≥∣B∣2/4. The closest points to z minimize this integral; disconnected pieces cannot lower it.
- Shared budget: both arms use the same cells. Summing their local squared-length costs and applying Cauchy–Schwarz with at most K cells gives the square of total weighted length, divided by 4K.
Letting the localization neighborhoods shrink makes that weighted length approach ∫ε1−εw(e)de.
Thus every measurable design obeys the same leading lower bound that short interval cells can attain.
Optimal Label Allocation
For a short interval of width h near score e, the leading ambiguity cost is f(e)h2/[4e(1−e)].
More labels reduce local widths. Cauchy–Schwarz allocates labels per unit score length in proportion to f(e)/[e(1−e)].
informal · Theorem T-6 Under fixed overlap and a continuous, strictly positive score density f, interval labels attain the exact limiting value of KΔK(H), although optimization allows every measurable label rule.
K→∞limKΔK(H)=K→∞limKDH(gK)=41(∫ε1−εe(1−e)f(e)de)2.
Here gK is the attaining interval-label design. The formula gives the leading ambiguity per label budget.
Place boundaries at equal increments of cumulative weight ∫εef(t)/[t(1−t)]dt.
Regions with more score mass or larger reciprocal-assignment factors receive narrower cells.
Honest Interval Length
Observe n independent, identically distributed released trial rows, with H known.
An interval is uniformly honest if it covers the ATE with probability at least 1−α under every causal law satisfying our model: Imbens and Manski (2004).
Let Ln,K,α⋆(H) be the smallest worst-case expected length, jointly choosing labels and an honest interval.
informal · Theorem T-5 With known H, independent trial rows, fixed overlap, score density f≥mf>0, and 0<α<1/2, optimal worst-case expected honest interval length has order K−1+n−1/2.
cε,mf,α(K−1+n−1/2)≤Ln,K,α⋆(H)≤Cε,α(K−1+n−1/2).
The positive constants depend only on their subscripts, not on H,n,K.
Observationally identical endpoint populations force retained ambiguity; trial sampling adds uncertainty. A label budget of order n balances the two.
Midpoint Weighting
Use equal-width labels. Replace the missing score by its label midpoint zr=ε+K(r−21)(1−2ε).
Tn=n1i=1∑n{zRiAiYi−1−zRi(1−Ai)Yi},Rn,K=ε4nlog(4/α)+2Kε(1−ε)1−2ε.
The estimator Tn averages midpoint-weighted outcomes. The radius Rn,K pays separately for sampling variation and score substitution.
For observed data x, use:
Cn(x)=[clampATE(Tn(x)−Rn,K),clampATE(Tn(x)+Rn,K)]
The function clampATE keeps endpoints in [−1,1].
Accounting for midpoint substitution makes this interval uniformly honest and attains the upper length order.
Independent Score Log
Fix a measurable release g with J labels. Observe n independent trial rows and an independent sample of m scores from H.
Call lenI(Prel;H,g) the population sharp width. Let Rn,m,α⋆(g) minimize worst-case expected positive excess length beyond that width among uniformly honest intervals.
Under our model and 0<α<1/2:
ctr(α)n−1/2≤Rn,m,α⋆(g)≤Cε,J,α,g(n−1/2+m−1/2).
Trial uncertainty gives the lower bound at every sample size. Score-log uncertainty also remains unavoidable:
m→∞liminfn≥1infmRn,m,α⋆(g)≥clog(g,α)>0.
The constants depend only on the indicated fixed quantities, not on H,n,m.
informal · Theorem T-7 For fixed measurable finite-label g, fixed overlap, 0<α<1/2, and independent iid trial and score-log samples, minimax excess length has order n−1/2+m−1/2 when n/m→ρ∈(0,∞).
Projection Construction
Estimate each cell’s outcome measure and assignment-weighted score measure:
λar(B)=n1i=1∑n1{Ri=r,Ai=a,Yi∈B},σar(B)=m1j=1∑mpa(Ejlog)1{g(Ejlog)=r,Ejlog∈B}.
For a set B, λar counts outcomes; σar weights logged scores. Both retain cell mass.
Define closeness by total-mass difference plus integrated cumulative distribution function distance: integrate the absolute gap between the measures’ cumulative distribution functions. This controls cell frequencies and movement of values.
Keep candidate arrays with matching outcome–score masses within empirical radii rn=αn4J and rm=αm2J(2−2ε). Call this set Bn,m.
Compute each candidate’s rank-pairing interval I(b), then form:
Jn,m=[−1,1]∩clconvb∈Bn,m⋃I(b).
This is the smallest closed interval containing all candidate intervals, within [−1,1]. On the coverage event, it contains the population sharp interval.
Overlap bounds weight changes. Integrated distribution-function distance controls quantile movement; retaining masses avoids division by small cell probabilities. Consequently, projection adds only order n−1/2+m−1/2 expected excess length, including with atoms and flat quantiles.
Takeaways
- Rank pairings recover the exact ATE range. DH(g)=0 iff E=h(R) almost surely: the example’s separate-score labels eliminate ambiguity because they recover each score.
- Half-zero, half-one outcomes attain worst-case ambiguity. Optimal ambiguity has order K−1 under a density floor; continuous positive densities give an exact constant and square-root label allocation.
- Honest inference accounts for both missing links and sampling: known-score length has order K−1+n−1/2; an independent score log gives excess length of order n−1/2+m−1/2 at a fixed positive sample-size ratio.
Appendix: Exact Release Uncertainty
The score-only formula is the attained maximum sharp ATE width for any finite-label release.
Suppose that:
- (Overlap.) The overlap condition in Assumption A-1 holds.
- (Score law.) H is a probability law on E=[ε,1−ε].
- (Release.) g:E→RJ is measurable.
Write Cr={e:g(e)=r}, p1(e)=e, p0(e)=1−e, and qar=∫Crpa(e)H(de). Let Prel be the released law Prel=r∈RJ∑a∈{0,1}∑2qar(δ(r,a,0)+δ(r,a,1)). Thus its observed outcome law Far is Bernoulli(1/2) in every positive-mass arm-label cell. The worst-case ambiguity DH(g) of Definition P-6 satisfies DH(g)=r∈RJ∑[c∈[1/(1−ε),1/ε]inf∫Cr∣1−ce∣H(de)+c∈[1/(1−ε),1/ε]inf∫Cr∣1−c(1−e)∣H(de)]. For each a∈{0,1} and each cell with qar>0, let Ear have law pa(e)1Cr(e)H(de)/qar, and let War=1/pa(Ear). Define μa=r:qar>0∑qar∫01QFar(u)QWar(1−u)du,μa=r:qar>0∑qar∫01QFar(u)QWar(u)du. The sharp ATE interval for Prel has length DH(g). Moreover, there are two probability laws P−,P+∈M(H,g,Prel), as defined in Definition P-2, such that EP−[Y1−Y0]=μ1−μ0,EP+[Y1−Y0]=μ1−μ0,EP+[Y1−Y0]−EP−[Y1−Y0]=DH(g).
Appendix: Optimal Label Allocation
A continuous, strictly positive score density yields an exact leading ambiguity constant attained by interval labels.
Let E=[ε,1−ε] be the score interval, and let H be a probability law on E. Suppose:
- (Overlap.) The overlap condition in Assumption A-1 holds.
- (Density.) H(de)=f(e)de on E.
- (Regularity.) The density f is continuous and strictly positive on E.
Let ΔK(H) and DH(g) be the optimal K-label ambiguity and the worst-case ambiguity defined in Definition P-8, Definition P-6. For every positive integer K, there is a measurable release gK:E→{0,…,K−1} whose label cells are intervals, such that K→∞limKΔK(H)=K→∞limKDH(gK)=41(∫ε1−εe(1−e)f(e)de)2.
Appendix: Honest Interval Length
Equal-width labels and midpoint weighting attain the minimax honest-length order under a positive density floor.
Consider the following conditions:
- (Overlap.) The overlap parameter ε satisfies Assumption A-1.
- (Miscoverage.) 0<α<1/2.
- (Trial size.) n≥1.
- (Label budget.) K≥1.
There are constants Cε,α>0 and cε,α>0, independent of H,n,K, such that every score law H on E=[ε,1−ε] satisfies cε,αn−1/2≤Ln,K,α⋆(H), where the minimax expected length is defined in Definition P-10. For every mf>0, there is a constant cε,mf,α>0 such that every H∈Hε,mflb, as defined in Definition P-18, satisfies cε,mf,α(K−1+n−1/2)≤Ln,K,α⋆(H)≤Cε,α(K−1+n−1/2). An interval attaining the upper bound uses the equal-width release with labels r∈RK={1,…,K}: gK(e)=1+min{⌊1−2εK(e−ε)⌋,K−1},zr=ε+K(r−21)(1−2ε). For released trial rows (Ri,Ai,Yi)i=1n, define Tn=n1i=1∑n{zRiAiYi−1−zRi(1−Ai)Yi},Rn,K=ε4nlog(4/α)+2Kε(1−ε)1−2ε. With clampATE(t)=max{−1,min{1,t}}, the interval Cn(x)=[clampATE(Tn(x)−Rn,K),clampATE(Tn(x)+Rn,K)] satisfies (gK,Cn)∈Jn,K,α(H), with honesty defined in Definition P-9. For every P∈P(H,gK), as defined in Definition P-1, its expected length under the n-fold released-row law satisfies ∫∣Cn(x)∣dPrel⊗n(x)≤Cε,α(K−1+n−1/2).
Appendix: Independent Score Log
Projection attains the two-source excess-length upper order, with separate trial and score-log lower bounds.
Let E=[ε,1−ε], and let g:E→RJ be a measurable finite-label release. Suppose:
- (Overlap.) The overlap condition in Assumption A-1 holds.
- (Miscoverage.) 0<α<1/2.
- (Joint sampling.) For every n,m, a joint sampling law QP,Hn,m is specified for each admissible causal law P and score law H. Its trial marginal consists of n independent draws from the released law. Its score-log marginal and independence from the trial sample satisfy Assumption A-9, Assumption A-10.
Set Lα=1+(1−2α)2,Sg=r∈RJ, e1<e2<e3e1,e2,e3∈Crsup(e3−e2)2+(e3−e1)2+(e2−e1)2(e3−e2)(1−e1/e2), and ctr(α)=4(1−2α)logLα,clog(g,α)=23(1−2α)logLαSg. Then 0<Sg≤1. There is a constant Cε,J,α,g>0 such that, for every positive n,m, the minimax excess risk of Definition P-15 satisfies ctr(α)n−1/2≤Rn,m,α⋆(g)≤Cε,J,α,g(n−1/2+m−1/2). For each such n,m, an interval procedure Cn,m belongs to the honest class of Definition P-14, agrees pointwise as a closed interval with the projection interval of Definition P-17, and, for every admissible pair (P,H), satisfies EQP,Hn,m[max{0,length(Cn,m)−length(I(Prel;H,g))}]≤Cε,J,α,g(n−1/2+m−1/2). In the external-score direction, m→∞liminfn≥1infmRn,m,α⋆(g)≥clog(g,α)>0. For every ρ>0, the normalized risk Tn,m,α(g) of Definition P-16 obeys 0<max{1+ρctr(α),1+ρclog(g,α)ρ}≤n,m→∞n/m→ρliminfTn,m,α(g)≤n,m→∞n/m→ρlimsupTn,m,α(g)≤Cε,J,α,g.