Estimating the Individual Treatment Effect with Different Treatment Group Sizes

Luyuan Song and Xiaojun Zhang ()
Additional contact information
Luyuan Song: School of Mathematical Sciences, University of Electronic Science and Technology of China, Chengdu 611731, China
Xiaojun Zhang: School of Mathematical Sciences, University of Electronic Science and Technology of China, Chengdu 611731, China

Mathematics, 2024, vol. 12, issue 8, 1-17

Abstract: Machine learning for causal inference, particularly at the individual level, has attracted intense interest in many domains. Existing techniques focus on controlling differences in distribution between treatment groups in a data-driven manner, eliminating the effects of confounding factors. However, few of the current methods adequately discuss the difference in treatment group sizes. Two approaches, a direct and an indirect one, deal with potential missing data for estimating individual treatment with binary treatments and different treatment group sizes. We embed the two methods into certain frameworks based on the domain adaption and representation. We validate the performance of our method by two benchmarks in the causal inference community: simulated data and real-world data. Experiment results verify that our methods perform well.

Keywords: causal inference; individual treatment effect; observational data; imbalanced dataset; binary treatments (search for similar items in EconPapers)
JEL-codes: C (search for similar items in EconPapers)
Date: 2024
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2227-7390/12/8/1224/pdf (application/pdf)
https://www.mdpi.com/2227-7390/12/8/1224/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jmathe:v:12:y:2024:i:8:p:1224-:d:1378514

Access Statistics for this article

Mathematics is currently edited by Ms. Emma He

More articles in Mathematics from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().