A New Interpretable Neural Network-Based Rule Model for Healthcare Decision Making
This addresses the need for interpretable AI in healthcare, offering a novel method that combines interpretability with performance, though it is incremental in building upon existing TTnet technology.
The paper tackles the problem of making neural networks interpretable for healthcare decision-making by introducing TT-rules, a framework that transforms neural networks into rule-based models while maintaining high performance, achieving equal or better results compared to other interpretable methods and handling large datasets with over 20K features.
In healthcare applications, understanding how machine/deep learning models make decisions is crucial. In this study, we introduce a neural network framework, $\textit{Truth Table rules}$ (TT-rules), that combines the global and exact interpretability properties of rule-based models with the high performance of deep neural networks. TT-rules is built upon $\textit{Truth Table nets}$ (TTnet), a family of deep neural networks initially developed for formal verification. By extracting the necessary and sufficient rules $\mathcal{R}$ from the trained TTnet model (global interpretability) to yield the same output as the TTnet (exact interpretability), TT-rules effectively transforms the neural network into a rule-based model. This rule-based model supports binary classification, multi-label classification, and regression tasks for small to large tabular datasets. After outlining the framework, we evaluate TT-rules' performance on healthcare applications and compare it to state-of-the-art rule-based methods. Our results demonstrate that TT-rules achieves equal or higher performance compared to other interpretable methods. Notably, TT-rules presents the first accurate rule-based model capable of fitting large tabular datasets, including two real-life DNA datasets with over 20K features.