Saved in:
Bibliographic Details
Main Authors: Shu, Kaixiang, Meng, Kai, Luo, Junqin
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2511.09239
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912704495616000
author Shu, Kaixiang
Meng, Kai
Luo, Junqin
author_facet Shu, Kaixiang
Meng, Kai
Luo, Junqin
contents Deep neural networks typically learn spatially entangled representations that conflate discriminative foreground features with spurious background correlations, thereby undermining model interpretability and robustness. We propose a novel understanding framework for gradient-based attribution from an information-theoretic perspective. We prove that, under mild conditions, the Vector-Jacobian Products (VJP) computed during backpropagation form minimal sufficient statistics of input features with respect to class labels. Motivated by this finding, we propose an encoding-decoding perspective : forward propagation encodes inputs into class space, while VJP in backpropagation decodes this encoding back to feature space. Therefore, we propose Spatial Information Bottleneck (S-IB) to spatially disentangle information flow. By maximizing mutual information between foreground VJP and inputs while minimizing mutual information in background regions, S-IB encourages networks to encode information only in class-relevant spatial regions. Since post-hoc explanation methods fundamentally derive from VJP computations, directly optimizing VJP's spatial structure during training improves visualization quality across diverse explanation paradigms. Experiments on five benchmarks demonstrate universal improvements across six explanation methods, achieving better foreground concentration and background suppression without method-specific tuning, alongside consistent classification accuracy gains.
format Preprint
id arxiv_https___arxiv_org_abs_2511_09239
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Spatial Information Bottleneck for Interpretable Visual Recognition
Shu, Kaixiang
Meng, Kai
Luo, Junqin
Computer Vision and Pattern Recognition
Deep neural networks typically learn spatially entangled representations that conflate discriminative foreground features with spurious background correlations, thereby undermining model interpretability and robustness. We propose a novel understanding framework for gradient-based attribution from an information-theoretic perspective. We prove that, under mild conditions, the Vector-Jacobian Products (VJP) computed during backpropagation form minimal sufficient statistics of input features with respect to class labels. Motivated by this finding, we propose an encoding-decoding perspective : forward propagation encodes inputs into class space, while VJP in backpropagation decodes this encoding back to feature space. Therefore, we propose Spatial Information Bottleneck (S-IB) to spatially disentangle information flow. By maximizing mutual information between foreground VJP and inputs while minimizing mutual information in background regions, S-IB encourages networks to encode information only in class-relevant spatial regions. Since post-hoc explanation methods fundamentally derive from VJP computations, directly optimizing VJP's spatial structure during training improves visualization quality across diverse explanation paradigms. Experiments on five benchmarks demonstrate universal improvements across six explanation methods, achieving better foreground concentration and background suppression without method-specific tuning, alongside consistent classification accuracy gains.
title Spatial Information Bottleneck for Interpretable Visual Recognition
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2511.09239