atlas

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

Federated Learning over Connected Modes

NeurIPS 2024 · Federated learning and privacy › Personalized and heterogeneous FL · approach: Non-IID federated learning · 3 citations (OpenAlex)

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 FLoCo, a federated learning method that trains a low-loss solution simplex of last-layer weights and assigns each client a subregion of it based on projected gradient signals, improving both global and personalized accuracy.

Problem

Statistical heterogeneity in federated learning causes slow global training from conflicting gradients and poor local (personalized) performance, with a typical trade-off between global and local accuracy.

Approach

Leverages linear mode connectivity: a solution simplex of model endpoints is trained jointly across clients, where each client samples models from an assigned subregion of the simplex derived by projecting its gradient signals onto the standard simplex (PCA plus Riesz s-energy projection). Simplex learning is applied only to the last layer, and an optional fine-tuning variant (FLoCo+) adds Ditto-s

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace global training with Floco solution-simplex training and client subregion assignmentCIFAR-10 5-Fold, CifarCNN · CIFAR-10 5-Fold, CifarCNN globalaccuracy: 60.68 -> 62.93Helped
Use Floco client subregion personalizationCIFAR-10 5-Fold, CifarCNN · CIFAR-10 5-Fold, CifarCNN localaccuracy: 74.03 -> 71.78Hurt
Replace global training with Floco solution-simplex training and client subregion assignmentCIFAR-10 Dir(0.3), CifarCNN · CIFAR-10 Dir(0.3), CifarCNN globalaccuracy: 60.74 -> 62.57Helped
Use Floco client subregion personalizationCIFAR-10 Dir(0.3), CifarCNN · CIFAR-10 Dir(0.3), CifarCNN localaccuracy: 76.42 -> 71.04Hurt
Apply simplex learning only to the last fully connected layer rather than all network parameters (ablation)Floco · CIFAR-10 global and localaccuracyNo clear effect
Assign subregions earlier and reduce the subregion radius (ablation)Floco · CIFAR-10 5-Fold, CifarCNN localaccuracy: 78.18% against 82.79%Helped
Vary subregion assignment round and radius (ablation)Floco · CIFAR-10 5-Fold, CifarCNN globalaccuracy: 70.66% against 69.30%No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: CIFAR-10 FEMNIST

Benchmarks: CIFAR-10 FEMNIST

Metrics: global test accuracy local test accuracy expected calibration error time-to-accuracy improvement worst 5% client accuracy

Baselines: FedAvg FedProx FedPer APFL Ditto FedRoD SuPerFed

Builds on: Ditto FedAvg simplex learning (Benton et al. 2021) SuPerFed

Closest papers by meaning

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.

Open scime draft

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