On the emergence of simplex symmetry in the final and penultimate layers of neural network classifiers

A recent numerical study observed that neural network classifiers enjoy a large degree of symmetry in the penultimate layer. Namely, if h(x)=Af(x)+b where A is a linear map and f is the output of the penultimate layer of the network (after activation), then all data points xi,1,…,xi,Ni in a class Ci are mapped to a single point yi by f and the points yi are located at the vertices of a regular k−1-dimensional standard simplex in a high-dimensional Euclidean space. We explain this observation analytically in toy models for highly expressive deep neural networks. In complementary examples, we demonstrate rigorously that even the final output of the classifier h is not uniform over data samples from a class Ci if h is a shallow network (or if the deeper layers do not bring the data samples into a convenient geometric configuration).

https://arxiv.org/abs/2012.05420