English
Related papers

Related papers: Neural collapse in the orthoplex regime

200 papers

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…

Machine Learning · Computer Science 2021-01-20 Jianfeng Lu , Stefan Steinerberger

There is a recently discovered and intriguing phenomenon called Neural Collapse: at the terminal phase of training a deep neural network for classification, the within-class penultimate feature means and the associated classifier vectors of…

Computer Vision and Pattern Recognition · Computer Science 2023-08-10 Tong Liang , Jim Davis

Imposing orthogonality on the layers of neural networks is known to facilitate the learning by limiting the exploding/vanishing of the gradient; decorrelate the features; improve the robustness. This paper studies the theoretical properties…

Statistics Theory · Mathematics 2023-01-16 El Mehdi Achour , François Malgouyres , Franck Mamalet

Neural Collapse (NC) is a recently observed phenomenon in neural networks that characterises the solution space of the final classifier layer when trained until zero training loss. Specifically, NC suggests that the final classifier layer…

Machine Learning · Computer Science 2024-11-05 Evan Markou , Thalaiyasingam Ajanthan , Stephen Gould

This work bridges two important concepts: the Neural Tangent Kernel (NTK), which captures the evolution of deep neural networks (DNNs) during training, and the Neural Collapse (NC) phenomenon, which refers to the emergence of symmetry and…

Machine Learning · Computer Science 2023-11-07 Mariia Seleznova , Dana Weitzner , Raja Giryes , Gitta Kutyniok , Hung-Hsu Chou

Neural collapse (NC) refers to the surprising structure of the last layer of deep neural networks in the terminal phase of gradient descent training. Recently, an increasing amount of experimental evidence has pointed to the propagation of…

Machine Learning · Computer Science 2023-05-23 Peter Súkeník , Marco Mondelli , Christoph Lampert

Graph Neural Networks (GNNs) are currently dominating in modeling graph-structure data, while their high reliance on graph structure for inference significantly impedes them from widespread applications. By contrast, Graph-regularized MLPs…

When training deep neural networks for classification tasks, an intriguing empirical phenomenon has been widely observed in the last-layer classifiers and features, where (i) the class means and the last-layer classifiers all collapse to…

Machine Learning · Computer Science 2022-03-15 Jinxin Zhou , Xiao Li , Tianyu Ding , Chong You , Qing Qu , Zhihui Zhu

Deep learning methods operate in regimes that defy the traditional statistical mindset. Neural network architectures often contain more parameters than training samples, and are so rich that they can interpolate the observed labels, even if…

Statistics Theory · Mathematics 2020-12-14 Song Mei , Andrea Montanari

Modern neural translation models based on the Transformer architecture are known for their high performance, particularly when trained on high-resource datasets. A standard next-token prediction training strategy, while widely adopted in…

Computation and Language · Computer Science 2026-02-20 Evgeniia Tokarchuk , Maya K. Nachesa , Sergey Troshin , Vlad Niculae

Neural Collapse (NC) gives a precise description of the representations of classes in the final hidden layer of classification neural networks. This description provides insights into how these networks learn features and generalize well…

Machine Learning · Computer Science 2023-08-08 Liam Parker , Emre Onal , Anton Stengel , Jake Intrater

Neural Collapse (NC) presents an elegant geometric structure that enables individual activations (features), class means and classifier (weights) vectors to reach \textit{optimal} inter-class separability during the terminal phase of…

Machine Learning · Computer Science 2024-08-15 Enhao Zhang , Chaohua Li , Chuanxing Geng , Songcan Chen

We study regularized deep neural networks (DNNs) and introduce a convex analytic framework to characterize the structure of the hidden layers. We show that a set of optimal hidden layer weights for a norm regularized DNN training problem…

Machine Learning · Computer Science 2021-06-14 Tolga Ergen , Mert Pilanci

We describe the class of convexified convolutional neural networks (CCNNs), which capture the parameter sharing of convolutional neural networks in a convex manner. By representing the nonlinear convolutional filters as vectors in a…

Machine Learning · Computer Science 2016-09-06 Yuchen Zhang , Percy Liang , Martin J. Wainwright

The modern strategy for training deep neural networks for classification tasks includes optimizing the network's weights even after the training error vanishes to further push the training loss toward zero. Recently, a phenomenon termed…

Machine Learning · Computer Science 2022-10-13 Tom Tirer , Joan Bruna

Catastrophic forgetting is a major problem in continual learning, and lots of approaches arise to reduce it. However, most of them are evaluated through task accuracy, which ignores the internal model structure. Recent research suggests…

Machine Learning · Computer Science 2026-03-06 Yunqin Zhu , Jun Jin

We analyze a new approach to Machine Learning coming from a modification of classical regularization networks by casting the process in the time dimension, leading to a sort of collapse of dimensionality in the problem of learning the model…

Machine Learning · Computer Science 2017-01-05 Marco Gori , Marco Maggini , Alessandro Rossi

In self-supervised representation learning, a common idea behind most of the state-of-the-art approaches is to enforce the robustness of the representations to predefined augmentations. A potential issue of this idea is the existence of…

Machine Learning · Computer Science 2021-08-26 Tianyu Hua , Wenxiao Wang , Zihui Xue , Sucheng Ren , Yue Wang , Hang Zhao

In the era of proliferation of large language and image generation models, the phenomenon of "model collapse" refers to the situation whereby as a model is trained recursively on data generated from previous generations of itself over time,…

Machine Learning · Computer Science 2024-05-02 Elvis Dohmatob , Yunzhen Feng , Julia Kempe

In general, approximating classes of functions defined over high-dimensional input spaces by linear combinations of a fixed set of basis functions or ``features'' is known to be hard. Typically, the worst-case error of the best basis set…

Machine Learning · Computer Science 2025-12-05 Jonathan Baxter , Peter Bartlett