Interpretable classifiers using rules and Bayesian analysis: Building a better stroke prediction model
Massachusetts Institute of Technology · University of Washington · +1 more institution
Abstract
We aim to produce predictive models that are not only accurate, but are also interpretable to human experts. Our models are decision lists, which consist of a series of if…then…statements (e.g., if high blood pressure, then stroke) that discretize a high-dimensional, multivariate feature space into a series of simple, readily interpretable decision statements. We introduce a generative model called Bayesian Rule Lists that yields a posterior distribution over possible decision lists. It employs a novel prior structure to encourage sparsity. Our experiments show that Bayesian Rule Lists has predictive accuracy on par with the current top algorithms for prediction in machine learning. Our method is motivated by…
Citation impact
- FWCI
- 57.78
- Percentile
- 100%
- References
- 83
Authors
4Topics & keywords
- Machine learning
- Computer science
- Artificial intelligence
- Bayesian probability
- Feature (linguistics)
- Data mining
- Peace, Justice and strong institutions