Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Interpretable multiclass classification by MDL-based rule lists

About

Interpretable classifiers have recently witnessed an increase in attention from the data mining community because they are inherently easier to understand and explain than their more complex counterparts. Examples of interpretable classification models include decision trees, rule sets, and rule lists. Learning such models often involves optimizing hyperparameters, which typically requires substantial amounts of data and may result in relatively large models. In this paper, we consider the problem of learning compact yet accurate probabilistic rule lists for multiclass classification. Specifically, we propose a novel formalization based on probabilistic rule lists and the minimum description length (MDL) principle. This results in virtually parameter-free model selection that naturally allows to trade-off model complexity with goodness of fit, by which overfitting and the need for hyperparameter tuning are effectively avoided. Finally, we introduce the Classy algorithm, which greedily finds rule lists according to the proposed criterion. We empirically demonstrate that Classy selects small probabilistic rule lists that outperform state-of-the-art classifiers when it comes to the combination of predictive performance and interpretability. We show that Classy is insensitive to its only parameter, i.e., the candidate set, and that compression on the training set correlates with classification performance, validating our MDL-based selection criterion.

Hugo M. Proen\c{c}a, Matthijs van Leeuwen• 2019

Related benchmarks

TaskDatasetResultRank
Binary ClassificationDiabetes
AUC0.726
34
Multi-class classificationYeast--
20
Binary ClassificationHeart
Mean AUC87.49
17
Binary ClassificationElectricity
AUC88.95
12
Binary Classificationjungle
AUC94.76
10
Binary ClassificationBank
AUC90.42
10
Binary Classificationcalhousing
AUC90.72
10
Binary Classificationcc default
AUC0.7631
10
Binary ClassificationCOMPAS
AUC70.8
10
Binary Classificationblood
AUC66.28
10
Showing 10 of 20 rows

Other info

Follow for update