English

Nearest Class-Center Simplification through Intermediate Layers

Machine Learning 2022-06-14 v2

Abstract

Recent advances in theoretical Deep Learning have introduced geometric properties that occur during training, past the Interpolation Threshold -- where the training error reaches zero. We inquire into the phenomena coined Neural Collapse in the intermediate layers of the networks, and emphasize the innerworkings of Nearest Class-Center Mismatch inside the deepnet. We further show that these processes occur both in vision and language model architectures. Lastly, we propose a Stochastic Variability-Simplification Loss (SVSL) that encourages better geometrical features in intermediate layers, and improves both train metrics and generalization.

Keywords

Cite

@article{arxiv.2201.08924,
  title  = {Nearest Class-Center Simplification through Intermediate Layers},
  author = {Ido Ben-Shaul and Shai Dekel},
  journal= {arXiv preprint arXiv:2201.08924},
  year   = {2022}
}
R2 v1 2026-06-24T08:58:17.112Z