Papers › Recurrent Early Exits for Federated Learning with Heterogeneous Clients

Recurrent Early Exits for Federated Learning with Heterogeneous Clients

23 May 2024arXiv:2405.14791archive 2025-07-28

Royson Lee, Javier Fernandez-Marques, Shell Xu Hu, Da Li, Stefanos Laskaridis, Łukasz Dudziak, Timothy Hospedales, Ferenc Huszár, Nicholas D. Lane

Federated learning (FL) has enabled distributed learning of a model across multiple clients in a privacy-preserving manner. One of the main challenges of FL is to accommodate clients with varying hardware capacities; clients have differing compute and memory requirements. To tackle this challenge, recent state-of-the-art approaches leverage the use of early exits. Nonetheless, these approaches fall short of mitigating the challenges of joint learning multiple exit classifiers, often relying on hand-picked heuristic solutions for knowledge distillation among classifiers and/or utilizing additional layers for weaker classifiers. In this work, instead of utilizing multiple classifiers, we propose a recurrent early exit approach named ReeFL that fuses features from different sub-models into a single shared classifier. Specifically, we use a transformer-based early-exit module shared among sub-models to i) better exploit multi-layer feature representations for task-specific prediction and ii) modulate the feature representation of the backbone model for subsequent predictions. We additionally present a per-client self-distillation approach where the best sub-model is automatically selected as the teacher of the other sub-models at each client. Our experiments on standard image and speech classification benchmarks across various emerging federated fine-tuning baselines demonstrate ReeFL's effectiveness over previous works.

PaperPDFCodeCode Syntology ran

In Syntology View this paper on Syntology: its repositories, every harvested function with whether it ran, its licence and the call to fetch it.

Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

For agents, Syntology's MCP tool lists every function and class Syntology harvested from this paper and whether it ran (how to connect): get_harvested_code_for_paper(arxiv_id="2405.14791")

Code

Syntology Ran 9 of 20 code samples harvested from 1 repository linked to this paper; 11 have no recorded run. Of those that ran: 3 ran · our draft was wrong; 1 ran · fixture could not drive it; 5 ran with no contract checked.

By repository: official repository: 20 samples from 1 repository, 9 ran. The run record, sample by sample. “Ran” means executed on a synthesized input, not that the code is correct or reproduces the paper.

royson/reefl officialmentioned in paperpytorch report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

20 samples harvested; 9 ran; 0 honoured the contract we drafted; 11 have no recorded run. Read from Syntology's graph 2026-09-25; that is when this build read the record, not when the samples ran.

3ran · our draft was wrong
1ran · fixture could not drive it
5ran
11unverified

Licence: 0 of the 20 samples are pointer only, meaning Syntology does not serve that copy's text. This page shows no code text for any sample; each one links to its file in the repository.

Harvested from royson/reefl. “Ran” means the sample executed on a synthesized input. It does not mean the output is correct, and nothing here reproduces the paper's results. “Honoured” and “violated” refer to a contract Syntology drafted from the code itself; “our draft was wrong” and “fixture could not drive it” are failures of Syntology's instrument, not of the code.

Each sample ends with its code_sha256, Syntology's identity for that exact code. An agent fetches the stored sample with Syntology's MCP tool get_code(code_sha256="…") (how to connect); click an identity to copy that call.

Repository labels, per sample. official repository: The archive marks this repository official for the paper. named in the paper: The archive records that the paper mentions this repository; it is not marked official. community (archive-listed): In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper. found in paper text by Syntology: Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted. community: Not in the archive's code links for this paper; a community repository Syntology harvested. Samples from a repository marked official are listed first. Licence labels name the repository's licence as recorded at harvest. “Pointer only” means Syntology does not serve that copy's text, for one of four reasons: no licence file was found; the licence was not identified; the licence is recorded as permissive but that copy's record is not marked cleared; or the licence is outside the permissive list Syntology serves text under (MIT, Apache-2.0, BSD and similar). Some licences outside that list permit redistribution, such as WTFPL, and GPL-3.0 under its conditions; they are simply not on the list. Hover a licence label for the reason. File links open the file on GitHub at the default branch, which may have changed since the harvest.

Adapter_Layer royson/reefl/src/models/reefl_vit.py official repository ran · metamorphic tier: invariant MIT (permissive) · 25b4f80e64754f5f · report
Linear royson/reefl/src/models/reefl_vit.py official repository ran · metamorphic tier: invariant MIT (permissive) · 00410aa030868e82 · report
LoRALinear royson/reefl/src/models/reefl_vit.py official repository ran MIT (permissive) · 14af14fc44efac97 · report
NormAndLinear royson/reefl/src/models/reefl_vit.py official repository ran · metamorphic tier: invariant MIT (permissive) · c6f8f65aec8f0cda · report
PatchEmbed royson/reefl/src/models/reefl_vit.py official repository ran · metamorphic tier: invariant MIT (permissive) · 50c83684a8d032d4 · report
get_func_from_config royson/reefl/src/models/reefl_vit.py official repository ran · our draft was wrong MIT (permissive) · a0433a63b0aa9b64 · report
init_ssf_scale_shift royson/reefl/src/models/reefl_vit.py official repository ran · our draft was wrong fingerprinted MIT (permissive) · 9873773b10315cc0 · report
prune royson/reefl/src/models/reefl_vit.py official repository ran · our draft was wrong MIT (permissive) · 9b6ef57a04624e80 · report
trunc_normal_ royson/reefl/src/models/reefl_vit.py official repository ran · fixture could not drive it MIT (permissive) · af8aefa30bd76920 · report
Attention royson/reefl/src/models/reefl_vit.py official repository unverified MIT (permissive) · dfbc1c3d50b97f50 · report
Block royson/reefl/src/models/reefl_vit.py official repository unverified MIT (permissive) · b106d531af0900bc · report
Mlp royson/reefl/src/models/reefl_vit.py official repository unverified MIT (permissive) · 335a33f331dc2ff7 · report
Ree royson/reefl/src/models/reefl_vit.py official repository unverified MIT (permissive) · 47dcd63472af63f4 · report
Transformer royson/reefl/src/models/reefl_vit.py official repository unverified MIT (permissive) · e16ec4a665b3d4df · report
VisionTransformer royson/reefl/src/models/reefl_vit.py official repository unverified MIT (permissive) · dbbfaa54d7e3f160 · report
init_bert_weights royson/reefl/src/models/reefl_vit.py official repository unverified MIT (permissive) · 0a5a8b2a981b7267 · report
reefl_vit_template royson/reefl/src/models/reefl_vit.py official repository unverified MIT (permissive) · 3abf98fe951b6095 · report
vit_base royson/reefl/src/models/reefl_vit.py official repository unverified MIT (permissive) · b50fb2ac893c8ce1 · report
vit_small royson/reefl/src/models/reefl_vit.py official repository unverified MIT (permissive) · 1befb46451d50591 · report
vit_tiny royson/reefl/src/models/reefl_vit.py official repository unverified MIT (permissive) · c856301a391d8154 · report

Tasks

Federated LearningKnowledge DistillationPrivacy Preserving

Results from the paper archive 2025-07-28

No leaderboard rows for this paper in the archive.

Methods

Knowledge Distillation

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections