Building more accurate decision trees with the additive tree
José Marcio Luna,Efstathios D Gennatas,Lyle H Ungar,Eric Eaton,Eric S Diffenderfer,Shane T Jensen,Charles B Simone,Jerome H Friedman,Timothy D Solberg,Gilmer Valdes,Efstathios D. Gennatas,Lyle H. Ungar,Eric S. Diffenderfer,Shane T. Jensen,Charles B. Simone,Jerome H. Friedman,Timothy D. Solberg
DOI: https://doi.org/10.1073/pnas.1816748116
IF: 11.1
2019-09-16
Proceedings of the National Academy of Sciences
Abstract:Significance As machine learning applications expand to high-stakes areas such as criminal justice, finance, and medicine, legitimate concerns emerge about high-impact effects of individual mispredictions on people’s lives. As a result, there has been increasing interest in understanding general machine learning models to overcome possible serious risks. Current decision trees, such as Classification and Regression Trees (CART), have played a predominant role in fields such as medicine, due to their simplicity and intuitive interpretation. However, such trees suffer from intrinsic limitations in predictive power. We developed the additive tree, a theoretical approach to generate a more accurate and interpretable decision tree, which reveals connections between CART and gradient boosting. The additive tree exhibits superior predictive performance to CART, as validated on 83 classification tasks.
multidisciplinary sciences