AIセキュリティポータル K Program
On the explainable properties of 1-Lipschitz Neural Networks: An Optimal Transport Perspective
Share
Abstract
Input gradients have a pivotal role in a variety of applications, including adversarial attack algorithms for evaluating model robustness, explainable AI techniques for generating Saliency Maps, and counterfactual explanations.However, Saliency Maps generated by traditional neural networks are often noisy and provide limited insights. In this paper, we demonstrate that, on the contrary, the Saliency Maps of 1-Lipschitz neural networks, learned with the dual loss of an optimal transportation problem, exhibit desirable XAI properties:They are highly concentrated on the essential parts of the image with low noise, significantly outperforming state-of-the-art explanation approaches across various models and metrics. We also prove that these maps align unprecedentedly well with human explanations on ImageNet.To explain the particularly beneficial properties of the Saliency Map for such models, we prove this gradient encodes both the direction of the transportation plan and the direction towards the nearest adversarial attack. Following the gradient down to the decision boundary is no longer considered an adversarial attack, but rather a counterfactual explanation that explicitly transports the input from one class to another. Thus, Learning with such a loss jointly optimizes the classification objective and the alignment of the gradient, i.e. the Saliency Map, to the transportation plan direction.These networks were previously known to be certifiably robust by design, and we demonstrate that they scale well for large problems and models, and are tailored for explainability using a fast and straightforward method.
Existence, stability and scalability of orthogonal convolutional neural networks
E. M. Achour, F. Malgouyres, F. Mamalet
Published: 2021
Sanity checks for saliency maps
J. Adebayo, J. Gilmer, M. Muelly, I. Goodfellow, M. Hardt, B. Kim
Published: 2018
Existence and stability results in the L1 theory of optimal transportation
L. Ambrosio, A. Pratelli
Published: 2003
A unified view of gradient-based attribution methods for deep neural networks
M. Ancona, E. Ceolini, A. C. Öztireli, M. H. Gross
Published: 2017
Sorting out Lipschitz function approximation
C. Anil, J. Lucas, R. Grosse
Published: 2019
A unified algebraic perspective on lipschitz neural networks
A. Araujo, A. J. Havens, B. Delattre, A. Allauzen, B. Hu
Published: 2023
Wasserstein generative adversarial networks
M. Arjovsky, S. Chintala, L. Bottou
Evaluating and aggregating feature-based model explanations
U. Bhatt, A. Weller, J. M. F. Moura
Published: 2020
An Iterative Algorithm for Computing the Best Estimate of an Orthogonal Matrix
Å. Björck, C. Bowie
Published: 1971
Share