Can We Train Machine Learning Methods to Outperform the High-dimensional Propensity Score Algorithm?

5 years ago

Can We Train Machine Learning Methods to Outperform the High-dimensional Propensity Score Algorithm?

Pang, Menglan, Karim, Mohammad Ehsanul, Platt, Robert W.

The use of retrospective health care claims datasets is frequently criticized for the lack of complete information on potential confounders. Utilizing patient’s health status–related information from claims datasets as surrogates or proxies for mismeasured and unobserved confounders, the high-dimensional propensity score algorithm enables us to reduce bias. Using a previously published cohort study of postmyocardial infarction statin use (1998–2012), we compare the performance of the algorithm with a number of popular machine learning approaches for confounder selection in high-dimensional covariate spaces: random forest, least absolute shrinkage and selection operator, and elastic net. Our results suggest that, when the data analysis is done with epidemiologic principles in mind, machine learning methods perform as well as the high-dimensional propensity score algorithm. Using a plasmode framework that mimicked the empirical data, we also showed that a hybrid of machine learning and high-dimensional propensity score algorithms generally perform slightly better than both in terms of mean squared error, when a bias-based analysis is used.

Publisher URL: https://journals.lww.com/epidem/Fulltext/2018/03000/Can_We_Train_Machine_Learning_Methods_to.5.aspx

DOI: 10.1097/EDE.0000000000000787