Unlocking Feature Visualization for Deeper Networks with MAgnitude Constrained Optimization

Thomas Fel,Thibaut Boissin,Victor Boutin,Agustin Picard,Paul Novello,Julien Colin,Drew Linsley,Tom Rousseau,Rémi Cadène,Laurent Gardes,Thomas Serre
DOI: https://doi.org/10.48550/arXiv.2306.06805
2023-10-30
Abstract:Feature visualization has gained substantial popularity, particularly after the influential work by Olah et al. in 2017, which established it as a crucial tool for explainability. However, its widespread adoption has been limited due to a reliance on tricks to generate interpretable images, and corresponding challenges in scaling it to deeper neural networks. Here, we describe MACO, a simple approach to address these shortcomings. The main idea is to generate images by optimizing the phase spectrum while keeping the magnitude constant to ensure that generated explanations lie in the space of natural images. Our approach yields significantly better results (both qualitatively and quantitatively) and unlocks efficient and interpretable feature visualizations for large state-of-the-art neural networks. We also show that our approach exhibits an attribution mechanism allowing us to augment feature visualizations with spatial importance. We validate our method on a novel benchmark for comparing feature visualization methods, and release its visualizations for all classes of the ImageNet dataset on <a class="link-external link-https" href="https://serre-lab.github.io/Lens/" rel="external noopener nofollow">this https URL</a>. Overall, our approach unlocks, for the first time, feature visualizations for large, state-of-the-art deep neural networks without resorting to any parametric prior image model.
Computer Vision and Pattern Recognition,Artificial Intelligence
What problem does this paper attempt to address?