IDEAS home Printed from https://ideas.repec.org/p/arx/papers/2506.12225.html
   My bibliography  Save this paper

Optimal treatment assignment rules under capacity constraints

Author

Listed:
  • Keita Sunada
  • Kohei Izumi

Abstract

We study treatment assignment problems under capacity constraints, where a planner aims to maximize social welfare by assigning treatments based on observable covariates. Such constraints, common when treatments are costly or limited in supply, introduce nontrivial challenges for deriving optimal statistical assignment rules because the planner needs to coordinate treatment assignment probabilities across the entire covariate distribution. To address these challenges, we reformulate the planner's constrained maximization problem as an optimal transport problem, which makes the problem effectively unconstrained. We then establish local asymptotic optimality results of assignment rules using a limits of experiments framework. Finally, we illustrate our method with a voucher assignment problem for private secondary school attendance using data from Angrist et al. (2006)

Suggested Citation

  • Keita Sunada & Kohei Izumi, 2025. "Optimal treatment assignment rules under capacity constraints," Papers 2506.12225, arXiv.org, revised Sep 2025.
  • Handle: RePEc:arx:papers:2506.12225
    as

    Download full text from publisher

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

    References listed on IDEAS

    as
    1. Daido Kido, 2022. "Distributionally Robust Policy Learning with Wasserstein Distance," Papers 2205.04637, arXiv.org, revised Aug 2022.
    2. Alfred Galichon, 2016. "Optimal Transport Methods in Economics," Economics Books, Princeton University Press, edition 1, number 10870.
    3. Alfred Galichon, 2016. "Optimal transport methods in economics," SciencePo Working papers hal-03256830, HAL.
    4. Tetenov, Aleksey, 2012. "Statistical treatment choice based on asymmetric minimax regret criteria," Journal of Econometrics, Elsevier, vol. 166(1), pages 157-165.
    5. Alfred Galichon, 2016. "Optimal transport methods in economics," Post-Print hal-03256830, HAL.
    6. Abhijit V. Banerjee & Sylvain Chassang & Sergio Montero & Erik Snowberg, 2020. "A Theory of Experimenters: Robustness, Randomization, and Balance," American Economic Review, American Economic Association, vol. 110(4), pages 1206-1230, April.
    7. Alfred Galichon, 2016. "Optimal transport methods in economics," SciencePo Working papers Main hal-03256830, HAL.
    8. Jos'e Luis Montiel Olea & Chen Qiu & Jorg Stoye, 2023. "Decision Theory for Treatment Choice Problems with Partial Identification," Papers 2312.17623, arXiv.org, revised Jun 2025.
    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. Molinari, Francesca, 2020. "Microeconometrics with partial identification," Handbook of Econometrics, in: Steven N. Durlauf & Lars Peter Hansen & James J. Heckman & Rosa L. Matzkin (ed.), Handbook of Econometrics, edition 1, volume 7, chapter 0, pages 355-486, Elsevier.
    2. Principi, Giulio & Wakker, Peter P. & Wang, Ruodu, 2025. "Antimonotonicity for preference axioms: the natural counterpart to comonotonicity," Theoretical Economics, Econometric Society, vol. 20(3), July.
    3. Arthur Charpentier & Emmanuel Flachaire & Ewen Gallic, 2023. "Optimal Transport for Counterfactual Estimation: A Method for Causal Inference," Papers 2301.07755, arXiv.org.
    4. Florian Gunsilius & Susanne M. Schennach, 2017. "A nonlinear principal component decomposition," CeMMAP working papers 16/17, Institute for Fiscal Studies.
    5. Itai Arieli & Yakov Babichenko & Fedor Sandomirskiy, 2023. "Feasible Conditional Belief Distributions," Papers 2307.07672, arXiv.org, revised Nov 2024.
    6. Beatrice Acciaio & Berenice Anne Neumann, 2025. "Characterization of transport optimizers via graphs and applications to Stackelberg–Cournot–Nash equilibria," Mathematics and Financial Economics, Springer, volume 19, number 3, March.
    7. Andrew Lyasoff, 2023. "The Time-Interlaced Self-Consistent Master System of Heterogeneous-Agent Models," Papers 2303.12567, arXiv.org, revised May 2025.
    8. Roger Koenker, 2017. "Quantile regression 40 years on," CeMMAP working papers 36/17, Institute for Fiscal Studies.
    9. Kuan‐Ming Chen & Yu‐Wei Hsieh & Ming‐Jen Lin, 2023. "Reducing Recommendation Inequality Via Two‐Sided Matching: A Field Experiment Of Online Dating," International Economic Review, Department of Economics, University of Pennsylvania and Osaka University Institute of Social and Economic Research Association, vol. 64(3), pages 1201-1221, August.
    10. Arthur Charpentier & Alfred Galichon & Lucas Vernet, 2019. "Optimal transport on large networks a practitioner guide," SciencePo Working papers Main hal-02173210, HAL.
    11. Alfred Galichon & Bernard Salanié, 2023. "Structural Estimation of Matching Markets with Transferable Utility," Post-Print hal-03935865, HAL.
    12. Ashwin Kambhampati & Carlos Segura‐Rodriguez, 2022. "The optimal assortativity of teams inside the firm," RAND Journal of Economics, RAND Corporation, vol. 53(3), pages 484-515, September.
    13. Wayne Yuan Gao & Rui Wang, 2023. "IV Regressions without Exclusion Restrictions," Papers 2304.00626, arXiv.org, revised Jul 2023.
    14. Alfred Galichon, 2021. "The Unreasonable Effectiveness of Optimal Transport in Economics," SciencePo Working papers Main hal-03936221, HAL.
    15. Schennach, Susanne M., 2020. "Mismeasured and unobserved variables," Handbook of Econometrics, in: Steven N. Durlauf & Lars Peter Hansen & James J. Heckman & Rosa L. Matzkin (ed.), Handbook of Econometrics, edition 1, volume 7, chapter 0, pages 487-565, Elsevier.
    16. Alfred Galichon, 2021. "The Unreasonable Effectiveness of Optimal Transport in Economics," Working Papers hal-03936221, HAL.
    17. Haiyan Liu & Bin Wang & Ruodu Wang & Sheng Chao Zhuang, 2023. "Distorted optimal transport," Papers 2308.11238, arXiv.org, revised May 2025.
    18. Pablo D. Fajgelbaum & Edouard Schaal, 2020. "Optimal Transport Networks in Spatial Equilibrium," Econometrica, Econometric Society, vol. 88(4), pages 1411-1452, July.
    19. Carlier, Guillaume & Dupuy, Arnaud & Galichon, Alfred & Sun, Yifei, 2021. "SISTA: Learning Optimal Transport Costs under Sparsity Constraints," IZA Discussion Papers 14397, Institute of Labor Economics (IZA).
    20. Vuillermot, Pierre-A. & Zambrini, J.-C., 2020. "On Bernstein processes generated by hierarchies of linear parabolic systems in Rd," Stochastic Processes and their Applications, Elsevier, vol. 130(5), pages 2974-3004.

    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:2506.12225. 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.