Double machine learning for a partially linear model with endogenous treatments and multivariate sample selection

Statistics and Computing. 2026. Vol. 36. Article 200.

Potanin B.

We propose double machine learning (DML) estimators for a partially linear model (PLM) with endogenous treatments and multivariate sample selection. We prove asymptotic normality of the estimators under mild regularity conditions and study finite sample properties on simulated data. The results demonstrate the importance of addressing sample selection in PLMs and the usefulness of the proposed estimators for avoiding selection bias. Moreover, we extend the proposed estimators to the case of endogenous switching.

Keywords: sample selection, неслучайный отбор, treatment effects, эффекты воздействия, double machine learning, partially linear model, двойное машинное обучение, частично линейная регрессия