English

Geometric Analysis of Unconstrained Feature Models with $d=K$

Machine Learning 2024-07-23 v2

Abstract

Recently, interesting empirical phenomena known as Neural Collapse have been observed during the final phase of training deep neural networks for classification tasks. We examine this issue when the feature dimension d is equal to the number of classes K. We demonstrate that two popular unconstrained feature models are strict saddle functions, with every critical point being either a global minimum or a strict saddle point that can be exited using negative curvatures. The primary findings conclusively confirm the conjecture on the unconstrained feature models in previous articles.

Keywords

Cite

@article{arxiv.2407.10702,
  title  = {Geometric Analysis of Unconstrained Feature Models with $d=K$},
  author = {Yi Shen and Shao Gu},
  journal= {arXiv preprint arXiv:2407.10702},
  year   = {2024}
}
R2 v1 2026-06-28T17:41:10.255Z