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

A Robust Prototype-Based Network with Interpretable RBF Classifier Foundations

About

Prototype-based classification learning methods are known to be inherently interpretable. However, this paradigm suffers from major limitations compared to deep models, such as lower performance. This led to the development of the so-called deep Prototype-Based Networks (PBNs), also known as prototypical parts models. In this work, we analyze these models with respect to different properties, including interpretability. In particular, we focus on the Classification-by-Components (CBC) approach, which uses a probabilistic model to ensure interpretability and can be used as a shallow or deep architecture. We show that this model has several shortcomings, like creating contradicting explanations. Based on these findings, we propose an extension of CBC that solves these issues. Moreover, we prove that this extension has robustness guarantees and derive a loss that optimizes robustness. Additionally, our analysis shows that most (deep) PBNs are related to (deep) RBF classifiers, which implies that our robustness guarantees generalize to shallow RBF classifiers. The empirical evaluation demonstrates that our deep PBN yields state-of-the-art classification accuracy on different benchmarks while resolving the interpretability shortcomings of other approaches. Further, our shallow PBN variant outperforms other shallow PBNs while being inherently interpretable and exhibiting provable robustness guarantees.

Sascha Saralajew, Ashish Rana, Thomas Villmann, Ammar Shaker• 2024

Related benchmarks

TaskDatasetResultRank
Image ClassificationStanford Cars
Accuracy86.9
635
Image ClassificationCUB-200 2011
Accuracy85.5
356
Fine-grained Image ClassificationCUB-200 2011
Accuracy86
300
Fine-grained Image ClassificationStanford Cars
Accuracy91
284
Interpretability EvaluationCUB-200 2011
Consistency Score69
10
Interpretability EvaluationStanford Cars
Consistency42
10
Showing 6 of 6 rows

Other info

Follow for update