arXiv · 2409.10329
InfoDisent: Explainability of Image Classification Models by Information Disentanglement
Abstract
In this work, we introduce InfoDisent, a hybrid approach to explainability based on the information bottleneck principle. InfoDisent enables the disentanglement of information in the final layer of any pretrained model into atomic concepts, which can be interpreted as prototypical parts. This approach merges the flexibility of post-hoc methods with the concept-level modeling capabilities of self-explainable neural networks, such as ProtoPNets. We demonstrate the effectiveness of InfoDisent through computational experiments and user studies across various datasets using modern backbones such as ViTs and convolutional networks. Notably, InfoDisent generalizes the prototypical parts approach to novel domains (ImageNet).
Explore related subjects
Keep this discovery
Łukasz Struski, Dawid Rymarczyk, Jacek Tabor. 2024-09-16. InfoDisent: Explainability of Image Classification Models by Information Disentanglement. https://arxiv.org/abs/2409.10329
Cite the original work for its findings. Save a collection to share your selection of sources.