Research

Research Interests

My research interests focus on developing statistical tools for robust causal inference and data-driven decision-making in social and biomedical sciences. Specifically, I have been working on

Policy learning in causal inference
Heterogeneous treatment effect estimation
Transfer Learning and Federated Learning

Working Papers and Publications

Safe Policy Learning under Regression Discontinuity Designs
Yi Zhang, Eli Ben-Michael, Kosuke Imai. (2022)
2023 American Statistical Association Student Paper Award in Social Statistics Section
[arXiv]

Heterogeneous Causal Effect Estimation in Underrepresented Populations: Federated and Transfer Learning Approaches
Larry Han, Yi Zhang, Rui Duan.
In preparation, 2023

Augmented Transfer Regression Learning with Semi-non-parametric Nuisance Models
Molei Liu*, Yi Zhang*, Katherine P Liao, Tianxi Cai. (2022)
Journal of Machine Learning Research (accepted with minor revisions), 2023
[arXiv]

Double/Debiased Machine Learning for Logistic Partially Linear Model
Molei Liu*, Yi Zhang*, Doudou Zhou*. (2020)
The Econometrics Journal, 2021
[arXiv][Code][Journal]

*: The authors contributed equally to this work or are listed in alphabetic order.