A Super-pixel-based Approach to the Stable Interpretation of Neural Networks

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Gong, Shizhan, Zhang, Jingwei, Dou, Qi, Farnia, Farzan
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910752421445632
author Gong, Shizhan
Zhang, Jingwei
Dou, Qi
Farnia, Farzan
author_facet Gong, Shizhan
Zhang, Jingwei
Dou, Qi
Farnia, Farzan
contents Saliency maps are widely used in the computer vision community for interpreting neural network classifiers. However, due to the randomness of training samples and optimization algorithms, the resulting saliency maps suffer from a significant level of stochasticity, making it difficult for domain experts to capture the intrinsic factors that influence the neural network's decision. In this work, we propose a novel pixel partitioning strategy to boost the stability and generalizability of gradient-based saliency maps. Through both theoretical analysis and numerical experiments, we demonstrate that the grouping of pixels reduces the variance of the saliency map and improves the generalization behavior of the interpretation method. Furthermore, we propose a sensible grouping strategy based on super-pixels which cluster pixels into groups that align well with the semantic meaning of the images. We perform several numerical experiments on CIFAR-10 and ImageNet. Our empirical results suggest that the super-pixel-based interpretation maps consistently improve the stability and quality over the pixel-based saliency maps.
format Preprint
id arxiv_https___arxiv_org_abs_2412_14509
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A Super-pixel-based Approach to the Stable Interpretation of Neural Networks
Gong, Shizhan
Zhang, Jingwei
Dou, Qi
Farnia, Farzan
Computer Vision and Pattern Recognition
Saliency maps are widely used in the computer vision community for interpreting neural network classifiers. However, due to the randomness of training samples and optimization algorithms, the resulting saliency maps suffer from a significant level of stochasticity, making it difficult for domain experts to capture the intrinsic factors that influence the neural network's decision. In this work, we propose a novel pixel partitioning strategy to boost the stability and generalizability of gradient-based saliency maps. Through both theoretical analysis and numerical experiments, we demonstrate that the grouping of pixels reduces the variance of the saliency map and improves the generalization behavior of the interpretation method. Furthermore, we propose a sensible grouping strategy based on super-pixels which cluster pixels into groups that align well with the semantic meaning of the images. We perform several numerical experiments on CIFAR-10 and ImageNet. Our empirical results suggest that the super-pixel-based interpretation maps consistently improve the stability and quality over the pixel-based saliency maps.
title A Super-pixel-based Approach to the Stable Interpretation of Neural Networks
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2412.14509