CausalSmith · seminar slides
Inference for Minimum Dose Policies
Raising every treatment below a minimum dose creates a point mass that changes how accurately we can estimate the resulting mean outcome.
slides for Minimax Inference for Threshold Modified Treatment Policies with Continuous Treatments
Motivation
A practitioner wants to raise low treatment doses while leaving higher doses unchanged.
For natural treatment A, threshold δ, and assigned treatment Aδ, the policy is
dδ(a)=max{a,δ},Aδ=dδ(A).
Every dose below δ becomes exactly δ. The policy creates an atom: positive probability concentrated at one dose.
We allow a predetermined threshold δn to change with sample size n, within 0≤δn≤δˉ<1.
Research Question
Díaz et al. (2021) develop estimation and root-n inference for modified treatment policies under regularity conditions.
The usual approach averages predicted responses under the policy and uses regular influence-function inference.
For an exact clamp, that average assigns positive weight to a regression value at one dose. Continuous treatment data supply no observations exactly at that dose.
Gaïffas (2005) studies the related difficulty of estimating a regression value where treatment density becomes small.
How accurately can we estimate the exact clamp mean, and how short can uniformly valid confidence intervals be?
Key Idea
Separate ordinary mean estimation from learning the response at the threshold.
The threshold regression error matters only after multiplication by the probability moved to that threshold.
Under fixed finite strata, bounded outcomes, known regression smoothness, and polynomial treatment-density thinning, optimal absolute-error risk and honest interval length have scale
rn=n−1/2+δnκ+1hnβ.
Here β>0 controls regression smoothness, κ≥0 controls density thinning, and hn is the local fitting window.
- n−1/2: ordinary sampling error.
- δnκ+1: mass moved by the policy.
- hnβ: error in learning the threshold response.
A shrinking threshold can make the weighted regression error small enough for root-n accuracy.
Example
Take one stratum, density π(a)=2a, and smoothness β=κ=1. The mass moved to the threshold is δn2.
Compare Bernoulli outcomes with mean m0(a)=1/2 against a triangular regression bump centered at δn, with height hn.
When δn≫n−1/4,
hn≍(nδn)−1/3,Kn≍nδnhn3,Δn≍δn2hn.
Kn is the n-sample Kullback–Leibler divergence: expected log-likelihood evidence distinguishing the alternatives. Δn is their difference in clamp means. The symbol ≍ means bounded above and below by fixed positive multiples.
The window keeps that evidence bounded, while the clamp mean changes by atom mass times bump height.
Model and Target
Observe n independent units Oi=(Xi,Ai,Yi) from law P. Baseline stratum X has J fixed possible values; treatment A and outcome Y lie in [0,1].
Write px=P(X=x), πx(a) for the conditional treatment density, and μxP(a) for the continuous conditional outcome regression.
qx(δ)=∫[0,δ]πx(a)da,νδ(P)=EP[Y1{A>δ}].
qx(δ) is the collapsed probability within stratum x; νδ(P) is the retained outcome contribution.
θδ(P)=νδ(P)+x∈X∑pxqx(δ)μxP(δ).
The target adds outcomes retained above the threshold to collapsed mass multiplied by its threshold response. Here X is the stratum set.
Assumptions
Treatment thinning. For fixed known κ≥0 and positive envelope constants c−,c+,
c−aκ≤πx(a)≤c+aκ.
The lower bound guarantees nearby observations; the upper bound controls nearby information and collapsed mass. Every stratum has probability at least a fixed pmin>0.
Quantitative smoothness. Fix known β>0, radius L>0, and polynomial degree ℓ=⌈β⌉−1.
μxP(t)−j=0∑ℓj!(μxP)(j)(s)(t−s)j≤L∣t−s∣β.
This Hölder condition bounds the error of a Taylor approximation between doses s,t; j indexes derivatives.
For our example, it says ∣μ(t)−μ(s)∣≤L∣t−s∣: nearby means cannot change arbitrarily fast.
Call the laws satisfying these conditions M.
Main Result
informal · Theorem T-2 Under our fixed-stratum Hölder and thinning conditions, every deterministic threshold path in [0,δˉ] has minimax absolute-error risk bounded above and below by constants times rn, for all sufficiently large n.
Rn⋆ is the smallest worst-case expected absolute error among all estimators.
For fixed disjoint split blocks Bn, each containing at least ⌊n/4⌋ observations, our stabilized total Gram estimator attains the upper bound:
crn≤Rn⋆≤P∈MsupEP[θn,BnTG−θδn(P)]≤Crn.
The superscript TG labels the estimator constructed below; c,C are positive comparison constants depending only on fixed model parameters.
Root-n accuracy depends on whether the weighted threshold-regression error is small enough.
Honest Inference
An interval is honest when its coverage guarantee holds uniformly over every law in M.
informal · Theorem T-3 Under our model conditions, every deterministic threshold path in [0,δˉ] and admissible fixed split sequence eventually gives coverage at least 1−α and optimal worst-case expected interval length comparable to rn.
Fix noncoverage probability 0<α<1/2. Our interval CIn satisfies
inf{P⊗n{θδn(P)∈CIn(O1,…,On)}:P∈M}≥1−α,
where P⊗n is the law of the independent sample.
crn≤sup{EP⊗n[len(CIn)]:P∈M}≤Crn.
Here len means interval length. Every honest interval has worst-case expected length at least crn. Constants may also depend on α.
Three Estimation Tasks
Split the sample into fixed disjoint blocks I0,I1,I2, each with at least ⌊n/4⌋ observations, so the three tasks use independent data.
- I0: estimate the retained contribution by averaging Yi1{Ai>δn}; call this νn.
- I1: estimate each joint collapsed mass pxqx(δn) by its sample proportion; call this bx.
- I2: estimate the threshold response; call this μx.
Combine them exactly as the target requires:
θn=Π[0,1][νn+x∈X∑bxμx].
Π[0,1] clips a value to the bounded outcome range.
The difficult task is estimating μx from nearby doses.
Local Regression
Fit a degree-ℓ polynomial using doses in [δn,δn+hn], separately within each stratum, and read its intercept.
Rescale treatment so the threshold becomes zero:
Ui=hnAi−δn,vℓ(u)=(1,u,…,uℓ)⊤.
Ui is the local dose coordinate; vℓ collects polynomial terms.
On a sufficiently informative realized design, intercept weights wix satisfy
i∈I2∑wixUij=1{j=0}.
For powers j=0,…,ℓ, the weighted fit reproduces the constant and cancels every higher-order polynomial term at the threshold. Only the controlled Taylor remainder remains.
In our β=1 example, ℓ=0: this is a local average.
Bandwidth Choice
A wider window supplies more observations but allows more approximation bias.
- The local count has scale nhn(δn+hn)κ.
- Regression noise decreases with the square root of that count.
- Smoothness bounds approximation error on scale hnβ.
Choose hn to balance noise and approximation error. For all sufficiently large n,
0<hn,hn≤1−δˉ,nhn2β+1(δn+hn)κ=1.
The first two conditions keep the fitting window inside treatment support; the equality makes the two errors comparable.
Multiplying regression error by collapsed mass produces δnκ+1hnβ.
How does the window change as the threshold approaches zero?
Bandwidth Regimes
The bandwidth transition occurs at
δedge,n=n−1/(2β+κ+1).
This edge scale compares the threshold’s distance from zero with the window needed to learn its response.
| Threshold location | Window width | Why | |---|---|---| | δn/δedge,n→0 | hn has order n−1/(2β+κ+1) | The window dominates the threshold; density across the window has scale hnκ. | | δn/δedge,n→∞ | hn≍(nδnκ)−1/(2β+1) | The threshold dominates the window; density across it has scale δnκ. |
At thresholds comparable to the edge scale, the window also has order n−1/(2β+κ+1).
For β=κ=1, the edge is n−1/4; above it, our example uses hn≍(nδn)−1/3.
Design Stabilization
A polynomial fit can fail when its window is empty or dose locations provide too little information to invert the fitting matrix.
Let Nx be the local observation count and Gx the sum of local cross-products vℓ(Ui)vℓ(Ui)⊤.
Use the fit only on the event Ωx where Nx>0 and every coefficient vector v satisfies
v⊤Gxv≥2λ⋆Nx∥v∥22.
The known positive constant λ⋆ supplies a uniform minimum amount of information per observation.
μx={Π[0,1](∑i∈I2wixYi),1/2,Ωx,Ωxc.
Use the clipped intercept when the design passes; otherwise use the bounded midpoint.
The density envelope guarantees population information uniformly as the threshold moves. Failed checks become exponentially rare in the local information count.
Bias and Noise
On a passing design, polynomial reproduction removes the Taylor terms. Smoothness and bounded-outcome concentration give the regression radius
Bx=Lhnβi∈I2∑∣wix∣+tα(i∈I2∑wix2)1/2.
- The first term bounds approximation bias using total absolute weight.
- The second bounds noise using the square root of the sum of squared weights.
- tα=log(12J/α)/2 calibrates the concentration allowance.
The design check keeps total absolute weight bounded and squared-weight sum at most a constant divided by Nx.
Bias-aware intervals, as in Armstrong and Kolesár (2018), allow for approximation error explicitly rather than treating the fitted intercept as unbiased.
Interval Construction
Let b0=tα/∣I0∣ and b1=tα/∣I1∣ cover sampling errors in the retained mean and atom masses.
For each stratum, set Dx=(bx+b1)Bx+b1 on a passing design: multiply regression uncertainty by an upper bound on atom mass, then add mass-estimation uncertainty.
On a failed design, set Dx=bx+b1, allowing for the full bounded-response uncertainty.
CIn=[θn−b0−x∈X∑Dx,θn+b0+x∈X∑Dx]∩[0,1].
This interval covers retained-mean error, mass error, regression bias, regression noise, and failed fits.
The same atom weighting that reduces estimation error also reduces the necessary interval width.
Why the Rate Is Unavoidable
Two pairs of hard-to-distinguish laws isolate the two error sources.
A global Bernoulli regression shift of n−1/2 changes the clamp mean on that scale while leaving bounded statistical evidence in n observations.
A localized pair Q0,Q1 changes regression only near the threshold:
μxQ1(t)−μxQ0(t)=ahnβb(hnt−δn)
Here a is a fixed small amplitude, and b is a nonnegative bump with height one at zero and support [−1,1].
The bandwidth balance keeps evidence for this bump bounded. Its threshold height is multiplied by collapsed mass, giving target separation at least a constant times δnκ+1hnβ.
An estimator must accommodate these separations; an honest interval must be long enough to cover them.
Threshold Regimes
The bandwidth transition concerns nearby information. The risk transition also accounts for the mass weighting that information receives.
δcrit,n=n−1/{2(βκ+2β+κ+1)},δedge,n=n−1/(2β+κ+1).
The critical scale marks where weighted regression error matches ordinary sampling error. It is asymptotically larger than the edge scale.
| Threshold path | Optimal risk and honest length | |---|---| | δn/δcrit,n→0 | rn≍n−1/2 | | δn/δcrit,n→c0>0 | Both error terms are comparable; rn≍n−1/2 | | δn/δcrit,n→∞, δn→0 | rn≍δnκ+1(nδnκ)−β/(2β+1) | | δn→δ0>0 | rn≍n−β/(2β+1) | | δn=0 | Ordinary bounded-mean inference; rn=n−1/2 |
Here c0 is a positive limiting ratio and δ0 a positive limiting threshold.
Example Revisited
For π(a)=2a and β=κ=1, regression error has scale hn, but the policy weights it by δn2.
Above the bandwidth edge n−1/4, the window has order (nδn)−1/3. The risk transition occurs at the larger threshold scale:
δn≍n−1/10⟺Δn≍n−1/2.
Here Δn is the clamp-mean separation between our triangular-bump alternatives.
- Below the critical scale, shrinking atom mass permits root-n accuracy.
- Above it, learning the threshold response determines overall difficulty.
- At zero threshold, the atom disappears and the target is the ordinary outcome mean.
Causal Assumptions
Let PF be a law including potential outcomes Y(a), the outcomes a unit would have at each dose a. The causal policy mean is
ψδ(PF)=EPF[Y(dδ(A))].
It averages outcomes after applying the minimum-dose rule to natural treatment.
An unobserved variable U contains the unit’s response characteristics; gx(a,U) gives its response at dose a in stratum x.
Consistency: with probability one, the same equations hold simultaneously for every dose a:
Y(a)=gX(a,U),Y=gX(A,U).
Conditional independence: A⊥U∣X. Within each stratum, treatment carries no additional information about response characteristics.
Mean continuity: mxF(a)=EPF{gx(a,U)∣X=x} is continuous on [0,δˉ]. This is the stratum’s average structural response, including at the clamp threshold.
Further Results
- Causal interpretation: Under the three causal conditions just given and observed law P∈M, ψδ(PF)=θδ(P); the same estimator and interval attain matching causal risk and honest-length bounds of order rn.
- Continuity alone: Under the same design conditions, replacing quantitative smoothness by continuity gives matched risk and honest-length scale sn=n−1/2+δnκ+1; at a fixed positive threshold, neither minimax quantity converges to zero.
Open Questions
Can we obtain sharp limiting constants for optimal honest interval length, beyond matching its order?
Optimizing local weights against their exact worst-case regression bias gives a starting point.
A complete construction must also incorporate retained-mean uncertainty, atom masses and their uncertainty, aggregation across strata, and protection against failed local designs.
Takeaways
- An exact minimum-dose policy creates a point mass, making a single regression value part of an otherwise ordinary mean.
- The difficulty is local regression error multiplied by mass moved: rn=n−1/2+δnκ+1hnβ.
- Under known smoothness and polynomial thinning, stabilized local fitting and explicit bias allowances attain matching estimation and honest-inference bounds.
Appendix: Minimax Clamp Frontier
The stabilized estimator attains the best possible worst-case absolute-error order.
Fix J,β,κ,L,c−,c+,pmin,δˉ,α satisfying the regime conditions:
- (Regime.) J≥1, β>0, κ≥0, L>0, c−>0, c−≤κ+1≤c+, pmin>0, pmin≤1/J, δˉ∈(0,1), and α∈(0,1/2).
Then there exist constants 0<c<C and an amplitude a∈(0,1/4], depending only on the displayed regime constants, such that, for every deterministic threshold sequence (δn)n≥1 with δn∈[0,δˉ] for every n and every deterministic sequence Bn=(I0,n,I1,n,I2,n) of three pairwise-disjoint sample blocks with ∣Ij,n∣≥⌊n/4⌋ for j=0,1,2, for all sufficiently large n, with hn the information-balance bandwidth from Definition P-3 and rn=n−1/2+δnκ+1hnβ as in Definition P-4, the observed minimax absolute-error risk Rn⋆ over the clamp model M in Definition P-1 and the worst-case i.i.d. risk over M of the stabilized estimator using Bn satisfy crn≤Rn⋆≤P∈MsupEP[θn,BnTG−θδn(P)]≤Crn. Moreover, for all sufficiently large n, the lower bound is witnessed inside the same clamp model by:
- (Global shift.) Laws P0,P1∈M satisfying Assumption A-1, with the same (X,A)-design distribution and the same px and πx functions, having Bernoulli outcomes, and obeying μxP1(a)=μxP0(a)+n−1/2for every x and a∈[0,1], with well-posed product chi-squared divergence, P1⊗n≪P0⊗n,∫(dP0⊗ndP1⊗n−1)2dP0⊗n<∞, and cn−1/2≤∣θδn(P1)−θδn(P0)∣,χ2(P1⊗n,P0⊗n)≤C.
- (Localized perturbation.) Laws Q0,Q1∈M satisfying Assumption A-1, with the same (X,A)-design distribution and the same px and πx functions, having Bernoulli outcomes, and admitting a continuous bump b:R→R satisfying b(0)=1, b(u)≥0 for every u, b(u)=0 for u∈/[−1,1], and ∣b(u)∣≤1 for u∈[−1,1], with regression shift μxQ1(t)−μxQ0(t)=ahnβb(hnt−δn) for every x and t∈[0,1]. The product chi-squared divergence is well posed, Q1⊗n≪Q0⊗n,∫(dQ0⊗ndQ1⊗n−1)2dQ0⊗n<∞, and cδnκ+1hnβ≤∣θδn(Q1)−θδn(Q0)∣,χ2(Q1⊗n,Q0⊗n)≤C.
Appendix: Honest Length Frontier
Uniformly valid confidence intervals attain, and must pay, the same worst-case expected-length order.
There are constants c,C∈(0,∞), with c<C, for which the following holds.
- (Regime constants.) The number of strata J is a positive integer, β>0, κ≥0, L>0, c−>0, c−≤κ+1≤c+, pmin>0, pmin≤1/J, δˉ∈(0,1), and α∈(0,1/2).
- (Threshold path.) The deterministic thresholds satisfy 0≤δn≤δˉ for every n.
- (Sample splits.) For every n, Bn=(I0,n,I1,n,I2,n) is a deterministic three-way split of the sample indices with pairwise disjoint blocks and ∣Ij,n∣≥⌊n/4⌋ for j=0,1,2.
For all sufficiently large n, set hn to be the information-balance bandwidth for δn from Definition P-3, and set rn=n−1/2+δnκ+1hnβ. Then the stabilized interval CIn has worst-case coverage inf{P⊗n{θδn(P)∈CIn(O1,…,On)}:P∈M}≥1−α, and its worst-case expected length is comparable to rn: crn≤sup{EP⊗n[len(CIn)]:P∈M}≤Crn. Moreover, the minimax honest expected length satisfies crn≤Ln⋆, where Ln⋆ is the infimum, over observed-sample confidence procedures with uniform coverage at least 1−α over M, of their worst-case expected length. Equivalently, every observed-sample confidence procedure Cn with inf{P⊗n{θδn(P)∈Cn(O1,…,On)}:P∈M}≥1−α has worst-case expected length at least crn: crn≤sup{EP⊗n[len(Cn)]:P∈M}.
Appendix: Causal Frontier Lift
The observed estimator and interval attain the same frontier for the causal clamp mean under the declared causal conditions.
Let J∈N, and let β,κ,L,c−,c+,pmin,δˉ,α∈R. Suppose that
- (Regime constants.) The constants satisfy 0<J,β>0,κ≥0,L>0,c−>0,c−≤κ+1≤c+, pmin>0,pmin≤J1,0<δˉ<1,0<α<21.
- (Threshold sequence.) The deterministic thresholds (δn)n∈N satisfy 0≤δn≤δˉ for every n∈N.
- (Split blocks.) For each n∈N, Bn=(I0,n,I1,n,I2,n) is a deterministic three-way split with pairwise disjoint blocks and ⌊4n⌋≤∣I0,n∣,⌊4n⌋≤∣I1,n∣,⌊4n⌋≤∣I2,n∣.
Let hn, the information-balance bandwidth for δn, be as in Definition P-3, and let rn=n−1/2+δnκ+1hnβ be the frontier rate from Definition P-4. Then there exist constants c,C∈R, with 0<c<C, depending only on these regime constants, such that, for every threshold sequence and every deterministic split-block sequence satisfying the preceding conditions, all sufficiently large n satisfy the following conclusions. The total-Gram estimator θn in Definition P-5, formed with split Bn, order ℓ=⌈β⌉−1, bandwidth hn, and threshold δn, is observed-sample measurable. The corresponding bias-aware interval CIn in Definition P-6, formed with the same split, order, bandwidth, and threshold, is an observed-sample measurable interval. For the causal minimax criteria Rn,F⋆ and Ln,F⋆ in Definition P-9, crn≤Rn,F⋆≤PF∈MFsupE(PF)⊗nθn−ψδn(PF)≤Crn, PF∈MFinf(PF)⊗n{ψδn(PF)∈CIn}≥1−α, PF∈MFsupE(PF)⊗nlen(CIn)≤Crn. Moreover, in the extended nonnegative-real order, after embedding the nonnegative real bounds into R≥0∪{∞}, crn≤Ln,F⋆≤Crn.
Appendix: Continuity Frontier Characterization
Qualitative continuity gives a separate frontier governed by sampling error and collapsed mass.
Fix J∈N and constants κ,c−,c+,pmin,δˉ,α∈R. Suppose 0<J,0≤κ,0<c−,c−≤κ+1≤c+,0<pmin≤J1,0<δˉ<1,0<α<21. Then there exist constants c,C∈R with 0<c<C such that the following statements hold.
- (Uniform frontier bounds.) For every deterministic threshold sequence (δn)n≥1 with δn∈[0,δˉ] for all n, and for every deterministic sequence B∙=(Bn)n≥1 of admissible three-way split blocks, all sufficiently large n satisfy the following bounds. With sn=n−1/2+δnκ+1, and with Rn,cont⋆, Ln,cont⋆, Rn,cont,F⋆, and Ln,cont,F⋆ as in Definition P-16, csn≤Rn,cont⋆≤P∈McontsupEP⊗nθn,cont−θδn,cont(P)≤Csn, the interval CIn,cont has uniform coverage at least 1−α over Mcont, its worst-case expected length is at most Csn, and csn≤Ln,cont⋆. Moreover, Rn,cont,F⋆=Rn,cont⋆,Ln,cont,F⋆=Ln,cont⋆, and the same estimator and interval satisfy Rn,cont,F⋆≤PF∈McontFsupE(PobsF)⊗nθn,cont−ψδn(PF)≤Csn, with causal coverage at least 1−α uniformly over McontF and causal worst-case expected length at most Csn.
- (Elbow and null-threshold rates.) For the continuity-only rate in Definition P-12, sn(δn=n−1/(2(κ+1)))≍n−1/2,sn(0)≍n−1/2.
- (Fixed positive thresholds.) For every deterministic threshold sequence (δn)n≥1 with δn∈[0,δˉ] for all n, every δ0∈(0,δˉ], and δn→δ0, Rn,cont⋆→0,Ln,cont⋆→0.