No Fear of Classifier Biases: Neural Collapse Inspired Federated Learning with Synthetic and Fixed Classifier
Not on the map. Why?
Abstract (arXiv)
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 calibration after FL training, but this approach falls short in improving the poor feature representations caused by training-time classifier biases. Resolving the classifier bias dilemma in FL requires a full understanding of the mechanisms behind the classifier. Recent advances in neural collapse have shown that the classifiers and feature prototypes under perfect training scenarios collapse into an optimal structure called simplex equiangular tight frame (ETF). Building on this neural collapse insight, we propose a solution to the FL's classifier bias problem by utilizing a synthetic and fixed ETF classifier during training. The optimal classifier structure enables all clients to learn unified and optimal feature representations even under extremely heterogeneous data. We devise several effective modules to better adapt the ETF structure in FL, achieving both high generalization and personalization. Extensive experiments demonstrate that our method achieves state-of-the-art performances on CIFAR-10, CIFAR-100, and Tiny-ImageNet.
Key facts. Cited by 55 papers in the map (data through 13 Oct 2026). Builds on 41 of its 58 references that are in the map. First posted on arXiv 17 Mar 2023.
Builds on
41 of 58 references.
References · 41
- 2015Deep Residual Learning for Image Recognitioncited by 51,950
- 2015Distilling the Knowledge in a Neural Networkcited by 10,180
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,653
- 2016Densely Connected Convolutional Networkscited by 6,707
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,426
- 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,786
- 2018Federated Optimization in Heterogeneous Networkscited by 2,981
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,129
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 716
- 2020Ensemble Distillation for Robust Model Fusion in Federated Learningcited by 682
- 2020Ditto: Fair and Robust Federated Learning Through Personalizationcited by 668
- 2021Federated Learning Based on Dynamic Regularizationcited by 559
29 more
- 2021Exploiting Shared Representations for Personalized Federated Learningcited by 539
- 2020An Efficient Framework for Clustered Federated Learningcited by 508
- 2020Balanced Meta-Softmax for Long-Tailed Visual Recognitioncited by 399
- 2021FedProto: Federated Prototype Learning across Heterogeneous Clientscited by 352
- 2021Federated Learning for Internet of Things: A Comprehensive Surveycited by 251
- 2021No Fear of Heterogeneity: Classifier Calibration for Federated Learning with Non-IID Datacited by 241
- 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 226
- 2020Federated Learning for Internet of Things: Recent Advances, Taxonomy, and Open Challengescited by 139
- 2022FedALA: Adaptive Local Aggregation for Personalized Federated Learningcited by 122
- 2023Neural Collapse Inspired Feature-Classifier Alignment for Few-Shot Class Incremental Learningcited by 115
- 2022Federated Learning with Label Distribution Skew via Logits Calibrationcited by 109
- 2022Extended Unconstrained Features Model for Exploring Deep Neural Collapsecited by 102
- 2021DENSE: Data-Free One-Shot Federated Learningcited by 100
- 2022Neural Collapse: A Review on Modelling Principles and Generalizationcited by 95
- 2019Reliable Federated Learning for Mobile Networkscited by 92
- 2021An Unconstrained Layer-Peeled Perspective on Neural Collapsecited by 87
- 2022Imbalance Trouble: Revisiting Neural-Collapse Geometrycited by 80
- 2022Tackling Data Heterogeneity in Federated Learning with Class Prototypescited by 67
- 2022Federated Learning on Heterogeneous and Long-Tailed Data via Classifier Re-Training with Federated Featurescited by 63
- 2021What Do We Mean by Generalization in Federated Learning?cited by 62
- 2023Revisiting Weighted Aggregation in Federated Learning with Neural Networkscited by 58
- 2022FedFA: Federated Learning with Feature Anchors to Align Features and Classifiers for Heterogeneous Datacited by 50
- 2022Neural Collapse Inspired Attraction-Repulsion-Balanced Loss for Imbalanced Learningcited by 43
- 2020Training Keyword Spotting Models on Non-IID Data with Federated Learningcited by 36
- 2022FedFM: Anchor-based Feature Matching for Data Heterogeneity in Federated Learningcited by 27
- 2022Towards Effective Clustered Federated Learning: A Peer-to-peer Framework with Adaptive Neighbor Matchingcited by 24
- 2022Understanding and Improving Transfer Learning of Deep Models via Neural Collapsecited by 24
- 2022Auto-FedRL: Federated Hyperparameter Optimization for Multi-institutional Medical Image Segmentationcited by 16
- 2022On the effectiveness of partial variance reduction in federated learning with heterogeneous datacited by 9
Built on by
Citing papers per year: 2023: 7; 2024: 25; 2025: 17; 2026: 6.
55 papers.
Sort:
By area:
- 2024FedTGP: Trainable Global Prototypes with Adaptive-Margin-Enhanced Contrastive Learning for Data and Model Heterogeneity in Federated Learningcited by 47
- 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Modelscited by 37
- 2024FedLLM-Bench: Realistic Benchmarks for Federated Learning of Large Language Modelscited by 30
- 2024Fair Federated Learning under Domain Skew with Local Consistency and Domain Diversitycited by 29
- 2025FedAWA: Adaptive Optimization of Aggregation Weights in Federated Learning Using Client Vectorscited by 19
- 2024Emerging Safety Attack and Defense in Federated Instruction Tuning of Large Language Modelscited by 18
- 2024An Upload-Efficient Scheme for Transferring Knowledge From a Server-Side Pre-trained Generator to Clients in Heterogeneous Federated Learningcited by 17
- 2024Neural Collapse versus Low-rank Bias: Is Deep Neural Collapse Really Optimal?cited by 17
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapsecited by 16
- 2024FedImpro: Measuring and Improving Client Update in Federated Learningcited by 15
- 2023FedSOL: Stabilized Orthogonal Learning with Proximal Restrictions in Federated Learningcited by 14
38 more
- 2024Navigate Beyond Shortcuts: Debiased Learning through the Lens of Neural Collapsecited by 13
- 2025Geometric Knowledge-Guided Localized Global Distribution Alignment for Federated Learningcited by 12
- 2024FuseFL: One-Shot Federated Learning through the Lens of Causality with Progressive Model Fusioncited by 11
- 2024FedLoGe: Joint Local and Generic Federated Learning under Long-tailed Datacited by 10
- 2024Accelerating Heterogeneous Federated Learning with Closed-form Classifierscited by 10
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformerscited by 9
- 2023PRIOR: Personalized Prior for Reactivating the Information Overlooked in Federated Learningcited by 8
- 2023FedFN: Feature Normalization for Alleviating Data Heterogeneity Problem in Federated Learningcited by 8
- 2024Improving Model Fusion by Training-time Neuron Alignment with Fixed Neuron Anchorscited by 7
- 2024Mamba-FSCIL: Dynamic Adaptation with Selective State Space Model for Few-Shot Class-Incremental Learningcited by 7
- 2025FedLWS: Federated Learning with Adaptive Layer-wise Weight Shrinkingcited by 7
- 2025Subgraph Federated Learning for Local Generalizationcited by 6
- 2023FediOS: Decoupling Orthogonal Subspaces for Personalization in Feature-skew Federated Learningcited by 5
- 2025You Are Your Own Best Teacher: Achieving Centralized-level Performance in Federated Learning under Heterogeneous and Long-tailed Datacited by 5
- 2023Pairwise Similarity Learning is SimPLEcited by 4
- 2024Covariances for Free: Exploiting Mean Distributions for Training-free Federated Learningcited by 3
- 2023Understanding Prompt Tuning for V-L Models Through the Lens of Neural Collapsecited by 2
- 2024FedDr+: Stabilizing Dot-regression with Global Feature Distillation for Federated Learningcited by 2
- 2024Tackling Feature-Classifier Mismatch in Federated Learning via Prompt-Driven Feature Transformationcited by 2
- 2024DualFed: Enjoying both Generalization and Personalization in Federated Learning via Hierachical Representationscited by 2
- 2025FedBM: Stealing Knowledge from Pre-trained Language Models for Heterogeneous Federated Learningcited by 2
- 2025Enhancing Visual Representation with Textual Semantics: Textual Semantics-Powered Prototypes for Heterogeneous Federated Learningcited by 2
- 2025Federated Adapter on Foundation Models: An Out-Of-Distribution Approachcited by 2
- 2025Neural Collapse-Inspired Multi-Label Federated Learning under Label-Distribution Skewcited by 2
- 2024Evaluating the Fairness of Neural Collapse in Medical Image Classificationcited by 1
- 2024Collapsed Language Models Promote Fairnesscited by 1
- 2024TPFL: A Trustworthy Personalized Federated Learning Framework via Subjective Logiccited by 1
- 2025FedORGP: Guiding Heterogeneous Federated Learning with Orthogonality Regularization on Global Prototypescited by 1
- 2025FedDEAP: Adaptive Dual-Prompt Tuning for Multi-Domain Federated Learningcited by 1
- 2025FedRE: A Representation Entanglement Framework for Model-Heterogeneous Federated Learningcited by 1
- 2026GFPL: Generative Federated Prototype Learning for Resource-Constrained and Data-Imbalanced Vision Taskcited by 1
- 2024Fixed Random Classifier Rearrangement for Continual Learningcited by 0
- 2024FedGuCci: Making Local Models More Connected in Landscape for Federated Learningcited by 0
- 2024Client2Vec: Improving Federated Learning by Distribution Shifts Aware Client Indexingcited by 0
- 2025Knowledge-Driven Federated Graph Learning on Model Heterogeneitycited by 0
- 2025Calibrating Biased Distribution in VFM-derived Latent Space via Cross-Domain Geometric Consistencycited by 0
- 2025Rethinking the Backbone in Class Imbalanced Federated Source Free Domain Adaptation: The Utility of Vision Foundation Modelscited by 0
- 2025From Coefficients to Directions: Rethinking Model Merging with Directional Alignmentcited by 0
50 most cited of 55 All 55 (JSON)
- 2026Understanding Head Geometry and Dynamics in Federated Regression through a Natural Solution Selection Rule: An Unconstrained Feature Model Analysis30 Sep 2026
- 2026FedLTLib: A Comprehensive Benchmark for Federated Long-Tail Learning14 Sep 2026
- 2026WHERE to Generate Matters: Budget-Aware Synthetic Augmentation for Label Skewed Federated Learning7 Jul 2026
- 2026FedReLa: Imbalanced Federated Learning via Re-Labeling24 Jun 2026
- 2026From Selection to Scheduling: Federated Geometry-Aware Correction Makes Exemplar Replay Work Better under Continual Dynamic Heterogeneity9 Apr 2026
- 2026GFPL: Generative Federated Prototype Learning for Resource-Constrained and Data-Imbalanced Vision Task25 Feb 2026
- 2025From Coefficients to Directions: Rethinking Model Merging with Directional Alignment29 Nov 2025
- 2025FedRE: A Representation Entanglement Framework for Model-Heterogeneous Federated Learning27 Nov 2025
- 2025FedDEAP: Adaptive Dual-Prompt Tuning for Multi-Domain Federated Learning21 Oct 2025
- 2025Neural Collapse-Inspired Multi-Label Federated Learning under Label-Distribution Skew16 Sep 2025
- 2025Rethinking the Backbone in Class Imbalanced Federated Source Free Domain Adaptation: The Utility of Vision Foundation Models10 Sep 2025
- 2025Calibrating Biased Distribution in VFM-derived Latent Space via Cross-Domain Geometric Consistency19 Aug 2025
38 more
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformers21 May 2025
- 2025Federated Adapter on Foundation Models: An Out-Of-Distribution Approach2 May 2025
- 2025FedAWA: Adaptive Optimization of Aggregation Weights in Federated Learning Using Client Vectors20 Mar 2025
- 2025FedLWS: Federated Learning with Adaptive Layer-wise Weight Shrinking19 Mar 2025
- 2025Enhancing Visual Representation with Textual Semantics: Textual Semantics-Powered Prototypes for Heterogeneous Federated Learning16 Mar 2025
- 2025You Are Your Own Best Teacher: Achieving Centralized-level Performance in Federated Learning under Heterogeneous and Long-tailed Data10 Mar 2025
- 2025Geometric Knowledge-Guided Localized Global Distribution Alignment for Federated Learning9 Mar 2025
- 2025Subgraph Federated Learning for Local Generalization6 Mar 2025
- 2025FedBM: Stealing Knowledge from Pre-trained Language Models for Heterogeneous Federated Learning24 Feb 2025
- 2025FedORGP: Guiding Heterogeneous Federated Learning with Orthogonality Regularization on Global Prototypes22 Feb 2025
- 2025Knowledge-Driven Federated Graph Learning on Model Heterogeneity22 Jan 2025
- 2024Covariances for Free: Exploiting Mean Distributions for Training-free Federated Learning18 Dec 2024
- 2024FuseFL: One-Shot Federated Learning through the Lens of Causality with Progressive Model Fusion27 Oct 2024
- 2024TPFL: A Trustworthy Personalized Federated Learning Framework via Subjective Logic16 Oct 2024
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapse7 Oct 2024
- 2024Collapsed Language Models Promote Fairness6 Oct 2024
- 2024DualFed: Enjoying both Generalization and Personalization in Federated Learning via Hierachical Representations25 Jul 2024
- 2024Tackling Feature-Classifier Mismatch in Federated Learning via Prompt-Driven Feature Transformation23 Jul 2024
- 2024Mamba-FSCIL: Dynamic Adaptation with Selective State Space Model for Few-Shot Class-Incremental Learning8 Jul 2024
- 2024Evaluating the Fairness of Neural Collapse in Medical Image Classification8 Jul 2024
- 2024Emerging Safety Attack and Defense in Federated Instruction Tuning of Large Language Models15 Jun 2024
- 2024FedLLM-Bench: Realistic Benchmarks for Federated Learning of Large Language Models7 Jun 2024
- 2024FedDr+: Stabilizing Dot-regression with Global Feature Distillation for Federated Learning4 Jun 2024
- 2024Accelerating Heterogeneous Federated Learning with Closed-form Classifiers3 Jun 2024
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Models28 May 2024
- 2024Fair Federated Learning under Domain Skew with Local Consistency and Domain Diversity26 May 2024
- 2024Client2Vec: Improving Federated Learning by Distribution Shifts Aware Client Indexing25 May 2024
- 2024Neural Collapse versus Low-rank Bias: Is Deep Neural Collapse Really Optimal?23 May 2024
- 2024Navigate Beyond Shortcuts: Debiased Learning through the Lens of Neural Collapse9 May 2024
- 2024An Upload-Efficient Scheme for Transferring Knowledge From a Server-Side Pre-trained Generator to Clients in Heterogeneous Federated Learning23 Mar 2024
- 2024FedGuCci: Making Local Models More Connected in Landscape for Federated Learning29 Feb 2024
- 2024Fixed Random Classifier Rearrangement for Continual Learning23 Feb 2024
- 2024FedImpro: Measuring and Improving Client Update in Federated Learning10 Feb 2024
- 2024Improving Model Fusion by Training-time Neuron Alignment with Fixed Neuron Anchors2 Feb 2024
- 2024FedLoGe: Joint Local and Generic Federated Learning under Long-tailed Data17 Jan 2024
- 2024FedTGP: Trainable Global Prototypes with Adaptive-Margin-Enhanced Contrastive Learning for Data and Model Heterogeneity in Federated Learning6 Jan 2024
- 2023FediOS: Decoupling Orthogonal Subspaces for Personalization in Feature-skew Federated Learning30 Nov 2023
- 2023FedFN: Feature Normalization for Alleviating Data Heterogeneity Problem in Federated Learning22 Nov 2023
50 newest of 55 All 55 (JSON)
- 2024FedTGP: Trainable Global Prototypes with Adaptive-Margin-Enhanced Contrastive Learning for Data and Model Heterogeneity in Federated Learningcited by 47
- 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Modelscited by 37
- 2024FedLLM-Bench: Realistic Benchmarks for Federated Learning of Large Language Modelscited by 30
- 2024Fair Federated Learning under Domain Skew with Local Consistency and Domain Diversitycited by 29
- 2025FedAWA: Adaptive Optimization of Aggregation Weights in Federated Learning Using Client Vectorscited by 19
- 2024An Upload-Efficient Scheme for Transferring Knowledge From a Server-Side Pre-trained Generator to Clients in Heterogeneous Federated Learningcited by 17
- 2024Neural Collapse versus Low-rank Bias: Is Deep Neural Collapse Really Optimal?cited by 17
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapsecited by 16
- 2024FedImpro: Measuring and Improving Client Update in Federated Learningcited by 15
- 2023FedSOL: Stabilized Orthogonal Learning with Proximal Restrictions in Federated Learningcited by 14
- 2024Navigate Beyond Shortcuts: Debiased Learning through the Lens of Neural Collapsecited by 13
12 most cited of 52 All 52 (JSON)
- 2026Understanding Head Geometry and Dynamics in Federated Regression through a Natural Solution Selection Rule: An Unconstrained Feature Model Analysis30 Sep 2026
- 2026FedLTLib: A Comprehensive Benchmark for Federated Long-Tail Learning14 Sep 2026
- 2026WHERE to Generate Matters: Budget-Aware Synthetic Augmentation for Label Skewed Federated Learning7 Jul 2026
- 2026FedReLa: Imbalanced Federated Learning via Re-Labeling24 Jun 2026
- 2026From Selection to Scheduling: Federated Geometry-Aware Correction Makes Exemplar Replay Work Better under Continual Dynamic Heterogeneity9 Apr 2026
- 2026GFPL: Generative Federated Prototype Learning for Resource-Constrained and Data-Imbalanced Vision Task25 Feb 2026
- 2025From Coefficients to Directions: Rethinking Model Merging with Directional Alignment29 Nov 2025
- 2025FedRE: A Representation Entanglement Framework for Model-Heterogeneous Federated Learning27 Nov 2025
- 2025FedDEAP: Adaptive Dual-Prompt Tuning for Multi-Domain Federated Learning21 Oct 2025
- 2025Neural Collapse-Inspired Multi-Label Federated Learning under Label-Distribution Skew16 Sep 2025
- 2025Rethinking the Backbone in Class Imbalanced Federated Source Free Domain Adaptation: The Utility of Vision Foundation Models10 Sep 2025
- 2025Calibrating Biased Distribution in VFM-derived Latent Space via Cross-Domain Geometric Consistency19 Aug 2025
12 newest of 52 All 52 (JSON)