Abstract
We provide a new formulation for the problem of learning the optimal classification tree of a given depth as a binary linear program. A limitation of previously proposed Mathematical Optimization formulations is that they create constraints and variables for every row in the training data. As a result, the running time of the existing Integer Linear programming (ILP) formulations increases dramatically with the size of data. In our new binary formulation, we aim to circumvent this problem by making the formulation size largely independent from the training data size. We show experimentally that our formulation achieves better performance than existing formulations on both small and large problem instances within shorter running time.
Original language | English |
---|---|
Title of host publication | 33rd AAAI Conference on Artificial Intelligence, AAAI 2019, 31st Innovative Applications of Artificial Intelligence Conference, IAAI 2019 and the 9th AAAI Symposium on Educational Advances in Artificial Intelligence, EAAI 2019 |
Place of Publication | Palo Alto, California USA |
Publisher | AAAI Press |
Pages | 1625-1632 |
Number of pages | 8 |
ISBN (Print) | 978-1-57735-809-1 |
DOIs | |
Publication status | Published - 2019 |
Event | 33rd AAAI Conference on Artificial Intelligence, AAAI 2019, 31st Annual Conference on Innovative Applications of Artificial Intelligence, IAAI 2019 and the 9th AAAI Symposium on Educational Advances in Artificial Intelligence, EAAI 2019 - Honolulu, United States Duration: 27 Jan 2019 → 1 Feb 2019 Conference number: 33 https://aaai.org/Conferences/AAAI-19/ |
Conference
Conference | 33rd AAAI Conference on Artificial Intelligence, AAAI 2019, 31st Annual Conference on Innovative Applications of Artificial Intelligence, IAAI 2019 and the 9th AAAI Symposium on Educational Advances in Artificial Intelligence, EAAI 2019 |
---|---|
Abbreviated title | AAAI 2019 |
Country/Territory | United States |
City | Honolulu |
Period | 27/01/19 → 1/02/19 |
Internet address |
Funding
The work is partially supported by the NWO project Real-time data-driven maintenance logistics (project number: 628.009.012).
Keywords
- Machine learning
- decision tree
- mathematical optimization