Using Tree-Based Machine Learning for Health Studies: Literature Review and Case Series
- PMID: 36498153
- PMCID: PMC9736500
- DOI: 10.3390/ijerph192316080
Using Tree-Based Machine Learning for Health Studies: Literature Review and Case Series
Abstract
Tree-based machine learning methods have gained traction in the statistical and data science fields. They have been shown to provide better solutions to various research questions than traditional analysis approaches. To encourage the uptake of tree-based methods in health research, we review the methodological fundamentals of three key tree-based machine learning methods: random forests, extreme gradient boosting and Bayesian additive regression trees. We further conduct a series of case studies to illustrate how these methods can be properly used to solve important health research problems in four domains: variable selection, estimation of causal effects, propensity score weighting and missing data. We exposit that the central idea of using ensemble tree methods for these research questions is accurate prediction via flexible modeling. We applied ensemble trees methods to select important predictors for the presence of postoperative respiratory complication among early stage lung cancer patients with resectable tumors. We then demonstrated how to use these methods to estimate the causal effects of popular surgical approaches on postoperative respiratory complications among lung cancer patients. Using the same data, we further implemented the methods to accurately estimate the inverse probability weights for a propensity score analysis of the comparative effectiveness of the surgical approaches. Finally, we demonstrated how random forests can be used to impute missing data using the Study of Women's Health Across the Nation data set. To conclude, the tree-based methods are a flexible tool and should be properly used for health investigations.
Keywords: causal inference; ensemble methods; missing data; sensitivity analysis; variable selection.
Conflict of interest statement
The authors declare no conflict of interest.
Figures
References
-
- Hernández B., Pennington S.R., Parnell A.C. Bayesian methods for proteomic biomarker development. EuPA Open Proteom. 2015;9:54–64. doi: 10.1016/j.euprot.2015.08.001. - DOI
-
- Hu L., Gu C. Estimation of causal effects of multiple treatments in healthcare database studies with rare outcomes. Health Serv. Outcomes Res. Methodol. 2021;21:287–308. doi: 10.1007/s10742-020-00234-4. - DOI
-
- Mazumdar M., Lin J.Y.J., Zhang W., Li L., Liu M., Dharmarajan K., Sanderson M., Isola L., Hu L. Comparison of statistical and machine learning models for healthcare cost data: A simulation study motivated by Oncology Care Model (OCM) data. BMC Health Serv. Res. 2020;20:350. doi: 10.1186/s12913-020-05148-y. - DOI - PMC - PubMed
Publication types
MeSH terms
Grants and funding
LinkOut - more resources
Full Text Sources
Medical
