atlas

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

Knowledge Transfer via Compact Model in Federated Learning (Student Abstract)

AAAI 2024 · Federated learning and privacy › Personalized and heterogeneous FL · approach: Non-IID federated learning · 0 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 a federated learning framework where clients pre-train large local models and transfer knowledge to a compact same-architecture global model via prediction loss and KL alignment of softmax outputs, reducing communication overhead.

Problem

Communication overhead in federated learning caused by frequent exchange of large global models, while preserving accuracy and client privacy.

Approach

Clients pre-train large local models; a compact global model (same architecture, smaller) is communicated and updated on clients using label loss plus KL divergence to the local model's softmax outputs, then aggregated by dataset-size-weighted averaging.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use compact-model knowledge transfer instead of FEDGENFederated learning · FMNIST (α=0.2)accuracy: 88.54 -> 89.21Helped
Use compact-model knowledge transfer instead of PrunFLFederated learning · FMNIST (α=0.2)communication overhead: 9.55E+01 -> 9.01E+01Helped
Use compact-model knowledge transfer instead of PrunFLFederated learning · FMNIST (α=0.2)training time: 1.35E+03 -> 7.24E+02Helped
Use compact-model knowledge transfer instead of FEDGENFederated learning · CIFAR10 (α=0.2)accuracy: 71.75 -> 72.16Helped
Use compact-model knowledge transfer instead of FEDGENFederated learning · CIFAR10 (α=0.2)communication overhead: 6.45E+04 -> 4.01E+04Helped
Use compact-model knowledge transfer instead of PrunFLFederated learning · CIFAR10 (α=0.2)training time: 8.91E+03 -> 7.51E+03Helped

Limitations

Experimental setup

Datasets: FMNIST CIFAR10

Benchmarks: FMNIST CIFAR10

Metrics: accuracy communication overhead (Mb) training time (s)

Baselines: FedAvg PrunFL FedGen

Builds on: FedAvg knowledge distillation

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