Enhancing Interpretability for Vision Models via Shapley Value Optimization
Preprint
Abstract
Deep neural networks have demonstrated remarkable performance across various domains, yet their decision-making processes remain opaque. Although many explanation methods are dedicated to bringing the obscurity of DNNs to light, they exhibit significant limitations: post-hoc explanation methods often struggle to faithfully reflect model behaviors, while self-explaining neural networks sacrifice performance and compatibility due to their specialized architectural designs. To address these challenges, we propose a novel self-explaining framework that integrates Shapley value estimation as an auxiliary task during training, which achieves two key advancements: 1) a fair allocation of the model prediction scores to image patches, ensuring explanations inherently align with the model's decision logic, and 2) enhanced interpretability with minor structural modifications, preserving model performance and compatibility. Extensive experiments on multiple benchmarks demonstrate that our method achieves state-of-the-art interpretability.Keywords
All Related Versions
This publication has 9 references indexed in Scilit:
- Benchmarking the Attribution Quality of Vision Models2024
- Negative Flux Aggregation to Estimate Feature Attributions2023
- ICAM-Reg: Interpretable Classification and Regression With Feature Attribution for Mapping Neurological Phenotypes in Individual ScansIEEE Transactions on Medical Imaging, 2023
- Large-Scale Unsupervised Semantic SegmentationIEEE Transactions on Pattern Analysis and Machine Intelligence, 2022
- Generating Hierarchical Explanations on Text Classification via Feature Interaction Detection2020
- Quantifying Attention Flow in Transformers2020
- Flickr30k Entities: Collecting Region-to-Phrase Correspondences for Richer Image-to-Sentence Models2015
- Microsoft COCO: Common Objects in Context2014
- Polynomial calculation of the Shapley value based on samplingComputers & Operations Research, 2009