LucidPPN: Unambiguous Prototypical Parts Network for User-centric Interpretable Computer Vision

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Pach, Mateusz, Rymarczyk, Dawid, Lewandowska, Koryna, Tabor, Jacek, Zieliński, Bartosz
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913359771729920
author Pach, Mateusz
Rymarczyk, Dawid
Lewandowska, Koryna
Tabor, Jacek
Zieliński, Bartosz
author_facet Pach, Mateusz
Rymarczyk, Dawid
Lewandowska, Koryna
Tabor, Jacek
Zieliński, Bartosz
contents Prototypical parts networks combine the power of deep learning with the explainability of case-based reasoning to make accurate, interpretable decisions. They follow the this looks like that reasoning, representing each prototypical part with patches from training images. However, a single image patch comprises multiple visual features, such as color, shape, and texture, making it difficult for users to identify which feature is important to the model. To reduce this ambiguity, we introduce the Lucid Prototypical Parts Network (LucidPPN), a novel prototypical parts network that separates color prototypes from other visual features. Our method employs two reasoning branches: one for non-color visual features, processing grayscale images, and another focusing solely on color information. This separation allows us to clarify whether the model's decisions are based on color, shape, or texture. Additionally, LucidPPN identifies prototypical parts corresponding to semantic parts of classified objects, making comparisons between data classes more intuitive, e.g., when two bird species might differ primarily in belly color. Our experiments demonstrate that the two branches are complementary and together achieve results comparable to baseline methods. More importantly, LucidPPN generates less ambiguous prototypical parts, enhancing user understanding.
format Preprint
id arxiv_https___arxiv_org_abs_2405_14331
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle LucidPPN: Unambiguous Prototypical Parts Network for User-centric Interpretable Computer Vision
Pach, Mateusz
Rymarczyk, Dawid
Lewandowska, Koryna
Tabor, Jacek
Zieliński, Bartosz
Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
Prototypical parts networks combine the power of deep learning with the explainability of case-based reasoning to make accurate, interpretable decisions. They follow the this looks like that reasoning, representing each prototypical part with patches from training images. However, a single image patch comprises multiple visual features, such as color, shape, and texture, making it difficult for users to identify which feature is important to the model. To reduce this ambiguity, we introduce the Lucid Prototypical Parts Network (LucidPPN), a novel prototypical parts network that separates color prototypes from other visual features. Our method employs two reasoning branches: one for non-color visual features, processing grayscale images, and another focusing solely on color information. This separation allows us to clarify whether the model's decisions are based on color, shape, or texture. Additionally, LucidPPN identifies prototypical parts corresponding to semantic parts of classified objects, making comparisons between data classes more intuitive, e.g., when two bird species might differ primarily in belly color. Our experiments demonstrate that the two branches are complementary and together achieve results comparable to baseline methods. More importantly, LucidPPN generates less ambiguous prototypical parts, enhancing user understanding.
title LucidPPN: Unambiguous Prototypical Parts Network for User-centric Interpretable Computer Vision
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2405.14331