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

Multi-Depth Concept Extraction for Post-Hoc Vision Encoder Explanation

About

Explainable AI methods for vision models aim to identify the parts of the input that are important for the final prediction and subsequently relate these regions to human-understandable concepts. Here, we propose focusing solely on the encoder and relating its intermediate outputs to the input, instead. We introduce Neuro-Activated Vision Explanations (NAVE), a post-hoc, unsupervised, and architecture-agnostic (across CNNs and ViTs) method for extracting and visualizing internal representations from frozen vision model encoders. Specifically, NAVE clusters composite feature activations from multiple encoder depths to produce interpretable segmentation maps with controllable granularity, requiring no fine-tuning or architectural modifications. Through extensive experiments, we quantitatively demonstrate that NAVE's concepts align with input semantics and can be used in downstream tasks. We further demonstrate NAVE as an inspection tool by analyzing how training strategies and architectures affect encoder representations. Overall, our results establish NAVE as an effective tool for post-hoc model inspection and enhancing transparency in vision models. \texttt{https://github.com/Ahcene-B/NAVE}

Ahc\`ene Boubekki, Samuel G. Fadel, Sebastian Mair• 2024

Related benchmarks

TaskDatasetResultRank
Image ClassificationCUB-200 (test)
Accuracy66
120
ClassificationMNIST (test)
Accuracy99.3
93
Object LocalizationVOC 07
AP@50%63.5
12
Object LocalizationVOC 12
AP@50%65
12
ClassificationSTL10 (test)
Robust Accuracy (RA)89.5
11
Unsupervised SegmentationCOCO 20k (subset)
Accuracy93.5
7
Object LocalizationCOCO 20k
AP@50%61
6
Showing 7 of 7 rows

Other info

Follow for update