Journal article
Hybrid Predictive Models: When an Interpretable Model Collaborates with a Black-box Model
Journal of machine learning research, Vol.22(137), pp.1-38
04/01/2021
Abstract
Interpretable machine learning has become a strong competitor for black-box models. However, the possible loss of the predictive performance for gaining understandability is often inevitable, especially when it needs to satisfy users with diverse backgrounds or high standards for what is considered interpretable. This tension puts practitioners in a dilemma of choosing between high accuracy (black-box models) and interpretability (interpretable models). In this work, we propose a novel framework for building a Hybrid Predictive Model that integrates an interpretable model with any pre-trained black-box model to combine their strengths. The interpretable model substitutes the black-box model on a subset of data where the interpretable model is most competent, gaining transparency at a low cost of the predictive accuracy. We design a principled objective function that considers predictive accuracy, model interpretability, and model transparency (defined as the percentage of data processed by the interpretable substitute.) Under this framework, we propose two hybrid models, one substituting with association rules and the other with linear models, and design customized training algorithms for both models. We test the hybrid models on structured data and text data where interpretable models collaborate with various state-of-the-art black-box models. Results show that hybrid models obtain an efficient trade-off between transparency and predictive performance, characterized by pareto frontiers. Finally, we apply the proposed model on a real-world patients dataset for predicting cardiovascular disease and propose multi-model Pareto frontiers to assist model selection in real applications.
Details
- Title: Subtitle
- Hybrid Predictive Models: When an Interpretable Model Collaborates with a Black-box Model
- Creators
- Tong Wang - Univ Iowa, Tippie Coll Business, Iowa City, IA 52242 USAQihang Lin - Univ Iowa, Tippie Coll Business, Iowa City, IA 52242 USA
- Resource Type
- Journal article
- Publication Details
- Journal of machine learning research, Vol.22(137), pp.1-38
- Publisher
- Microtome Publ
- ISSN
- 1532-4435
- eISSN
- 1533-7928
- Number of pages
- 38
- Language
- English
- Date published
- 04/01/2021
- Academic Unit
- Business Analytics
- Record Identifier
- 9984380449602771
Metrics
1 Record Views