Pigeonhole Design: Balancing Sequential Experiments from an Online Matching Perspective 3: For p ≥ 2 Continuous Covariates, Expected Discrepancy O(T^(1−φ) + T^((p−1)φ)), with (log T)^(3/2) at p = 2Research Paper
Motivation
Online platforms run randomized experiments on subjects (users, sessions, orders) that arrive one at a time and must be assigned to a control or a treatment group on arrival, before later subjects are seen. The precision of a treatment-effect estimate depends on how similar the two groups are in their observed covariates. When the groups are badly balanced, the estimator's variance grows, and a large literature on covariate balancing (blocking, rerandomization, matched pairs) addresses this for offline experiments, where all covariates are known in advance.
J. Zhao and Z. Zhou (arXiv:2201.12936, Management Science, 2025) cast the online problem as an online matching problem. The cost of an assignment is the discrepancy, the weight of a minimum-cost perfect matching between the two groups, and they compare three designs against an oblivious adversary that fixes the covariate sequence in advance:
- the clairvoyant matched-pair design;
- the completely randomized design;
- their online pigeonhole design.
For one continuous covariate the pigeonhole design attains against for complete randomization. This mission concerns the multi-dimensional result, Theorem 9.
Setting
subjects arrive in order; is even. Subject carries a covariate vector with discrete coordinates, the -th in a finite set of size , and continuous coordinates in . Distances are Euclidean in . An assignment puts each subject in control () or treatment ().
The discrepancy of a split with is
the value of the assignment problem (1) of the paper.
The completely randomized design picks uniformly among the subsets of size ; is the mean of .
A pigeonhole design fixes a partition of the covariate space into pigeonholes. When subject arrives:
- if one group already has subjects, joins the other group;
- otherwise joins the group that has fewer subjects in its own pigeonhole, and on a tie a fair coin decides.
Theorem 9 uses the pigeonholes
uniform cells of edge on each continuous axis, for parameters and . The expected discrepancy is the mean of over the design's coins.
Formalization targets
Goal: Theorem 9 (p. 26)
There are and , depending only on , such that for every even and every admissible arrival sequence,
Milestones
- Lemma EC.1: Hoeffding for a binomial, .
- Lemma 5, : if and the subjects occupy each cell of the grid exactly once, then .
- Lemma 5, : on the full grid of cells, .
- Balance: every run of a pigeonhole design has .
- Component 1: the subjects paired within their pigeonhole before the stopping time contribute at most , on every run.
- Unmatched subjects: at most one unmatched subject per pigeonhole, at most the number of pigeonholes, and .
- Component 2: the remaining subjects contribute in expectation.
Corollary 2 (the case ) is an extra theorem after the goal.
Significance
At Theorem 9 gives for . This is the order of the matched-pair design (Theorem 7), which sees all covariates in advance. An online design that commits to each assignment on arrival therefore loses at most a constant factor in dimension , and at most a factor at . For the completely randomized design no comparable upper bound is known for general sequences (p. 21). Lemma 5 is the paper's bound for complete randomization on the special family of full grids, and is of independent interest as a random-labelling matching estimate.
The results are proved on paper; none of them has a machine-checked proof, and the platform holds no item on them. A formalization would close the gaps of the printed argument:
- Lemma 5's induction for ends at rather than the stated .
- Component 2 applies Lemma 5 to unmatched subjects, which occupy distinct cells but do not fill the grid.
- The case is argued per discrete support, although the balancing rule acts across supports.
Every statement in the mission is believed true.
Difficulty
The matched part (Component 1) is deterministic: paired subjects share a pigeonhole of diameter . The difficulty lies in the subjects left over: the unmatched ones and the subjects after . Their labels come from coins, but which subjects end up unmatched, and when occurs, depend on the same coins. So the leftover set is not a fixed set with independent fair labels, and the conditional binomial law used on p. ec22 does not hold literally.
A second obstacle is the random-matching estimate itself. Bounding the expected optimal matching between randomly labelled points of a grid at the rates () and () is a multi-scale problem of Ajtai–Komlós–Tusnády type. A single-scale count of imbalances per cell gives only times a power of that is too large.
Formalization scope
- Subjects are indexed ; an assignment is a map to
Bool(false= control). The covariate space isEuclideanSpace ℝ (Fin q ⊕ Fin p). Admissible sequences have discrete coordinates in and continuous ones in , with (the paper's normalisation). - Discrepancy: an infimum over bijections between the groups. An unbalanced split has the junk value . The designs, the matched set and its complement are balanced on every run, so the junk is never reached.
- Cells: the printed edge gives a partition only when is an integer. The formalization uses cells per axis of edge , with the last cell closed. A pigeonhole is keyed by the exact discrete values and the cell index on each axis. The partition is the printed one, not a quantified family.
- Randomness: the pigeonhole design uses one coin per subject, consulted only on a tie. Its expected discrepancy is the average over all coin vectors, which is exactly the paper's law. The completely randomized design averages over all control sets.
- Asymptotics: means even, admissible. and are chosen before and and depend only on the fixed parameters. is the natural logarithm.
- Lemma 5's family : subjects in , one per cell, stated as injectivity of the cell map.
- Stopping objects: is the first at which a group holds subjects. Within a pigeonhole the subjects before are paired in arrival order, and the last one is unmatched when their number is odd.
- Corrections to the page: the second cell printed as on p. ec21 is read as . The family's is read as . The bound becomes under the ceiling and the discrete supports.
The following formalizations would trivialize the mission and are ruled out:
- a bound on the discrepancy of one run, or of the best run, instead of the average over all coin vectors;
- constants that depend on or ;
- an unbalanced design that exploits the junk value ;
- an arbitrary partition in place of the printed grid.
A complete development needs a binomial Hoeffding bound and an exact analysis of the design's recursion (balance, alternation within a pigeonhole before ). It also needs a multi-scale bound on random bichromatic matchings of grid points. That bound is reusable for Lemma 5 and for matching estimators generally. Contributions on any milestone are welcome, as are proofs of Lemma 5 for arbitrary subsets of grid points, which is what Component 2 needs.
Selected references
- J. Zhao and Z. Zhou, Pigeonhole Design: Balancing Sequential Experiments from an Online Matching Perspective, Management Science, 2025; arXiv:2201.12936v6, 2024. https://arxiv.org/abs/2201.12936
- M. Ajtai, J. Komlós and G. Tusnády, On optimal matchings, Combinatorica 4 (1984) 259–264. https://doi.org/10.1007/BF02579135
- N. Fournier and A. Guillin, On the rate of convergence in Wasserstein distance of the empirical measure, Probability Theory and Related Fields 162 (2015) 707–738. https://doi.org/10.1007/s00440-014-0583-7
- W. Hoeffding, Probability inequalities for sums of bounded random variables, Journal of the American Statistical Association 58 (1963) 13–30. https://doi.org/10.1080/01621459.1963.10500830