Scalable Rule-Based Representation Learning for Interpretable Classification

Abstract

Rule-based models, e.g., decision trees, are widely used in scenarios demanding high model interpretability for their transparent inner structures and good model expressivity. However, rule-based models are hard to optimize, especially on large data sets, due to their discrete parameters and structures. Ensemble methods and fuzzy/soft rules are commonly used to improve performance, but they sacrifice the model interpretability. To obtain both good scalability and interpretability, we propose a new classifier, named Rule-based Representation Learner (RRL), that automatically learns interpretable non-fuzzy rules for data representation and classification. To train the non-differentiable RRL effectively, we project it to a continuous space and propose a novel training method, called Gradient Grafting, that can directly optimize the discrete model using gradient descent. An improved design of logical activation functions is also devised to increase the scalability of RRL and enable it to discretize the continuous features end-to-end. Exhaustive experiments on nine small and four large data sets show that RRL outperforms the competitive interpretable approaches and can be easily adjusted to obtain a trade-off between classification accuracy and model complexity for different scenarios. Our code is available at: https://github.com/12wang3/rrl.

arXiv commentAccepted by NeurIPS 2021; Interpretable ML; Neuro-Symbolic AI

References in corpus (13)

• Adam: A Method for Stochastic Optimization
• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithms
• Categorical Reparameterization with Gumbel-Softmax
• Towards A Rigorous Science of Interpretable Machine Learning
• Interpretable machine learning: definitions, methods, and applications
• Binarized Neural Networks: Training Deep Neural Networks with Weights and Activations Constrained to +1 or -1
• Interpretable classifiers using rules and Bayesian analysis: Building a better stroke prediction model
• Binary Neural Networks: A Survey
• Distilling a Neural Network Into a Soft Decision Tree
• Deep Neural Decision Trees
• Making Tree Ensembles Interpretable
• Learning Algorithms via Neural Logic Networks