GAN-based Generation and Automatic Selection of Explanations for Neural Networks
This addresses the slow, manual process of interpreting neural networks for researchers in explainable AI, though it is incremental as it builds on existing optimization methods.
The paper tackles the problem of efficiently generating and selecting interpretable examples for neural network neurons by introducing a new metric using Fréchet Inception Distance to evaluate hyper-parameter settings automatically, avoiding manual evaluation, and applies it to a music classification model to successfully generate vocal and non-vocal examples.
One way to interpret trained deep neural networks (DNNs) is by inspecting characteristics that neurons in the model respond to, such as by iteratively optimising the model input (e.g., an image) to maximally activate specific neurons. However, this requires a careful selection of hyper-parameters to generate interpretable examples for each neuron of interest, and current methods rely on a manual, qualitative evaluation of each setting, which is prohibitively slow. We introduce a new metric that uses Fréchet Inception Distance (FID) to encourage similarity between model activations for real and generated data. This provides an efficient way to evaluate a set of generated examples for each setting of hyper-parameters. We also propose a novel GAN-based method for generating explanations that enables an efficient search through the input space and imposes a strong prior favouring realistic outputs. We apply our approach to a classification model trained to predict whether a music audio recording contains singing voice. Our results suggest that this proposed metric successfully selects hyper-parameters leading to interpretable examples, avoiding the need for manual evaluation. Moreover, we see that examples synthesised to maximise or minimise the predicted probability of singing voice presence exhibit vocal or non-vocal characteristics, respectively, suggesting that our approach is able to generate suitable explanations for understanding concepts learned by a neural network.