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

Connor Lennon, Edward Rubin, Glen Waddell

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 M…