CausalSmith · seminar slides
Second-Order Risk in Multi-Arm Randomization
For fixed binary potential outcomes and a prespecified multi-arm contrast, we reduce the minimax design problem to a finite orbit game, identify the first-order constant, and pin down the universal n−4/3 second-order scale.
slides for Second-order Minimax Risk in Multi-arm Binary Randomization
Overview
- We study randomized experiments with fixed n, fixed K, binary potential outcomes, and a zero-sum contrast c.
- The object is worst-case design-based mean squared error, optimized jointly over randomization and estimation.
- The main reduction replaces labeled schedules by response-type counts.
- The first-order minimax constant is C0(c)=Lc2/4.
- A clipped shrinkage estimator improves the first-order envelope at scale n−4/3.
- Exact rational finite programs give computable upper and lower certificates.
informal · Theorem T-1 The labeled minimax game has exactly the same value as a finite game over response-type counts.
Motivation
- Multi-arm trials often target a prespecified comparison, such as one regimen versus an average of alternatives.
- In an ACTG 175-style four-regimen binary-endpoint setting, the finite population is the enrolled cohort and each unit has one binary potential outcome per regimen.
- The design question is how to randomize before outcomes are seen when the adversary may choose any complete binary schedule.
- Equal allocation is a natural default, but the contrast itself determines which arms matter for minimax risk.
- We ask: what is the best possible worst-case risk, and what randomization-estimator pair attains its first refinements?
Setup
- A response type records a unit's binary potential outcomes across all K arms.
- A complete schedule assigns one response type to each labeled unit.
- A zero-sum contrast c turns that schedule into the finite-population target τc, a treatment comparison.
- The minimax risk ρn(c) lets us choose both the assignment law and a clipped estimator.
- Clipping is to the feasible target range, so the estimator lives on the same scale as the contrast.
The unrestricted labeled-schedule minimax risk for contrast c is ρn(c)=D∈Δ(AKn),τinfz∈TKnmaxED[{τ((Ai,Yiobs)i=1n)−τc(z)}2].
Orbit Reduction
- Unit labels are arbitrary for the target and for worst-case design-based risk.
- Averaging any procedure over common relabelings produces an invariant procedure.
- An invariant procedure depends on the data through allocation counts and arm-success counts.
- The adversary's state becomes the response-type count vector m, not the full labeled schedule.
- This turns a large labeled problem into a finite decision problem.
Fix n∈N, K≥2, and a nonzero zero-sum contrast c in the labeled-schedule and finite-orbit games of Definition P-1, Definition P-3. Then:
- (Lossless symmetrization.) For every labeled procedure p, there are an invariant labeled procedure pˉ and an orbit procedure q such that pˉ is the common-unit-permutation average of p, and for every complete labeled schedule z∈TKn, Rn(pˉ;z)≤∣Sn∣1σ∈Sn∑Rn(p;σz). The orbit procedure q is dominated by this averaged invariant representative in the corresponding orbit risk.
- (Exact invariant correspondence.) Invariant labeled procedures are in bijection with orbit procedures (π,δ). Under this bijection, each invariant labeled procedure and its orbit representative realize the same allocation-count and observation-count rule, and for every z∈TKn, Rn(q;counts(z))=Rn(pˉ;z).
- (Target preservation.) If m=counts(z), with mt=#{i:ti=t}, then τc(z)=n1i∑a∈AK∑cati,a=n1t∑mta∈AK∑cata=τc(m).
- (Value equality.) The unrestricted labeled-schedule minimax risk equals the finite response-type orbit minimax value: ρn(c)=Gn(c).
- (Orbit saddle point.) There exist an orbit procedure q⋆ and a prior ν on Mn,K such that, for every response-count vector m and every orbit procedure q, Rn(q⋆;m)≤Gn(c)andGn(c)≤Eν[Rn(q;m)].
Multi-arm Scope
- For K=2 and c=(1,−1), the orbit game recovers the two-arm binary target through the counts of positive and negative effect classes.
- For every fixed K≥3, the same orbit equality holds with 2K response types.
- This gives one finite minimax representation for binary treatment contrasts with any fixed number of arms.
informal · Theorem T-2 The two-arm benchmark and the fixed-K multi-arm orbit game live in the same exact response-type framework.
Related Literature
- Fisher (1935), Splawa-Neyman (1990), Rubin (1974), and Holland (1986) anchor the design-based potential-outcomes view.
- Horvitz and Thompson (1952), Hansen et al. (1953), and Särndal et al. (1992) provide the finite-population sampling language.
- Kallus (2018, 2020), Bai (2023), and Aronow and Lopatto (2026) give close minimax design and sampling comparisons.
- Sudijono et al. (2026) provide the sharp two-arm binary calibration.
- Our contribution is the exact multi-arm binary orbit game, its first-order constant, the n−4/3 second-order scale, and finite-program certificates.
First-order Result
- Let Lc be the ℓ1 norm of the contrast.
- Allocate independently on active arms with probabilities proportional to ∣ca∣.
- Estimate the centered contrast with a projected contrast-weighted Horvitz--Thompson rule.
- This procedure attains the finite-sample first-order envelope C0(c)/n.
- The asymptotic first-order constant is C0(c)=Lc2/4.
Let Sc={a∈AK:ca=0},Lc=a∈AK∑∣ca∣,qa⋆=Lc∣ca∣(a∈Sc). Assign each Ai independently on Sc with Pr(Ai=a)=qa⋆. The projected contrast-weighted Horvitz--Thompson rule is τcHT⋆=proj[−Lc/2,Lc/2][n−1i=1∑na∈Sc∑ca1{Ai=a}qa⋆Yiobs−1/2].
Fix a natural number K and a nonzero zero-sum contrast c on the treatment-arm set AK, in the labeled-schedule game of Definition P-1. Let Lc, qa⋆, and τcHT⋆ be the contrast norm, contrast-weighted assignment probabilities, and projected contrast-weighted Horvitz--Thompson rule from Definition P-4, and set C0(c)=4Lc2. Then the unrestricted labeled-schedule minimax risk satisfies n→∞limnρn(c)=C0(c). Moreover, for every n≥1, the independent contrast-weighted design together with τcHT⋆ attains the finite-sample worst-case bound z∈TKnmaxRn(D⋆,τcHT⋆;z)≤nC0(c).
Key Idea
- The contrast-weighted rule equalizes the worst-case contribution of the active arms at first order.
- The naive projected Horvitz--Thompson estimator is centered correctly, but it leaves room near the hardest response-type boundary.
- The shrinkage rule keeps the same assignment design and modifies only the estimator.
- It shrinks the normalized centered score toward zero inside a bandwidth of order n−1/3.
- That local correction produces a risk improvement at order n−4/3.
Second-order Result
informal · Theorem T-7 For every fixed nonzero zero-sum contrast, clipped shrinkage improves the first-order envelope by at least a positive multiple of n−4/3 once n passes a contrast-dependent threshold, and the improvement is at most a constant multiple of n−4/3.
For every fixed K≥2 and every nonzero zero-sum contrast c, let λc, the smallest nonzero normalized response-type contrast magnitude, be λc=min{Lc/2∑a∈AKcata:t∈TK, a∈AK∑cata=0}. Define C0(c)=4Lc2,Ac=2λc−16λc,Bc=167λc,κc=21min{Ac,Bc}, and dn(c)=nC0(c)−ρn(c),sn=n4/3. The following assertions hold:
- (Spacing.) The constants satisfy 0<λc≤1,0<κc.
- (First-order envelope.) For every n, ρn(c)≤nC0(c).
- (Uniform shrinkage bound.) There is a finite integer Nc such that, for every integer m≥Nc, 4λcmexp{−m1/3/8}≤2λc−16λc, and, for every n≥Nc, the explicit clipped-shrinkage procedure τnsh satisfies z∈TKnsupRn(D⋆,τnsh;z)≤C0(c){n−1−κcn−4/3}.
- (Second-order envelope.) The normalized second-order improvement satisfies C0(c)κc≤n→∞liminfn4/3dn(c)≤n→∞limsupn4/3dn(c)≤43C0(c).
- (Regular-variation index.) For every positive sequence an, every β, and every C>0, if an is regularly varying with index β and andn(c)→C, then β=4/3.
- The exponent 4/3 is universal for fixed K and fixed contrast.
- In the ACTG-style running example, a regimen-versus-average contrast falls under this multi-arm conclusion when at least three arms enter the contrast.
Converse
- The lower side embeds a two-arm hard subproblem inside any fixed contrast.
- The comparison scales the two-arm minimax risk by C0(c).
- For contrasts with exactly two active arms, the comparison becomes exact.
- A coarse two-arm information bound supplies the n−4/3 converse scale.
informal · Theorem T-6 Every fixed contrast inherits the two-arm lower difficulty up to the factor C0(c), with exact value transfer for two active arms.
informal · Theorem T-14 The two-arm binary minimax risk is at least the first-order value minus 43n−4/3.
Proof Sketch
- First, symmetrization removes unit labels and preserves the target.
- Second, the first-order upper bound follows from contrast-weighted inverse-probability scores and projection to the target interval.
- Third, shrinkage separates schedules into a central region and a separated region.
- In the central region, shrinking the normalized score reduces variance enough to dominate the introduced bias.
- In the separated region, the response-type contrast spacing keeps the shrinkage loss controlled.
- The two-arm lower bound uses a scalar Bayesian information inequality after reducing schedules to effect-class counts.
informal · Lemma L-1 A two-arm prior over effect-class counts induces a complete-schedule prior whose Bayes risk is captured by a scalar binomial experiment.
informal · Lemma L-2 The scalar Bayes risk is bounded below by a prior- and likelihood-information ratio.
Finite Certificates
- The orbit game is finite, so rational contrasts admit finite linear programs.
- The primal side chooses allocation-count masses and grid-valued estimator actions.
- The dual side yields response-count multipliers, interpreted as a least-favorable prior certificate.
- Replacing grid actions by their conditional barycenter gives an invariant estimator.
- The upper and lower endpoints differ by a mesh term.
For K≥2, n,M≥1, and a rational nonzero zero-sum contrast c∈QK, define the rational contrast grid-action value Λn,M(c) as the optimal value of the following finite linear program. Let hc=Lc/2,ΓM,c={−hc+jhc/M:j=0,…,2M}. The program minimizes u over variables πr≥0 and wr,x,g≥0, subject to r∈Rn,K∑πr=1, g∈ΓM,c∑wr,x,g=πrfor every (r,x), and r∈Rn,K∑x∑g∈ΓM,c∑Pm(x∣r)wr,x,g{g−τc(m)}2≤ufor every m∈Mn,K.
informal · Theorem T-8 For every rational contrast, exact rational primal-dual certificates bracket ρn(c) with mesh gap at most C0(c)/(4M2).
Real Contrasts
- Rational certificates transfer to real contrasts through square-root risk continuity.
- The distance is half the ℓ1 distance between contrasts, matching the target range.
- Rational approximants can be chosen close enough that the transfer loss is negligible at the n4/3 scale.
- For an exactly rational contrast, the transferred bracket has width at most C0(c)/(4n2) when M=n.
informal · Theorem T-9 Root minimax risk is Lipschitz in the contrast under half-ℓ1 distance.
Fix K≥2 and a nonzero zero-sum real contrast c on AK. For real contrasts c1,c2, write η(c1,c2)=21a∈AK∑∣c1(a)−c2(a)∣,C0(c1)=4Lc12. A tuple (B,U,R−,R+) is a transferred certificate for (n,M,c,q), where q∈QK∖{0} is zero-sum, if there exist an invariant allocation-count distribution π, a rational grid-action weight w, a rational epigraph value u, a rational response-count prior ν, an invariant barycenter rule δ, a projected upper procedure, and a schedule prior such that the exact rational primal-dual grid certificate and barycenter identities hold for q,π,w,u,ν,δ, and B=Bn,q(ν),U=Un,M(q;π,δ),u=Λn,M(q), B≤ρn(q)≤U≤u≤B+4M2C0(q), R−={max(0,B−η(c,q))}2,R+=(U+η(c,q))2, R−≤ρn(c)≤R+, and R+−R−≤4M25C0(q)+4C0(q)η(c,q)n−1/2+3η(c,q)2. The projected upper procedure certifies worst-case risk at most R+, and the schedule prior certifies Bayes risk at least R−. Then the following hold.
- (Finite-grid transfer.) For every rational zero-sum contrast q∈QK∖{0} and every pair of integers n,M≥1, there exist B,U,R−,R+∈R forming a transferred certificate for (n,M,c,q).
- (Real-contrast approximation.) There exist rational zero-sum contrasts c(n)∈QK∖{0} and real sequences Bn,Un,Rn−,Rn+ such that, for every arm a∈AK, c(n)(a)⟶c(a), n5/6η(c,c(n))⟶0, for every n≥1, (Bn,Un,Rn−,Rn+) is a transferred certificate for (n,n,c,c(n)), and n4/3(Rn+−Rn−)⟶0.
- (Exact rational case.) If a rational zero-sum contrast q∈QK∖{0} embeds coefficientwise as c, then for every n≥1 there exist B,U,R−,R+∈R forming a transferred certificate for (n,n,c,q) and satisfying U−B≤4n2C0(c).
Three-arm Diagnostic
- The diagnostic contrast is c†=(1,−1/2,−1/2).
- A scalar signed score keeps the target direction but collapses arm-label information.
- At n=3, an estimator reading the full observed arm labels and outcomes attains a strictly smaller worst-case risk than any rule based on the scalar score.
- This shows why the orbit game keeps arm-success counts, rather than only a one-dimensional signed statistic.
For K=3, let c†=(1,−1/2,−1/2), and let D⋆ be the independent design assigning each unit with probabilities q⋆=(1/2,1/4,1/4). For any n≥1, any schedule z∈T3n, and any unit i, define Zi=sign(cAi†)(2Yiobs−1),μi=ti,1−2ti,2+ti,3. Then the following statements hold.
- (Score moments.) For every n≥1, every z∈T3n, and every unit i, ED⋆(Zi∣z)=μi,VarD⋆(Zi∣z)=1−μi2.
- (Average-score risk.) For every n≥1 and every z∈T3n, ED⋆{n1i∑Zi−τc†(z)}2z=n1−n21i∑μi2.
- (Scalar minimax value.) For the scalar experiment at n=3 that retains (Z1,Z2,Z3) and lets each μi range over the five feasible values, its minimax value is V3Z=1−23.
- (Full-data certificate.) There exists a clipped full-data estimator using the observed arm labels and outcomes under D⋆ whose worst-case risk over z∈T33 is 4000000511653. Moreover, 4000000511653<13600018213<1−23.
- (Scalar moment envelope.) For every n, every real M, and every feasible three-arm scalar total M=∣∑iμi∣, the minimum attainable value of ∑iμi2 among scalar mean vectors with ∣∑iμi∣=M is {M/2,(3M−n)/2,M≤n/2,M>n/2.
Thus the scalar signed-score experiment has exact value V3Z at n=3, while the full observed arm-label-and-outcome experiment under the same independent q⋆ design attains the smaller certified worst-case risk displayed above.
Additional Results
- Four further statements package the same ingredients for specific supports and for the three-arm example.
informal · Theorem T-4 The three-arm diagnostic LP gives rational upper and lower certificates, with grid error negligible at the second-order scale when M=n2.
informal · Theorem T-11 The three-arm grid program has exact rational primal-dual certificates and brackets ρn† within 1/(4M2).
informal · Theorem T-12 For contrasts with at least three active arms, normalized improvements have positive bounded subsequential limits and are approximated by rational certificate clusters.
informal · Theorem T-13 The support-two value transfers exactly from the two-arm game, while shrinkage and finite brackets apply to every fixed nonzero zero-sum contrast.
Takeaways
- The exact orbit reduction is the computational and conceptual spine: labels vanish, response-type counts remain.
- The first-order benchmark is contrast-weighted allocation with constant C0(c)=Lc2/4.
- The same allocation, paired with clipped shrinkage, improves worst-case risk on the universal n−4/3 scale.
- Finite rational programs provide procedure and prior certificates for fixed-n minimax risk.
- For multi-arm binary trials with prespecified contrasts, the contrast support determines whether the exact two-arm transfer applies or only the universal order sandwich with its finite brackets.