atlas

Research map / Federated learning and privacy / Communication-efficient FL

Decentralized Directed Collaboration for Personalized Federated Learning

CVPR 2024 · Federated learning and privacy › Communication-efficient FL · approach: Local drift correction · 31 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 DFedPGP, a decentralized personalized federated learning method that combines partial model personalization (private linear classifier) with stochastic gradient push over time-varying directed graphs, with an O(1/sqrt(T)) non-convex convergence guarantee.

Problem

Personalized federated learning in decentralized peer-to-peer settings suffers from suboptimal personalization and unguaranteed convergence when using undirected, symmetric topologies under data and resource heterogeneity.

Approach

Decouple the model into a shared feature extractor and a private linear classifier, and exchange only shared parameters over a directed, asymmetric topology using push-sum with de-biasing weights, with alternating local updates of the personal and shared parts.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use DFedPGP instead of FedRepCIFAR-10 PFL · CIFAR-10 Dirichlet α=0.3test accuracy (%): 84.50±.05 -> 85.61±.05Helped
use DFedPGP instead of FedBABUCIFAR-100 PFL · CIFAR-100 Dirichlet α=0.1test accuracy (%): 60.23±.07 -> 66.26±.25Helped
use DFedPGP instead of OSGPCIFAR-100 PFL · CIFAR-100 Pathological c=10test accuracy (%): 71.08±.52 -> 72.19±.21Helped
add partial personalization to DFedAvgM (ablation)DFedAvgM · CIFAR-10 Dirichlet α=0.3test accuracy (%): 82.49 -> 84.69Helped
add partial personalization to DFedAvgM (ablation)DFedAvgM · CIFAR-10 Pathological c=2test accuracy (%): 90.23 -> 90.90Helped
add directed communication and partial personalization to OSGP (ablation)OSGP · CIFAR-10 Dirichlet α=0.3test accuracy (%): 83.14 -> 85.61Helped
add directed communication and partial personalization to OSGP (ablation)OSGP · CIFAR-10 Pathological c=2test accuracy (%): 90.72 -> 91.26Helped
increase the number of participating neighbors (ablation)DFedPGP · CIFAR-10 Dirichlet α=0.3test accuracyHelped
reduce participating clients to 5 or 10 rather than 50, 100, or 200 (ablation)DFedPGP · CIFAR-10 Dirichlet α=0.3test accuracyHelped
increase communication network connectivity and exchange with more neighborsDFedPGP · non-convex convergence analysisconvergence rateHelped

Surprising findings

Limitations

Experimental setup

Datasets: CIFAR-10 CIFAR-100 Tiny-ImageNet

Benchmarks: CIFAR-10 CIFAR-100 Tiny-ImageNet

Metrics: test accuracy communication rounds to target accuracy

Baselines: Local FedAvg FedPer FedRep FedBABU Ditto DFedAvgM OSGP Dis-PFL

Builds on: Stochastic Gradient Push (SGP) FedRep FedPer FedBABU Push-sum

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