We propose an efficient algorithm to visualise symmetries in neural networks. Typically, models are defined with respect to a parameter space, where non-equal parameters can produce the same input-output map. Our proposed method, GENNI, allows us to efficiently identify parameters that are functionally equivalent and then visualise the subspace of the resulting equivalence class. By doing so, we are now able to better explore questions surrounding identifiability, with applications to optimisation and generalizability, for commonly used or newly developed neural network architectures.
@article{arxiv.2011.07407,
title = {GENNI: Visualising the Geometry of Equivalences for Neural Network Identifiability},
author = {Daniel Lengyel and Janith Petangoda and Isak Falk and Kate Highnam and Michalis Lazarou and Arinbjörn Kolbeinsson and Marc Peter Deisenroth and Nicholas R. Jennings},
journal= {arXiv preprint arXiv:2011.07407},
year = {2020}
}