English
Related papers

Related papers: Neural Collapse for Cross-entropy Class-Imbalanced…

200 papers

Neural collapse provides an elegant mathematical characterization of learned last layer representations (a.k.a. features) and classifier weights in deep classification models. Such results not only provide insights but also motivate new…

Machine Learning · Computer Science 2023-10-30 Jiachen Jiang , Jinxin Zhou , Peng Wang , Qing Qu , Dustin Mixon , Chong You , Zhihui Zhu

Neural collapse is a highly symmetric geometric pattern of neural networks that emerges during the terminal phase of training, with profound implications on the generalization performance and robustness of the trained networks. To…

Machine Learning · Computer Science 2022-04-26 Wenlong Ji , Yiping Lu , Yiliang Zhang , Zhun Deng , Weijie J. Su

While cross entropy (CE) is the most commonly used loss to train deep neural networks for classification tasks, many alternative losses have been developed to obtain better empirical performance. Among them, which one is the best to use is…

Machine Learning · Computer Science 2022-10-11 Jinxin Zhou , Chong You , Xiao Li , Kangning Liu , Sheng Liu , Qing Qu , Zhihui Zhu

Loss reweighting is a widely used strategy for long-tailed classification, but existing reweighting strategies often rely on heuristics and rarely define a well-specified target. Inspired by Neural Collapse (NC), the ideal simplex…

Machine Learning · Computer Science 2026-05-12 Jinping Wang , Zixin Tong , Zhiwu Xie , Zhiqiang Gao

Recent studies have noted an intriguing phenomenon termed Neural Collapse, that is, when the neural networks establish the right correlation between feature spaces and the training targets, their last-layer features, together with the…

Computer Vision and Pattern Recognition · Computer Science 2025-09-22 Yining Wang , Junjie Sun , Chenyue Wang , Mi Zhang , Min Yang

The empirical emergence of neural collapse -- a surprising symmetry in the feature representations of the training data in the penultimate layer of deep neural networks -- has spurred a line of theoretical research aimed at its…

Machine Learning · Computer Science 2025-05-22 Peter Súkeník , Christoph H. Lampert , Marco Mondelli

Recent findings reveal that over-parameterized deep neural networks, trained beyond zero training-error, exhibit a distinctive structural pattern at the final layer, termed as Neural-collapse (NC). These results indicate that the final…

Machine Learning · Computer Science 2024-03-01 Tina Behnia , Christos Thrampoulidis

Neural collapse (NC) and its multi-layer variant, deep neural collapse (DNC), describe a structured geometry that occurs in the features and weights of trained deep networks. Recent theoretical work by Sukenik et al. using a deep…

Machine Learning · Computer Science 2025-10-07 Connall Garrod , Jonathan P. Keating

Supervised classification has a theoretical optimum, Neural Collapse (NC), yet neither of its two dominant paradigms reaches it in practice. Cross entropy (CE) leaves radial degrees of freedom unconstrained and converges to a degenerate…

Machine Learning · Computer Science 2026-05-22 Panagiotis Koromilas , Theodoros Giannakopoulos , Mihalis A. Nicolaou , Yannis Panagakis

The recent work of Papyan, Han, & Donoho (2020) presented an intriguing "Neural Collapse" phenomenon, showing a structural property of interpolating classifiers in the late stage of training. This opened a rich area of exploration studying…

Machine Learning · Computer Science 2022-02-18 Like Hui , Mikhail Belkin , Preetum Nakkiran

Data heterogeneity is an inherent challenge that hinders the performance of federated learning (FL). Recent studies have identified the biased classifiers of local models as the key bottleneck. Previous attempts have used classifier…

Machine Learning · Computer Science 2023-08-29 Zexi Li , Xinyi Shang , Rui He , Tao Lin , Chao Wu

Neural Collapse refers to the curious phenomenon in the end of training of a neural network, where feature vectors and classification weights converge to a very simple geometrical arrangement (a simplex). While it has been observed…

Machine Learning · Computer Science 2024-11-14 Jingtong Su , Ya Shi Zhang , Nikolaos Tsilivis , Julia Kempe

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…

Machine Learning · Computer Science 2022-06-14 Ido Ben-Shaul , Shai Dekel

Deep neural networks (DNNs) exhibit a surprising structure in their final layer known as neural collapse (NC), and a growing body of works has currently investigated the propagation of neural collapse to earlier layers of DNNs -- a…

Machine Learning · Computer Science 2024-10-22 Peter Súkeník , Marco Mondelli , Christoph Lampert

Incremental learning often encounter challenges such as overfitting to new data and catastrophic forgetting of old data. Existing methods can effectively extend the model for new tasks while freezing the parameters of the old model, but…

Computer Vision and Pattern Recognition · Computer Science 2025-12-29 Chuangxin Zhang , Guangfeng Lin , Enhui Zhao , Kaiyang Liao , Yajun Chen

Continual Learning (CL) seeks to build an agent that can continuously learn a sequence of tasks, where a key challenge, namely Catastrophic Forgetting, persists due to the potential knowledge interference among different tasks. On the other…

Machine Learning · Computer Science 2026-03-10 Zheng Wang , Wanhao Yu , Li Yang , Sen Lin

In this paper, we provide a computable characterization of the geometry of optimal representations in Contrastive Learning (CL) when the classes are imbalanced. When classes are balanced and the representation dimension is greater than the…

Machine Learning · Computer Science 2026-05-13 Thuan Nguyen , Shuchin Aeron , D. Richard Brown , Prakash Ishwar

When training a neural network for classification, the feature vectors of the training set are known to collapse to the vertices of a regular simplex, provided the dimension $d$ of the feature space and the number $n$ of classes satisfies…

Machine Learning · Computer Science 2026-03-24 James Alcala , Rayna Andreeva , Vladimir A. Kobzar , Dustin G. Mixon , Sanghoon Na , Shashank Sule , Yangxinyu Xie

Deep learning has achieved impressive performance across various medical imaging tasks. However, its inherent bias against specific groups hinders its clinical applicability in equitable healthcare systems. A recently discovered phenomenon,…

Computer Vision and Pattern Recognition · Computer Science 2024-11-01 Kaouther Mouheb , Marawan Elbatel , Stefan Klein , Esther E. Bron

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