arXiv · 2012.08465
Neural Collapse with Cross-Entropy Loss
Abstract
We consider the variational problem of cross-entropy loss with $n$ feature vectors on a unit hypersphere in $\mathbb{R}^d$. We prove that when $d \geq n - 1$, the global minimum is given by the simplex equiangular tight frame, which justifies the neural collapse behavior. We also prove that as $n \rightarrow \infty$ with fixed $d$, the minimizing points will distribute uniformly on the hypersphere and show a connection with the frame potential of Benedetto & Fickus.
Explore related subjects
Keep this discovery
Jianfeng Lu, Stefan Steinerberger. 2020-12-15. Neural Collapse with Cross-Entropy Loss. https://arxiv.org/abs/2012.08465
Cite the original work for its findings. Save a collection to share your selection of sources.