Fix your classifier: the marginal value of training the last weight layer
Not on the map. Why?
Abstract (arXiv)
Neural networks are commonly used as models for classification for a wide variety of tasks. Typically, a learned affine transformation is placed at the end of such models, yielding a per-class value used for classification. This classifier can have a vast number of parameters, which grows linearly with the number of possible classes, thus requiring increasingly more resources. In this work we argue that this classifier can be fixed, up to a global scale constant, with little or no loss of accuracy for most tasks, allowing memory and computational benefits. Moreover, we show that by initializing the classifier with a Hadamard matrix we can speed up inference as well. We discuss the implications for current understanding of neural network models.
Key facts. Cited by 43 papers in the map (data through 7 Oct 2026). First posted on arXiv 14 Jan 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 13; 2022: 7; 2023: 8; 2024: 6; 2025: 1; 2026: 8.
43 papers.
Sort:
- 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 226
- 2022Inducing Neural Collapse in Imbalanced Learning: Do We Really Need a Learnable Classifier at the End of Deep Neural Network?cited by 100
- 2022Constrained Few-shot Class-incremental Learningcited by 93
- 2021On the Validity of Modeling SGD with Stochastic Differential Equations (SDEs)cited by 73
- 2024nGPT: Normalized Transformer with Representation Learning on the Hyperspherecited by 68
- 2021Dissecting Supervised Contrastive Learningcited by 64
- 2021No One Representation to Rule Them All: Overlapping Features of Training Methodscited by 57
- 2021Self-Supervised Classification Networkcited by 27
- 2021Regular Polytope Networkscited by 24
- 2022No Reason for No Supervision: Improved Generalization in Supervised Modelscited by 23
- 2021On the Periodic Behavior of Neural Network Training with Batch Normalization and Weight Decaycited by 20
- 2022SphereFed: Hyperspherical Federated Learningcited by 19
31 more
- 2021CoReS: Compatible Representations via Stationaritycited by 14
- 2023Neural Collapse Terminus: A Unified Solution for Class Incremental Learning and Its Variantscited by 14
- 2024Guiding Neural Collapse: Optimising Towards the Nearest Simplex Equiangular Tight Framecited by 14
- 2023Inducing Neural Collapse to a Fixed Hierarchy-Aware Frame for Reducing Mistake Severitycited by 13
- 2021Generalized Learning Vector Quantization for Classification in Randomized Neural Networks and Hyperdimensional Computingcited by 12
- 2022Dimensionality Reduced Training by Pruning and Freezing Parts of a Deep Neural Network, a Surveycited by 12
- 2024Towards Continual Learning Desiderata via HSIC-Bottleneck Orthogonalization and Equiangular Embeddingcited by 9
- 2023Engineering the Neural Collapse Geometry of Supervised-Contrastive Losscited by 8
- 2023Reset It and Forget It: Relearning Last-Layer Weights Improves Continual and Transfer Learningcited by 6
- 2021Slot Machines: Discovering Winning Combinations of Random Weights in Neural Networkscited by 5
- 2021Ex uno plures: Splitting One Model into an Ensemble of Subnetworkscited by 5
- 2023Maximally Compact and Separated Features with Regular Polytope Networkscited by 5
- 2023Factorizers for Distributed Sparse Block Codescited by 4
- 2023Pairwise Similarity Learning is SimPLEcited by 4
- 2024Out-of-Distribution Detection with Prototypical Outlier Proxycited by 4
- 2024Hyperspherical Classification with Dynamic Label-to-Prototype Assignmentcited by 3
- 2024Rethinking Decoders for Transformer-based Semantic Segmentation: A Compression Perspectivecited by 3
- 2021LogAvgExp Provides a Principled and Performant Global Pooling Operatorcited by 2
- 2022Reinforcement Learning with Large Action Spaces for Neural Machine Translationcited by 2
- 2023The Role of Codeword-to-Class Assignments in Error-Correcting Codes: An Empirical Studycited by 2
- 2025GIF: Generative Inspiration for Face Recognition at Scalecited by 2
- 2022The Fixed Sub-Center: A Better Way to Capture Data Complexitycited by 1
- 2026What Do Scan-Derived Class Prototypes Add? Disentangling Supervision, Prototype Content and Query Protocol in Recognition over Frozen Foundation Featurescited by 1
- 2021Classifier Crafting: Turn Your ConvNet into a Zero-Shot Learner!cited by 0
- 2026Simplex Deep Linear Discriminant Analysiscited by 0
- 2026RST: certifying and constructing prescribed information in variational autoencoderscited by 0
- 2026Adversarial Attack and Disturbance Detection by Hadamard-Coded Output Representations for Object Detection and Semantic Segmentationcited by 0
- 2026Contrastive-Collapsed Loss for Flexible and Geometrically Optimal Embeddings and Faster Convergencecited by 0
- 2026Federated Lightweight Fine-Tuningcited by 0
- 2026OPAL: Orthonormal Prototype Alignment Learning for Interpretable Image Classificationcited by 0
- 2026SOLO: Pretraining Billion-Parameter Language Models with Shared-Output Local Learningcited by 0
- 2026SOLO: Pretraining Billion-Parameter Language Models with Shared-Output Local Learning28 Sep 2026
- 2026What Do Scan-Derived Class Prototypes Add? Disentangling Supervision, Prototype Content and Query Protocol in Recognition over Frozen Foundation Features3 Sep 2026
- 2026OPAL: Orthonormal Prototype Alignment Learning for Interpretable Image Classification30 Aug 2026
- 2026Federated Lightweight Fine-Tuning20 Jul 2026
- 2026Contrastive-Collapsed Loss for Flexible and Geometrically Optimal Embeddings and Faster Convergence14 Jul 2026
- 2026Adversarial Attack and Disturbance Detection by Hadamard-Coded Output Representations for Object Detection and Semantic Segmentation8 Jun 2026
- 2026RST: certifying and constructing prescribed information in variational autoencoders18 May 2026
- 2026Simplex Deep Linear Discriminant Analysis4 Jan 2026
- 2025GIF: Generative Inspiration for Face Recognition at Scale5 May 2025
- 2024Out-of-Distribution Detection with Prototypical Outlier Proxy22 Dec 2024
- 2024Rethinking Decoders for Transformer-based Semantic Segmentation: A Compression Perspective5 Nov 2024
- 2024Guiding Neural Collapse: Optimising Towards the Nearest Simplex Equiangular Tight Frame2 Nov 2024
31 more
- 2024nGPT: Normalized Transformer with Representation Learning on the Hypersphere1 Oct 2024
- 2024Hyperspherical Classification with Dynamic Label-to-Prototype Assignment25 Mar 2024
- 2024Towards Continual Learning Desiderata via HSIC-Bottleneck Orthogonalization and Equiangular Embedding17 Jan 2024
- 2023Pairwise Similarity Learning is SimPLE13 Oct 2023
- 2023Reset It and Forget It: Relearning Last-Layer Weights Improves Continual and Transfer Learning12 Oct 2023
- 2023Engineering the Neural Collapse Geometry of Supervised-Contrastive Loss2 Oct 2023
- 2023Neural Collapse Terminus: A Unified Solution for Class Incremental Learning and Its Variants3 Aug 2023
- 2023Factorizers for Distributed Sparse Block Codes24 Mar 2023
- 2023Inducing Neural Collapse to a Fixed Hierarchy-Aware Frame for Reducing Mistake Severity10 Mar 2023
- 2023The Role of Codeword-to-Class Assignments in Error-Correcting Codes: An Empirical Study10 Feb 2023
- 2023Maximally Compact and Separated Features with Regular Polytope Networks15 Jan 2023
- 2022Reinforcement Learning with Large Action Spaces for Neural Machine Translation6 Oct 2022
- 2022SphereFed: Hyperspherical Federated Learning19 Jul 2022
- 2022No Reason for No Supervision: Improved Generalization in Supervised Models30 Jun 2022
- 2022Dimensionality Reduced Training by Pruning and Freezing Parts of a Deep Neural Network, a Survey17 May 2022
- 2022Constrained Few-shot Class-incremental Learning30 Mar 2022
- 2022The Fixed Sub-Center: A Better Way to Capture Data Complexity24 Mar 2022
- 2022Inducing Neural Collapse in Imbalanced Learning: Do We Really Need a Learnable Classifier at the End of Deep Neural Network?17 Mar 2022
- 2021CoReS: Compatible Representations via Stationarity15 Nov 2021
- 2021LogAvgExp Provides a Principled and Performant Global Pooling Operator2 Nov 2021
- 2021No One Representation to Rule Them All: Overlapping Features of Training Methods20 Oct 2021
- 2021On the Periodic Behavior of Neural Network Training with Batch Normalization and Weight Decay29 Jun 2021
- 2021Generalized Learning Vector Quantization for Classification in Randomized Neural Networks and Hyperdimensional Computing17 Jun 2021
- 2021Ex uno plures: Splitting One Model into an Ensemble of Subnetworks9 Jun 2021
- 2021A Geometric Analysis of Neural Collapse with Unconstrained Features6 May 2021
- 2021Regular Polytope Networks29 Mar 2021
- 2021Classifier Crafting: Turn Your ConvNet into a Zero-Shot Learner!20 Mar 2021
- 2021Self-Supervised Classification Network19 Mar 2021
- 2021On the Validity of Modeling SGD with Stochastic Differential Equations (SDEs)24 Feb 2021
- 2021Dissecting Supervised Contrastive Learning17 Feb 2021
- 2021Slot Machines: Discovering Winning Combinations of Random Weights in Neural Networks16 Jan 2021
- 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 226
- 2022Inducing Neural Collapse in Imbalanced Learning: Do We Really Need a Learnable Classifier at the End of Deep Neural Network?cited by 100
- 2022Constrained Few-shot Class-incremental Learningcited by 93
- 2021On the Validity of Modeling SGD with Stochastic Differential Equations (SDEs)cited by 73
- 2024nGPT: Normalized Transformer with Representation Learning on the Hyperspherecited by 68
- 2021Dissecting Supervised Contrastive Learningcited by 64
- 2021No One Representation to Rule Them All: Overlapping Features of Training Methodscited by 57
- 2021Self-Supervised Classification Networkcited by 27
- 2021Regular Polytope Networkscited by 24
- 2022No Reason for No Supervision: Improved Generalization in Supervised Modelscited by 23
- 2021On the Periodic Behavior of Neural Network Training with Batch Normalization and Weight Decaycited by 20
- 2022SphereFed: Hyperspherical Federated Learningcited by 19
12 most cited of 43 All 43 (JSON)
- 2026SOLO: Pretraining Billion-Parameter Language Models with Shared-Output Local Learning28 Sep 2026
- 2026What Do Scan-Derived Class Prototypes Add? Disentangling Supervision, Prototype Content and Query Protocol in Recognition over Frozen Foundation Features3 Sep 2026
- 2026OPAL: Orthonormal Prototype Alignment Learning for Interpretable Image Classification30 Aug 2026
- 2026Federated Lightweight Fine-Tuning20 Jul 2026
- 2026Contrastive-Collapsed Loss for Flexible and Geometrically Optimal Embeddings and Faster Convergence14 Jul 2026
- 2026Adversarial Attack and Disturbance Detection by Hadamard-Coded Output Representations for Object Detection and Semantic Segmentation8 Jun 2026
- 2026RST: certifying and constructing prescribed information in variational autoencoders18 May 2026
- 2026Simplex Deep Linear Discriminant Analysis4 Jan 2026
- 2025GIF: Generative Inspiration for Face Recognition at Scale5 May 2025
- 2024Out-of-Distribution Detection with Prototypical Outlier Proxy22 Dec 2024
- 2024Rethinking Decoders for Transformer-based Semantic Segmentation: A Compression Perspective5 Nov 2024
- 2024Guiding Neural Collapse: Optimising Towards the Nearest Simplex Equiangular Tight Frame2 Nov 2024
12 newest of 43 All 43 (JSON)