Visualization of Supervised and Self-Supervised Neural Networks via Attribution Guided Factorization

By Shir Gur, Ameen Ali, Lior Wolf. 2020

43 citations recorded by the source corpus.

Neural network visualization techniques mark image locations by their relevancy to the network's classification. Existing methods are effective in highlighting the regions that affect the resulting classification the most. However, as we show, these methods are limited in their ability to identify the support for alternative classifications, an effect we name the saliency bias hypothesis. In this work, we integrate two lines of research: gradient-based methods and attribution-based methods, and develop an algorithm that provides per-class explainability. The algorithm back-projects the per pixel local influence, in a manner that is guided by the local attributions, while correcting for salient features that would otherwise bias the explanation. In an extensive battery of experiments, we demonstrate the ability of our methods to class-specific visualization, and not just the predicted label. Remarkably, the method obtains state of the art results in benchmarks that are commonly applied to gradient-based methods as well as in those that are employed mostly for evaluating attribution methods. Using a new unsupervised procedure, our method is also successful in demonstrating that self-supervised methods learn semantic information. Our code is available at: https://github.com/shirgur/AGFVisualization.

Subfields: Computer Vision and Pattern Recognition, Safety, Robustness & Interpretability

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Dacheng Tao
  3. Andrew Zisserman
  4. Trevor Darrell
  5. Andrea Vedaldi
  6. Ming-Hsuan Yang
  7. Cordelia Schmid
  8. Xiaogang Wang

In Safety, Robustness & Interpretability

  1. Hila Chefer
  2. Julien Mairal
  3. Lior Wolf
  4. Shir Gur
  5. Herv'e J'egou
  6. Ishan Misra
  7. Armand Joulin
  8. Piotr Bojanowski

Citations in the corpus