IDEAS home Printed from https://ideas.repec.org/p/arx/papers/2608.10177.html

Graph-Laplacian Variance Estimators for Finely Stratified Experiments

Author

Listed:
  • Yuehao Bai
  • Xun Huang
  • Joseph P. Romano
  • Azeem M. Shaikh
  • Max Tabord-Meehan

Abstract

This paper considers design-based inference on the average treatment effect in finely stratified experiments, where uncertainty arises only from the randomized treatment assignment. We focus on settings in which units are first stratified into groups of fixed size according to baseline covariates and, then within each group, exactly one unit is assigned to treatment. In this setting, we introduce a class of graph-Laplacian variance estimators in which strata form the vertices of a weighted graph and edge weights determine how between-stratum comparisons are aggregated. The canonical estimator of Imai (2008) corresponds to a complete graph with edge weights normalized so that each stratum has weighted degree one, while a paired-stratum estimator arises from a perfect matching graph. For the subclass of degree-calibrated graphs, in which each vertex has weighted degree one, we derive an exact bias identity showing that the corresponding estimators are upward-biased, with bias governed by squared differences in the true stratum-level treatment effects across adjacent strata. As a result, any such estimator may be used for valid inference. The identity further suggests that paired-stratum estimators constructed from a covariate-based perfect matching can induce small biases when treatment effects vary smoothly with the covariates. Without such smoothness, however, we show that paired-stratum estimators can exhibit large worst-case bias, and that, within the class of degree-calibrated estimators, the complete-graph estimator is minimax optimal for normalized bias under a weak bound on treatment-effect heterogeneity. Motivated by this contrast, we propose a regularized graph estimator that controls worst-case normalized bias while preserving much of the locality of the paired-stratum estimator. Simulations illustrate the resulting tradeoff between locality and worst-case protection.

Suggested Citation

  • Yuehao Bai & Xun Huang & Joseph P. Romano & Azeem M. Shaikh & Max Tabord-Meehan, 2026. "Graph-Laplacian Variance Estimators for Finely Stratified Experiments," Papers 2608.10177, arXiv.org.
  • Handle: RePEc:arx:papers:2608.10177
    as

    Download full text from publisher

    File URL: https://arxiv.org/pdf/2608.10177
    File Function: Latest version
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Colin B Fogarty, 2018. "Regression-assisted inference for the average treatment effect in paired experiments," Biometrika, Biometrika Trust, vol. 105(4), pages 994-1000.
    2. Xinran Li & Peng Ding, 2017. "General Forms of Finite Population Central Limit Theorems with Applications to Causal Inference," Journal of the American Statistical Association, Taylor & Francis Journals, vol. 112(520), pages 1759-1769, October.
    3. Max Cytrynbaum & Fredrik Savje, 2026. "Coupling Designs for Randomized Experiments with Complex Treatments," Papers 2604.09858, arXiv.org, revised Jul 2026.
    4. Yuehao Bai & Joseph P. Romano & Azeem M. Shaikh, 2022. "Inference in Experiments With Matched Pairs," Journal of the American Statistical Association, Taylor & Francis Journals, vol. 117(540), pages 1726-1737, October.
    5. Nicole E. Pashley & Luke W. Miratrix, 2021. "Insights on Variance Estimation for Blocked and Matched Pairs Designs," Journal of Educational and Behavioral Statistics, , vol. 46(3), pages 271-296, June.
    6. Yuehao Bai & Hongchang Guo & Azeem M. Shaikh & Max Tabord-Meehan, 2025. "Inference in Experiments with Matched Pairs and Imperfect Compliance," Journal of Business & Economic Statistics, Taylor & Francis Journals, vol. 43(3), pages 627-642, July.
    7. Yuehao Bai & Jizhou Liu & Max Tabord‐Meehan, 2024. "Inference for matched tuples and fully blocked factorial designs," Quantitative Economics, Econometric Society, vol. 15(2), pages 279-330, May.
    8. Yuehao Bai, 2022. "Optimality of Matched-Pair Designs in Randomized Controlled Trials," Papers 2206.07845, arXiv.org.
    9. Christopher Harshaw & Fredrik Sävje & Daniel A. Spielman & Peng Zhang, 2024. "Balancing Covariates in Randomized Experiments with the Gram–Schmidt Walk Design," Journal of the American Statistical Association, Taylor & Francis Journals, vol. 119(548), pages 2934-2946, October.
    10. Colin B. Fogarty, 2018. "On mitigating the analytical limitations of finely stratified experiments," Journal of the Royal Statistical Society Series B, Royal Statistical Society, vol. 80(5), pages 1035-1056, November.
    11. Bai, Yuehao & Liu, Jizhou & Shaikh, Azeem M. & Tabord-Meehan, Max, 2024. "Inference in cluster randomized trials with matched pairs," Journal of Econometrics, Elsevier, vol. 245(1).
    12. Imbens,Guido W. & Rubin,Donald B., 2015. "Causal Inference for Statistics, Social, and Biomedical Sciences," Cambridge Books, Cambridge University Press, number 9780521885881.
    13. Bai, Yuehao & Jiang, Liang & Romano, Joseph P. & Shaikh, Azeem M. & Zhang, Yichong, 2024. "Covariate adjustment in experiments with matched pairs," Journal of Econometrics, Elsevier, vol. 241(1).
    14. Clément de Chaisemartin & Jaime Ramirez-Cuellar, 2024. "At What Level Should One Cluster Standard Errors in Paired and Small-Strata Experiments?," American Economic Journal: Applied Economics, American Economic Association, vol. 16(1), pages 193-212, January.
    15. Max Cytrynbaum, 2023. "Covariate Adjustment in Stratified Experiments," Papers 2302.03687, arXiv.org, revised Jul 2024.
    16. Yuehao Bai, 2022. "Optimality of Matched-Pair Designs in Randomized Controlled Trials," American Economic Review, American Economic Association, vol. 112(12), pages 3911-3940, December.
    17. Liang Jiang & Xiaobin Liu & Peter C. B. Phillips & Yichong Zhang, 2024. "Bootstrap Inference for Quantile Treatment Effects in Randomized Experiments with Matched Pairs," The Review of Economics and Statistics, MIT Press, vol. 106(2), pages 542-556, March.
    18. Max Cytrynbaum, 2024. "Covariate adjustment in stratified experiments," Quantitative Economics, Econometric Society, vol. 15(4), pages 971-998, November.
    19. Fangzhou Su & Peng Ding, 2021. "Model‐assisted analyses of cluster‐randomized experiments," Journal of the Royal Statistical Society Series B, Royal Statistical Society, vol. 83(5), pages 994-1015, November.
    Full references (including those not matched with items on IDEAS)

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Max Cytrynbaum, 2026. "The Limits of Experimental Design: Covariate Balance Beyond Low Dimension," Papers 2608.18057, arXiv.org.

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Yuehao Bai & Xun Huang & Joseph P. Romano & Azeem M. Shaikh & Max Tabord-Meehan, 2025. "A New Design-Based Variance Estimator for Finely Stratified Experiments," Papers 2503.10851, arXiv.org, revised May 2025.
    2. Wang, Yuhao & Li, Xinran, 2025. "Asymptotic theory of the best-choice rerandomization using the Mahalanobis distance," Journal of Econometrics, Elsevier, vol. 251(C).
    3. Yuehao Bai & Azeem M. Shaikh & Max Tabord-Meehan, 2024. "A Primer on the Analysis of Randomized Experiments and a Survey of some Recent Advances," Papers 2405.03910, arXiv.org, revised Apr 2025.
    4. Bai, Yuehao & Liu, Jizhou & Shaikh, Azeem M. & Tabord-Meehan, Max, 2024. "Inference in cluster randomized trials with matched pairs," Journal of Econometrics, Elsevier, vol. 245(1).
    5. Max Cytrynbaum, 2026. "The Limits of Experimental Design: Covariate Balance Beyond Low Dimension," Papers 2608.18057, arXiv.org.
    6. Liu, Jizhou, 2026. "Inference for two-stage experiments under covariate-adaptive randomization," Journal of Econometrics, Elsevier, vol. 253(C).
    7. Max Cytrynbaum & Fredrik Savje, 2026. "Coupling Designs for Randomized Experiments with Complex Treatments," Papers 2604.09858, arXiv.org, revised Jul 2026.
    8. Zhao, Anqi & Ding, Peng, 2024. "No star is good news: A unified look at rerandomization based on p-values from covariate balance tests," Journal of Econometrics, Elsevier, vol. 241(1).
    9. Haoge Chang, 2023. "Design-based Estimation Theory for Complex Experiments," Papers 2311.06891, arXiv.org, revised May 2025.
    10. Yuehao Bai & Jizhou Liu & Max Tabord-Meehan, 2022. "Inference for Matched Tuples and Fully Blocked Factorial Designs," Papers 2206.04157, arXiv.org, revised Nov 2023.
    11. Max Cytrynbaum, 2024. "Covariate adjustment in stratified experiments," Quantitative Economics, Econometric Society, vol. 15(4), pages 971-998, November.
    12. Zikai Li, 2025. "Bridging Stratification and Regression Adjustment: Batch-Adaptive Stratification with Post-Design Adjustment in Randomized Experiments," Papers 2510.22908, arXiv.org.
    13. Yuehao Bai & Jizhou Liu & Azeem M. Shaikh & Max Tabord-Meehan, 2023. "On the Efficiency of Highly Stratified Experiments," Papers 2307.15181, arXiv.org, revised Mar 2026.
    14. Zhexiao Lin & Peng Ding, 2025. "Unifying regression-based and design-based causal inference in time-series experiments," Papers 2510.22864, arXiv.org.
    15. David McKenzie, 2025. "Designing and analysing powerful experiments: practical tips for applied researchers," Fiscal Studies, John Wiley & Sons, vol. 46(3), pages 305-322, September.
    16. Brandon Hauser & Mauricio Olivares, 2025. "A primer on power and sample size calculations for randomisation inference with experimental data," Fiscal Studies, John Wiley & Sons, vol. 46(3), pages 349-371, September.
    17. Jiang, Liang & Li, Liyao & Miao, Ke & Zhang, Yichong, 2025. "Adjustments with many regressors under covariate-adaptive randomizations," Journal of Econometrics, Elsevier, vol. 249(PB).
    18. Bai, Yuehao & Jiang, Liang & Romano, Joseph P. & Shaikh, Azeem M. & Zhang, Yichong, 2024. "Covariate adjustment in experiments with matched pairs," Journal of Econometrics, Elsevier, vol. 241(1).
    19. Jiang, Liang & Phillips, Peter C.B. & Tao, Yubo & Zhang, Yichong, 2023. "Regression-adjusted estimation of quantile treatment effects under covariate-adaptive randomizations," Journal of Econometrics, Elsevier, vol. 234(2), pages 758-776.
    20. Jiawei Fu & Cyrus Samii & Ye Wang, 2026. "Inference for Group Interaction Experiments," Papers 2607.02385, arXiv.org, revised Aug 2026.

    More about this item

    NEP fields

    This paper has been announced in the following NEP Reports:

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:arx:papers:2608.10177. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: arXiv administrators (email available below). General contact details of provider: https://arxiv.org/ .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.