B-Cos Networks: Alignment Is All We Need for Interpretability

Böhle, Moritz and Fritz, Mario and Schiele, Bernt
(2022) B-Cos Networks: Alignment Is All We Need for Interpretability.
In: Conference on Computer Vision and Pattern Recognition, 21-24.06.2022, New Orleans, Louisiena, USA.
Conference: CVPR IEEE Conference on Computer Vision and Pattern Recognition

Full text not available from this repository.
Official URL: https://openaccess.thecvf.com/content/CVPR2022/htm...


We present a new direction for increasing the interpretability of deep neural networks (DNNs) by promoting weight-input alignment during training. For this, we propose to replace the linear transforms in DNNs by our B-cos transform. As we show, a sequence (network) of such transforms induces a single linear transform that faithfully summarises the full model computations. Moreover, the B-cos transform introduces alignment pressure on the weights during optimisation. As a result, those induced linear transforms become highly interpretable and align with task-relevant features. Importantly, the B-cos transform is designed to be compatible with existing architectures and we show that it can easily be integrated into common models such as VGGs, ResNets, InceptionNets, and DenseNets, whilst maintaining similar performance on ImageNet. The resulting explanations are of high visual quality and perform well under quantitative metrics for interpretability. Code available at github.com/moboehle/B-cos.


Actions (login required)

View Item View Item