Neural collapse under cross-entropy loss

Authors

Lu, J; Steinerberger, S

Abstract

We consider the variational problem of cross-entropy loss with n feature vectors on a unit hypersphere in Rd. We prove that when d≥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→∞ with fixed d, the minimizing points will distribute uniformly on the hypersphere and show a connection with the frame potential of Benedetto & Fickus.

Citation

Lu, J., and S. Steinerberger. “Neural collapse under cross-entropy loss.” Applied and Computational Harmonic Analysis 59 (July 1, 2022): 224–41. https://doi.org/10.1016/j.acha.2021.12.011.

Publication Links