EconBase
← All papers

Machine learning the first stage in 2SLS: Practical guidance from bias decomposition and simulation

Connor Lennon, Edward Rubin, Glen Waddell

arXiv 19 May 2025 · Econometrics

arXiv:2505.13422 · PDF · DOI · OpenAlex · Extracted main text

Abstract

Machine learning (ML) primarily evolved to solve "prediction problems." The first stage of two-stage least squares (2SLS) is a prediction problem, suggesting potential gains from ML first-stage assistance. However, little guidance exists on when ML helps 2SLS$\unicode{x2014}$or when it hurts. We investigate the implications of inserting ML into 2SLS, decomposing the bias into three informative components. Mechanically, ML-in-2SLS procedures face issues common to prediction and causal-inference settings$\unicode{x2014}$and their interaction. Through simulation, we show linear ML methods (e.g., post-Lasso) work well, while nonlinear methods (e.g., random forests, neural nets) generate substantial bias in second-stage estimates$\unicode{x2014}$potentially exceeding the bias of endogenous OLS.

Citation extraction

61
references
102
in-text mentions
61
distinct cited
0
self-citations
10,205
main-text words

appendix boundary found by appendix_command · 74% of the source is main text. Read the extracted text to check this.

Most heavily cited references

The works this paper leans on most, across its whole bibliography — not restricted to papers in our corpus. Ranked by composite intensity, which combines how often a work is mentioned, how many sections mention it, and how much of that falls in the main text rather than the appendix.

ReferenceIntensityMentionsSectionsMain text
1Angrist, J. D. and Frandsen, B (2022) Machine Labor1.00064100%
2Belloni, A., Chen, D., Chernozhukov, V., and Hansen, C (2012) Sparse Models and Methods for Optimal Instruments With an Application to Eminent Domain0.97715593%
3Chen, J., Chen, D. L., and Lewis, G (2020) Mostly Harmless Machine Learning: Learning Optimal Instruments in Linear IV Models0.97112592%
4Bennett, A., Kallus, N., and Schnabel, T (2019) Deep generalized method of moments for instrumental variable analysis0.7373367%
5Hartford, J., Lewis, G., Leyton-Brown, K., and Taddy, M (2017) Deep IV: A Flexible Approach for Counterfactual Prediction0.7373367%
6Angrist, J. D. and Krueger, A. B (2001) Instrumental Variables and the Search for Identification: From Supply and Demand to Natural Experiments0.64422100%
7Singh, R., Sahani, M., and Gretton, A (2019) Kernel instrumental variable regression0.64422100%
8Angrist, J. and Pischke, J.-S (2009) Mostly Harmless Econometrics: An Empiricist's Companion0.5112250%
9Belloni, A., Chernozhukov, V., and Hansen, C (2013) Inference on Treatment Effects after Selection among High-Dimensional Controls0.5112250%
10Kilbertus, N., Kusner, M. J., and Silva, R (2020) A class of algorithms for general instrumental variable models0.5112250%

Showing the top 10 of 61 scored citations.