Research map / Federated learning and privacy / Personalized and heterogeneous FL
Geometric Knowledge-Guided Localized Global Distribution Alignment for Federated Learning
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes GGEUR, which estimates the global covariance/geometric shape of CLIP embedding distributions from local client statistics and uses it to augment embeddings locally, addressing label and domain skew in federated learning.
Problem
Data heterogeneity (label skew, domain skew, and their coexistence) in federated learning causes misalignment between local and global data distributions, leading to divergent local optimization and unstable global training.
Approach
Approximates the global covariance matrix from local class covariances and means without sharing data, derives the global geometric shape (eigenvectors and eigenvalues) per class, and uses random linear combinations of these eigenvectors to augment local CLIP embeddings; in multi-domain settings, class prototypes from other domains are augmented to simulate cross-domain global distributions.
Key results
- On CIFAR-100 with beta=0.1, FedAvg (CLIP+MLP) improves from 68.22 to 77.70 with GGEUR, outperforming FedMix, FedGen, and FedFA.
- On Digits, GGEUR raises FedAvg (CLIP+MLP) average accuracy from 76.60 to 82.36 and lowers cross-domain STD from 16.25 to 13.84.
- On Office-Home-LDS (beta=0.1), GGEUR improves FedAvg (CLIP+MLP) average accuracy by 13.85 points (70.14 to 83.99) and reduces STD from 8.89 to 5.36.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Add GGEUR to FedAvg (CLIP+MLP) | FedAvg (CLIP+MLP) · CIFAR-100, β=0.01 | Top-1 accuracy (%): 58.71 -> 75.72 | Helped |
| Add GGEUR to FedAvg (CLIP+MLP) | FedAvg (CLIP+MLP) · CIFAR-10, β=0.01 | Top-1 accuracy (%): 90.87 -> 94.39 | Helped |
| Add GGEUR to FedAvg (CLIP+MLP) | FedAvg (CLIP+MLP) · Tiny-ImageNet, β=0.01 | Top-1 accuracy (%): 53.03 -> 64.27 | Helped |
| Add GGEUR Steps 1 and 2 | FedAvg (CLIP+MLP) · Digits | average Top-1 accuracy (%): 76.60 -> 82.36 | Helped |
| Apply GGEUR Step 1 only (ablation) | FedAvg (CLIP+MLP) · Digits | average Top-1 accuracy (%): 76.60 -> 80.18 | Helped |
| Add GGEUR Step 2 after Step 1 (ablation) | FedAvg (CLIP+MLP) with Step 1 · Digits | average Top-1 accuracy (%): 80.18 -> 82.36 | Helped |
| Add GGEUR Steps 1 and 2 | FedAvg (CLIP+MLP) · Digits | accuracy standard deviation: 16.25 -> 13.84 | Helped |
| Add GGEUR to FedAvg (CLIP+MLP) | FedAvg (CLIP+MLP) · Office-Home-LDS, β=0.1 | average Top-1 accuracy (%): 70.14 -> 83.99 | Helped |
| Add GGEUR to FedAvg (CLIP+MLP) | FedAvg (CLIP+MLP) · Office-Home-LDS, β=0.1 | accuracy standard deviation: 8.89 -> 5.36 | Helped |
| Add GGEUR to MOON (CLIP+MLP) | MOON (CLIP+MLP) · Office-Home-LDS, β=0.1 | average Top-1 accuracy (%): 76.83 -> 78.08 | Helped |
Surprising findings
- Simply using CLIP features with a single-layer MLP already outperforms many existing FL methods on domain-skew datasets, which the authors attribute to foundation model advances.
- FedNTD without CLIP achieves 56.43 average on Digits but its CLIP+MLP variant reaches 78.86, showing backbone choice dominates some comparisons.
Limitations
- Geometric similarity across domains is shown empirically on Digits only, so cross-domain shared geometry may not generalize.
- Gaussian-style random augmentation along eigen-directions relies on the assumption that the top eigenvector structure captures the global distribution; accuracy of covariance approximation under extreme heterogeneity is not analyzed in the main text.
- Evaluation uses CLIP ViT-B/16 backbones, so gains may depend on the strength of the pretrained foundation model.
Experimental setup
Datasets: CIFAR-10 CIFAR-100 Tiny-ImageNet Digits Office-Caltech PACS Office-Home
Benchmarks: CIFAR-10 CIFAR-100 Tiny-ImageNet Digits Office-Caltech PACS Office-Home-LDS
Metrics: Top-1 accuracy standard deviation of accuracy across domains
Baselines: FedMix FEDGEN FedFA FedTPG FedCLIP SCAFFOLD MOON FedDyn FedOPT FedProto FedNTD
Builds on: FedAvg FedFA FedMix FedGen CLIP
Closest papers by meaning
- FedScar: Correcting Geometric Bias for Flatness-Consistent Federated Learning (ICML 2026 · similarity 0.89)
- FedPDG: Prediction Discrepancy–Guided Data Generation for Heterogeneous Federated Learning (ICML 2026 · similarity 0.88)
- Federated Graph Learning under Domain Shift with Generalizable Prototypes (AAAI 2024 · similarity 0.86)
- FLUX: Efficient Descriptor-Driven Clustered Federated Learning under Arbitrary Distribution Shifts (NeurIPS 2025 · similarity 0.86)
- FedGPS: Statistical Rectification Against Data Heterogeneity in Federated Learning (NeurIPS 2025 · similarity 0.86)
- One Arrow, Two Hawks: Sharpness-aware Minimization for Federated Learning via Global Model Trajectory (ICML 2025 · similarity 0.86)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
