AI Paper Atlas

Research map / Federated learning and privacy / Personalized and heterogeneous FL

Non-IID federated learning

Counter data heterogeneity with drift decoupling, distillation and re-labeling.

198 accepted papers at ICML, NeurIPS, ICLR, CVPR and AAAI (2016–2026) · an approach in Personalized and heterogeneous FL, keyword cluster “domain · label · multi”

State History Tried Baselines Checklist Preprints Who Export

Where it stands

Not enough benchmark data Fewer than four papers report the same benchmark, so there is no record to follow.

66 papers in 2023–24 and 108 in 2025–26 (at ICML, NeurIPS, ICLR, CVPR and AAAI); its share of the topic held steady, from 26.5% to 27.8%.

Open questions the data points to

How this is decided

A benchmark is tracked when at least four papers of this approach report their own method on the same dataset and metric in at least two different years. Its record is the best number reported up to 2024 compared with the best reported in 2025–26: still improving means at least 1% better, flat means at most 0.2% better with at least two papers from 2025–26 trying. Numbers on a different scale from the rest (a fraction among percentages) and far outliers are left out. Results come from different papers with different setups, so they are signals, not head-to-head comparisons. The share of papers is counted at ICML, NeurIPS, ICLR, CVPR and AAAI.

How it got here

2016: 2162017: 0172018: 0182019: 2192020: 0202021: 5212022: 15222023: 22232024: 44242025: 42252026: 6626

Most cited paper of each year

Earliest papers here

What has been tried

117 papers here report 1,164 measured changes (a component added, removed or swapped, compared with a baseline): helped 71%, no clear effect 5%, hurt 20%, mixed 4%. Papers mostly publish changes that helped, so "helped" is over-represented.

Ablations in recent papers

What was changedMeasured onResultEffectPaper
replace adaptive-margin contrastive learning with standard contrastive lossCifar100, practical · accuracy46.94 -> 40.11HurtFedTGP: Trainable Global Prototypes with Adaptive-Margin-Enhanced Contrastive Learning for Data and Model Heterogeneity in Federated Learning (AAAI 2024 · 114 citations)
Add collaborative data condensation (CDC, SWD regularization)CIFAR10 α=0.02 · accuracy+0.99 (65.15 vs 64.16)HelpedAn Aggregation-Free Federated Learning for Tackling Data Heterogeneity (CVPR 2024 · 61 citations)
replace relaxed contrastive loss with supervised contrastive lossCIFAR-100, α=0.05, 1000R · accuracy54.63 -> 42.93HurtRelaxed Contrastive Learning for Federated Learning (CVPR 2024 · 43 citations)
add the additional local knowledge-transfer lossTiny-ImageNet, practical · test accuracy24.39 -> 28.17HelpedAn Upload-Efficient Scheme for Transferring Knowledge From a Server-Side Pre-trained Generator to Clients in Heterogeneous Federated Learning (CVPR 2024 · 30 citations)
add classifier variance regularizationSTL-10 (α=0.01) · accuracy24.1±1.3 -> 30.4±1.4HelpedFedUV: Uniformity and Variance for Heterogeneous Federated Learning (CVPR 2024 · 23 citations)
use balanced classifiers rather than local classifiers as clustering inputCIFAR-10, sudden drift, Dir(0.5), 100 clients · generalized accuracy68.39±0.71 -> 73.21±0.82HelpedClassifier Clustering and Feature Alignment for Federated Learning under Distributed Concept Drift (NeurIPS 2024 · 21 citations)
Add gradient alignment to agent valuationPROTEINS · model gradient fairness0.736±0.145 -> 0.787±0.052HelpedTowards Fair Graph Federated Learning via Incentive Mechanisms (AAAI 2024 · 20 citations)
Add the global acyclicity constraintSachs · FDR69.23 -> 52.63HelpedFederated Causality Learning with Explainable Adaptive Optimization (AAAI 2024 · 19 citations)
Add embedding projectionCifar100, HtFE_8, cross-device · test accuracy (%)29.54 -> 30.80 (+1.26)HelpedFedSA: A Unified Representation Learning via Semantic Anchors for Prototype-based Federated Learning (AAAI 2025 · 18 citations)
Add TERLCIFAR10, CIFAR100 · Top-1 accuracy70.85 -> 72.06; 68.24 -> 70.01; 60.67 -> 62.53; 57.58 -> 59.46HelpedCross-Silo Feature Space Alignment for Federated Learning on Clients with Imbalanced Data (AAAI 2025 · 15 citations)
omit DAT and use only CKFCIFAR-10, β=0.2 · local accuracy74.34% -> 73.74%HurtFedGMKD: An Efficient Prototype Federated Learning Framework through Knowledge Distillation and Discrepancy-Aware Aggregation (NeurIPS 2024 · 15 citations)
Add the stage-one auxiliary lossMERGE-FA · accuracy88.44 ± 0.36 -> 91.51 ± 0.36HelpedOn Disentanglement of Asymmetrical Knowledge Transfer for Modality-Task Agnostic Federated Learning (AAAI 2024 · 14 citations)

Baseline numbers that disagree

The same baseline on the same benchmark and metric, reported more than 5% apart by different papers here. Training budgets, data splits and reimplementations differ, so state which number you compare with.

BaselineBenchmarkMetricReportedPapers
FedProxTinyImageNet testtop-1 accuracy23.2 – 50.632 papers, e.g. CVPR 2021 and AAAI 2023
SCAFFOLDFashion-MNIST NIID-1accuracy (%)77.02 – 89.652 papers, e.g. ICLR 2024 and ICML 2023
FedAvgCIFAR100 α=∞test accuracy (%)67.2 – 74.542 papers, e.g. ICLR 2025 and ICLR 2023

What reviewers will expect

Datasets and baselines named in the 58 papers of this approach from 2025–2026 that have a full-text summary. Leaving out the common ones is a frequent review question.

Datasets

  • CIFAR-100 · 23 of 58 papers
  • CIFAR-10 · 21 of 58 papers
  • Tiny-ImageNet · 8 of 58 papers
  • PACS · 5 of 58 papers
  • Fashion-MNIST · 4 of 58 papers
  • CINIC-10 · 4 of 58 papers
  • MNIST · 4 of 58 papers
  • Office-Home · 4 of 58 papers

Baselines

  • FedAvg · 22 of 58 papers
  • FedProx · 14 of 58 papers
  • SCAFFOLD · 12 of 58 papers
  • FedProto · 7 of 58 papers
  • MOON · 6 of 58 papers
  • FedDyn · 6 of 58 papers
  • FedSAM · 5 of 58 papers
  • FedKD · 4 of 58 papers

Recent preprints

Loading recent arXiv preprints…

    Who publishes on it

    Institutions on the most papers of this approach since 2023 (of 174 papers; from OpenAlex and the papers' affiliations).

    Use it

    Download BibTeX of 36 key papersOpen the topic in the interactive map

    Get alerts. Describe your idea in scime atlas and save it (free scime account): you get a weekly email when new preprints land close to it.

    Same keyword cluster

    Elsewhere in the topic

    Benchmark numbers, ablations and baseline numbers are extracted by AI from each paper's full text and can be wrong; the paper is the authority. Data: results index of 2026-11-02, citation counts from OpenAlex (2026-10-09).

    Get the next scime Dispatch and a monthly field update

    Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy