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

Estimating Treatment Effects Under Bounded Heterogeneity

Author

Listed:
  • Soonwoo Kwon
  • Liyang Sun

Abstract

Specifications that impose constant treatment effects are common but biased, while fully flexible alternatives can be imprecise or infeasible. Under a bound on treatment effect heterogeneity, we propose a generalized ridge estimator, $\texttt{regulaTE}$, that yields heterogeneity-aware confidence intervals (CIs). The ridge penalty is chosen to optimally trade off worst-case bias and variance in a Gaussian homoskedastic setting; the resulting CIs remain tight more generally and are valid even under lack of overlap. Varying the bound enables sensitivity analysis to departures from constant effects, which we illustrate in leading empirical applications of unconfoundedness and staggered adoption designs.

Suggested Citation

  • Soonwoo Kwon & Liyang Sun, 2025. "Estimating Treatment Effects Under Bounded Heterogeneity," Papers 2510.05454, arXiv.org, revised Apr 2026.
  • Handle: RePEc:arx:papers:2510.05454
    as

    Download full text from publisher

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

    References listed on IDEAS

    as
    1. Goodman-Bacon, Andrew, 2021. "Difference-in-differences with variation in treatment timing," Journal of Econometrics, Elsevier, vol. 225(2), pages 254-277.
    2. Clément de Chaisemartin & Xavier D'Haultfœuille, 2020. "Two-Way Fixed Effects Estimators with Heterogeneous Treatment Effects," American Economic Review, American Economic Association, vol. 110(9), pages 2964-2996, September.
    3. Christoph Rothe, 2017. "Robust Confidence Intervals for Average Treatment Effects Under Limited Overlap," Econometrica, Econometric Society, vol. 85, pages 645-660, March.
    4. Joshua D. Angrist & Jörn-Steffen Pischke, 2009. "Mostly Harmless Econometrics: An Empiricist's Companion," Economics Books, Princeton University Press, edition 1, number 8769, December.
    5. Guido W. Imbens & Jeffrey M. Wooldridge, 2009. "Recent Developments in the Econometrics of Program Evaluation," Journal of Economic Literature, American Economic Association, vol. 47(1), pages 5-86, March.
    6. Anna Aizer & Shari Eli & Joseph Ferrie & Adriana Lleras-Muney, 2016. "The Long-Run Impact of Cash Transfers to Poor Families," American Economic Review, American Economic Association, vol. 106(4), pages 935-971, April.
    7. Timothy B. Armstrong & Michal Kolesár, 2018. "Optimal Inference in a Class of Regression Models," Econometrica, Econometric Society, vol. 86(2), pages 655-683, March.
    8. D’Amour, Alexander & Ding, Peng & Feller, Avi & Lei, Lihua & Sekhon, Jasjeet, 2021. "Overlap in observational studies with high-dimensional covariates," Journal of Econometrics, Elsevier, vol. 221(2), pages 644-654.
    9. Cl'ement de Chaisemartin & Antoine Deeb, 2024. "Estimating treatment-effect heterogeneity across sites, in multi-site randomized experiments with few units per site," Papers 2405.17254, arXiv.org, revised Dec 2024.
    10. Stelios Michalopoulos & Elias Papaioannou, 2016. "The Long-Run Effects of the Scramble for Africa," American Economic Review, American Economic Association, vol. 106(7), pages 1802-1848, July.
    11. Richard K. Crump & V. Joseph Hotz & Guido W. Imbens & Oscar A. Mitnik, 2009. "Dealing with limited overlap in estimation of average treatment effects," Biometrika, Biometrika Trust, vol. 96(1), pages 187-199.
    12. Heiler, Phillip & Kazak, Ekaterina, 2021. "Valid inference for treatment effect parameters under irregular identification and many extreme propensity scores," Journal of Econometrics, Elsevier, vol. 222(2), pages 1083-1108.
    13. Carlos Dobkin & Amy Finkelstein & Raymond Kluender & Matthew J. Notowidigdo, 2018. "The Economic Consequences of Hospital Admissions," American Economic Review, American Economic Association, vol. 108(2), pages 308-352, February.
    14. Gustavo J. Bobonis & Luis R. Cámara Fuertes & Rainer Schwabe, 2016. "Monitoring Corruptible Politicians," American Economic Review, American Economic Association, vol. 106(8), pages 2371-2405, August.
    15. Paul Goldsmith-Pinkham & Peter Hull & Michal Kolesár, 2024. "Contamination Bias in Linear Regressions," American Economic Review, American Economic Association, vol. 114(12), pages 4015-4051, December.
    16. Susan Athey & Guido W. Imbens & Stefan Wager, 2018. "Approximate residual balancing: debiased inference of average treatment effects in high dimensions," Journal of the Royal Statistical Society Series B, Royal Statistical Society, vol. 80(4), pages 597-623, September.
    17. Sasaki, Yuya & Ura, Takuya, 2022. "Estimation And Inference For Moments Of Ratios With Robustness Against Large Trimming Bias," Econometric Theory, Cambridge University Press, vol. 38(1), pages 66-112, February.
    18. Claudia Noack & Christoph Rothe, 2024. "Bias‐Aware Inference in Fuzzy Regression Discontinuity Designs," Econometrica, Econometric Society, vol. 92(3), pages 687-711, May.
    19. Timothy B. Armstrong & Michal Kolesár & Soonwoo Kwon, 2020. "Bias-Aware Inference in Regularized Regression Models," Working Papers 2020-2, Princeton University. Economics Department..
    20. Timothy B. Armstrong & Michal Kolesár, 2021. "Finite‐Sample Optimal Estimation and Inference on Average Treatment Effects Under Unconfoundedness," Econometrica, Econometric Society, vol. 89(3), pages 1141-1177, May.
    21. Charles F. Manski & John V. Pepper, 2018. "How Do Right-to-Carry Laws Affect Crime Rates? Coping with Ambiguity Using Bounded-Variation Assumptions," The Review of Economics and Statistics, MIT Press, vol. 100(2), pages 232-244, May.
    22. Giovanni Favara & Jean Imbs, 2015. "Credit Supply and the Price of Housing," American Economic Review, American Economic Association, vol. 105(3), pages 958-992, March.
    23. Alejandro Sanchez-Becerra, 2023. "Robust inference for the treatment effect variance in experiments using machine learning," Papers 2306.03363, arXiv.org.
    24. Chenchuan (Mark) Li & Ulrich K. Müller, 2021. "Linear regression with many controls of limited explanatory power," Quantitative Economics, Econometric Society, vol. 12(2), pages 405-442, May.
    25. Roth, Jonathan & Sant’Anna, Pedro H.C. & Bilinski, Alyssa & Poe, John, 2023. "What’s trending in difference-in-differences? A synthesis of the recent econometrics literature," Journal of Econometrics, Elsevier, vol. 235(2), pages 2218-2244.
    26. Guido Imbens & Stefan Wager, 2019. "Optimized Regression Discontinuity Designs," The Review of Economics and Statistics, MIT Press, vol. 101(2), pages 264-278, May.
    27. Monica Martinez-Bravo, 2014. "The Role of Local Officials in New Democracies: Evidence from Indonesia," American Economic Review, American Economic Association, vol. 104(4), pages 1244-1287, April.
    28. repec:hal:pseose:hal-01301589 is not listed on IDEAS
    29. Sun, Liyang & Abraham, Sarah, 2021. "Estimating dynamic treatment effects in event studies with heterogeneous treatment effects," Journal of Econometrics, Elsevier, vol. 225(2), pages 175-199.
    30. Xinwei Ma & Jingshen Wang, 2020. "Robust Inference Using Inverse Probability Weighting," Journal of the American Statistical Association, Taylor & Francis Journals, vol. 115(532), pages 1851-1860, December.
    31. Patrick Kline & Raffaele Saggio & Mikkel Sølvsten, 2020. "Leave‐Out Estimation of Variance Components," Econometrica, Econometric Society, vol. 88(5), pages 1859-1898, September.
    32. Jonathan M.V. Davis & Sara B. Heller, 2017. "Using Causal Forests to Predict Treatment Heterogeneity: An Application to Summer Jobs," American Economic Review, American Economic Association, vol. 107(5), pages 546-550, May.
    33. Matthew A. Masten & Alexandre Poirier, 2020. "Inference on breakdown frontiers," Quantitative Economics, Econometric Society, vol. 11(1), pages 41-111, January.
    34. Joshua D. Angrist, 1998. "Estimating the Labor Market Impact of Voluntary Military Service Using Social Security Data on Military Applicants," Econometrica, Econometric Society, vol. 66(2), pages 249-288, March.
    Full references (including those not matched with items on IDEAS)

    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. Sloczynski, Tymon, 2018. "A General Weighted Average Representation of the Ordinary and Two-Stage Least Squares Estimands," IZA Discussion Papers 11866, IZA Network @ LISER.
    2. Ganesh Karapakula, 2023. "Stable Probability Weighting: Large-Sample and Finite-Sample Estimation and Inference Methods for Heterogeneous Causal Effects of Multivalued Treatments Under Limited Overlap," Papers 2301.05703, arXiv.org, revised Jan 2023.
    3. Jacob Dorn, 2025. "How Much Weak Overlap Can Doubly Robust T-Statistics Handle?," Papers 2504.13273, arXiv.org, revised Apr 2025.
    4. Tymon S{l}oczy'nski, 2018. "Interpreting OLS Estimands When Treatment Effects Are Heterogeneous: Smaller Groups Get Larger Weights," Papers 1810.01576, arXiv.org, revised May 2020.
    5. Phillip Heiler & Michael C. Knaus, 2025. "Heterogeneity Analysis with Heterogeneous Treatments," Papers 2507.01517, arXiv.org, revised Feb 2026.
    6. Roth, Jonathan & Sant’Anna, Pedro H.C. & Bilinski, Alyssa & Poe, John, 2023. "What’s trending in difference-in-differences? A synthesis of the recent econometrics literature," Journal of Econometrics, Elsevier, vol. 235(2), pages 2218-2244.
    7. Mogstad, Magne & Torgovitsky, Alexander, 2024. "Instrumental variables with unobserved heterogeneity in treatment effects," Handbook of Labor Economics,, Elsevier.
    8. Martin Huber, 2019. "An introduction to flexible methods for policy evaluation," Papers 1910.00641, arXiv.org.
    9. Timothy B. Armstrong & Patrick Kline & Liyang Sun, 2025. "Adapting to Misspecification," Econometrica, Econometric Society, vol. 93(6), pages 1981-2005, November.
    10. Benjamin Lu & Eli Ben-Michael & Avi Feller & Luke Miratrix, 2023. "Is It Who You Are or Where You Are? Accounting for Compositional Differences in Cross-Site Treatment Effect Variation," Journal of Educational and Behavioral Statistics, , vol. 48(4), pages 420-453, August.
    11. Li, Daiyue & Jin, Yanhong & Cheng, Mingwang, 2024. "Unleashing the power of industrial robotics on firm productivity: Evidence from China," Journal of Economic Behavior & Organization, Elsevier, vol. 224(C), pages 500-520.
    12. Mark Kattenberg & Bas Scheer & Jurre Thiel, 2023. "Causal forests with fixed effects for treatment effect heterogeneity in difference-in-differences," CPB Discussion Paper 452, CPB Netherlands Bureau for Economic Policy Analysis.
    13. Peter Hull & Michal Kolesár & Christopher Walters, 2022. "Labor by design: contributions of David Card, Joshua Angrist, and Guido Imbens," Scandinavian Journal of Economics, Wiley Blackwell, vol. 124(3), pages 603-645, July.
    14. Dmitry Arkhangelsky & Guido Imbens, 2023. "Causal Models for Longitudinal and Panel Data: A Survey," Papers 2311.15458, arXiv.org, revised Jun 2024.
    15. Tamara Bischof & Boris Kaiser, 2021. "Who cares when you close down? The effects of primary care practice closures on patients," Health Economics, John Wiley & Sons, Ltd., vol. 30(9), pages 2004-2025, September.
    16. Kline, Patrick, 2024. "Firm wage effects," Handbook of Labor Economics,, Elsevier.
    17. Gregory Faletto, 2023. "Fused Extended Two-Way Fixed Effects for Difference-in-Differences With Staggered Adoptions," Papers 2312.05985, arXiv.org, revised Apr 2025.
    18. Baraldi, Anna Laura & Cantabene, Claudia & De Iudicibus, Alessandro & Fosco, Giovanni & Papagni, Erasmo, 2025. "Shocks and Selection: How Earthquakes Shape Local Political Representation," MPRA Paper 125061, University Library of Munich, Germany, revised 18 Jun 2025.
    19. Philipp Bach & Sven Klaassen & Jannis Kueck & Mara Mattes & Martin Spindler, 2025. "Sensitivity Analysis for Treatment Effects in Difference-in-Differences Models using Riesz Representation," Papers 2510.09064, arXiv.org.
    20. Dor Leventer, 2025. "Conditional Triple Difference-in-Differences," Papers 2502.16126, arXiv.org, revised Jun 2025.

    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:2510.05454. 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: http://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.