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

Deformable ProtoPNet: An Interpretable Image Classifier Using Deformable Prototypes

About

We present a deformable prototypical part network (Deformable ProtoPNet), an interpretable image classifier that integrates the power of deep learning and the interpretability of case-based reasoning. This model classifies input images by comparing them with prototypes learned during training, yielding explanations in the form of "this looks like that." However, while previous methods use spatially rigid prototypes, we address this shortcoming by proposing spatially flexible prototypes. Each prototype is made up of several prototypical parts that adaptively change their relative spatial positions depending on the input image. Consequently, a Deformable ProtoPNet can explicitly capture pose variations and context, improving both model accuracy and the richness of explanations provided. Compared to other case-based interpretable models using prototypes, our approach achieves state-of-the-art accuracy and gives an explanation with greater context. The code is available at https://github.com/jdonnelly36/Deformable-ProtoPNet.

Jon Donnelly, Alina Jade Barnett, Chaofan Chen• 2021

Related benchmarks

TaskDatasetResultRank
Image ClassificationCUB-200 2011
Accuracy86.4
356
Image ClassificationCUB-200-2011 (test)
Top-1 Acc85.66
286
Image ClassificationImageNet-1k (val)
Accuracy1.48
199
Image ClassificationCaltech101 (test)
Accuracy97.22
159
Image ClassificationCaltech101
Base Accuracy95.95
129
Fine-grained Image ClassificationStanford Dogs (test)
Accuracy86.5
124
Image ClassificationStanford Dogs (test)
Top-1 Acc86.5
113
Image ClassificationCars (test)
Accuracy88.7
57
ClassificationAWA2 (test)--
22
Image ClassificationAWA2 (unseen)
Accuracy19.71
6
Showing 10 of 10 rows

Other info

Code

Follow for update