arrow
返回

Optimal classification trees

delete2017-04-03
delete416
delete
OA
AI
D
Dimitris Bertsimas *
J
Jack Dunn
DOI:10.1007/s10994-017-5633-9delete
delete原文链接
delete分享
delete收藏
查看原文
摘要

摘要

En 中文
State-of-the-art decision tree methods apply heuristics recursively to create each split in isolation, which may not capture well the underlying characteristics of the dataset. The optimal decision tree problem attempts to resolve this by creating the entire decision tree at once to achieve global optimality. In the last 25 years, algorithmic advances in integer optimization coupled with hardware improvements have resulted in an astonishing 800 billion factor speedup in mixed-integer optimization (MIO). Motivated by this speedup, we present optimal classification trees, a novel formulation of the decision tree problem using modern MIO techniques that yields the optimal decision tree for axes-aligned splits. We also show the richness of this MIO formulation by adapting it to give optimal classification trees with hyperplanes that generates optimal decision trees with multivariate splits. Synthetic tests demonstrate that these methods recover the true decision tree more closely than heuristics, refuting the notion that optimal methods overfit the training data. We comprehensively benchmark these methods on a sample of 53 datasets from the UCI machine learning repository. We establish that these MIO methods are practically solvable on real-world datasets with sizes in the 1000s, and give average absolute improvements in out-of-sample accuracy over CART of 1-2 and 3-5% for the univariate and multivariate cases, respectively. Furthermore, we identify that optimal classification trees are likely to outperform CART by 1.2-1.3% in situations where the CART accuracy is high and we have sufficient training data, while the multivariate version outperforms CART by 4-7% when the CART accuracy or dimension of the dataset is low.
Keyword:
Classification
Decision trees
Mixed integer optimization
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

Machine Learning 封面图
Machine Learning
IF:
2.9
论文数:
2.7K
被引数:
3.4W

机构

暂无机构信息
引用论文

引用论文

err分享
err收藏
Metals and organic species associated with fine and coarse aerosol particles in an electronic waste recycling plant
err2023-02-06
err0
PREAI
errGiulia Pazzi; Francesca Buiarelli; Patrizia Di Filippo; Donatella Pomata; Carmela Riccardi; Franco Lucarelli; Fabio Giardi; Elisa Sonego; Roberta Galarini; Stefano Lorenzetti; Laura Goracci; Giulia Simonetti
err分享
err收藏
LEAST QUANTILE REGRESSION VIA MODERN OPTIMIZATION
err2014-12-01
err36
errOAAI
errBertsimas, Dimitris; Mazumder, Rahul
err分享
err收藏
Use of Ca–alginate as a novel support for TiO2 immobilization in methylene blue decolorisation
err2009-04-01
err0
PREAI
errJuliana Q. Albarelli; Diego T. Santos; Sharon Murphy; Michael Oelgemöller
err分享
err收藏
Fisher's decision tree
err2013-11-01
err46
PREAI
errLopez-Chau, Asdrubal; Cervantes, Jair; Lopez-Garcia, Lourdes; Garcia Lamont, Farid
err分享
err收藏
Sustained increase in hippocampal sharp-wave ripple activity during slow-wave sleep after learning
err2008-04-02
err0
errOAAI
errOxana Eschenko; Wiâm Ramadan; Matthias Mölle; Jan Born; Susan J. Sara
err分享
err收藏
BEST SUBSET SELECTION VIA A MODERN OPTIMIZATION LENS
err2016-04-01
err486
errOAAI
errBertsimas, Dimitris; King, Angela; Mazumder, Rahul
err分享
err收藏
Fluorescence enhancement of warfarin induced by interaction with β‐cyclodextrin
err2009-08-10
err0
PREAI
errJacob M. Vasquez; Andrew Vu; Jerome S. Schultz; Valentine I. Vullev
err分享
err收藏
学者 查看更多内容