KDD2021

ProtoPShare: Prototypical Parts Sharing for Similarity Discovery in Interpretable Image Classification

Dawid Rymarczyk, Lukasz Struski, Jacek Tabor, Bartosz Zielinski

被引用 78 次

摘要

In this work, we introduce an extension to ProtoPNet called ProtoPShare which shares prototypical parts between classes. To obtain prototype sharing we prune prototypical parts using a novel data-dependent similarity. Our approach substantially reduces the number of prototypes needed to preserve baseline accuracy and finds prototypical similarities between classes. We show the effectiveness of ProtoPShare on the CUB-200-2011 and the Stanford Cars datasets and confirm the semantic consistency of its prototypical parts in user-study.