We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

stat.ML

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Statistics > Machine Learning

Title: Learning Certifiably Optimal Rule Lists for Categorical Data

Abstract: We present the design and implementation of a custom discrete optimization technique for building rule lists over a categorical feature space. Our algorithm produces rule lists with optimal training performance, according to the regularized empirical risk, with a certificate of optimality. By leveraging algorithmic bounds, efficient data structures, and computational reuse, we achieve several orders of magnitude speedup in time and a massive reduction of memory consumption. We demonstrate that our approach produces optimal rule lists on practical problems in seconds. Our results indicate that it is possible to construct optimal sparse rule lists that are approximately as accurate as the COMPAS proprietary risk prediction tool on data from Broward County, Florida, but that are completely interpretable. This framework is a novel alternative to CART and other decision tree methods for interpretable modeling.
Comments: A short version of this work appeared in KDD '17 as "Learning Certifiably Optimal Rule Lists"
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG)
Journal reference: Journal of Machine Learning Research 18(234):1-78, 2018
Cite as: arXiv:1704.01701 [stat.ML]
  (or arXiv:1704.01701v4 [stat.ML] for this version)

Submission history

From: Elaine Angelino [view email]
[v1] Thu, 6 Apr 2017 04:02:35 GMT (252kb,D)
[v2] Thu, 30 Nov 2017 19:00:01 GMT (6947kb,D)
[v3] Wed, 20 Jun 2018 17:49:44 GMT (3478kb,D)
[v4] Fri, 3 Aug 2018 22:51:24 GMT (3478kb,D)

Link back to: arXiv, form interface, contact.