On the Symmetries of Deep Learning Models and their Internal Representations
Abstract
Symmetry is a fundamental tool in the exploration of a broad range of complex systems. In machine learning symmetry has been explored in both models and data. In this paper we seek to connect the symmetries arising from the architecture of a family of models with the symmetries of that family's internal representation of data. We do this by calculating a set of fundamental symmetry groups, which we call the intertwiner groups of the model. We connect intertwiner groups to a model's internal representations of data through a range of experiments that probe similarities between hidden states across models with the same architecture. Our work suggests that the symmetries of a network are propagated into the symmetries in that network's representation of data, providing us with a better understanding of how architecture affects the learning and prediction process. Finally, we speculate that for ReLU networks, the intertwiner groups may provide a justification for the common practice of concentrating model interpretability exploration on the activation basis in hidden layers rather than arbitrary linear combinations thereof.
Keywords
Cite
@article{arxiv.2205.14258,
title = {On the Symmetries of Deep Learning Models and their Internal Representations},
author = {Charles Godfrey and Davis Brown and Tegan Emerson and Henry Kvinge},
journal= {arXiv preprint arXiv:2205.14258},
year = {2023}
}
Comments
CG and DB contributed equally. V2: clarified relationship between $\mu_{\mathrm{CKA}}$ and existing instances of CKA. V3: more experiments, alternative stitching capacity comparison, GeLU intertwiner group. V4: minor typo corrections. V4: failure of PSD property for max kernel used in $\mu_{\mathrm{CKA}}$ (thanks to Derek Lim)