A Measured Communication-Quality Frontier for Federated LoRA Fine-Tuning with Adaptive Phase-Switching
Organizations: Independent Researcher, Fairport, NY, 14450, USA.
Abstract
Federated fine-tuning of large language models with low-rank adaptation (LoRA) reduces the number of trainable parameters, but communication remains the dominant cost, and protocols are usually compared by parameter-count ratios rather than by measured bytes. This paper measures per-round upload and download bytes for five federated LoRA protocols, three of them from prior work, and places them on a single communication-quality frontier scored by held-out instruction-following loss. The frontier has a knee. ReverseAdaptive, which learns both LoRA factors before freezing one once the relative improvement in training loss falls below a dimensionless threshold, sits at that knee: it cuts measured round-trip communication by 40.5% relative to FLoRA at a held-out loss cost of 0.0063, and beats FFA-LoRA, which freezes that factor at initialization, by 0.0182 in held-out loss, more than twenty times the largest per-method seed standard deviation. The same threshold carries to LLaMA-3.2-3B without retuning, where it saves 30.0%.
Figures & tables
| Parameter | Value |
|---|---|
| Base models | TinyLlama-1.1B-Chat-v1.0 Zhang et al. (2024) ; LLaMA-3.2-3B Dubey et al. (2024) |
| Datasets | Alpaca Taori et al. (2023) ; Dolly-15k Conover et al. (2023) ; 3000 samples each |
| Data partitions | IID; Dirichlet ; Dirichlet |
| Clients / round | 10 / 10 (full participation) |
| Rounds | 15 |
| Local epochs | 1 |
| Method | MB | Savings | Final loss | Held-out | vs. FLoRA |
|---|---|---|---|---|---|
| FLoRA | 2578.13 | – | – | ||
| FedIT | 2578.13 | 0.00% | |||
| Two-Phase | 1863.13 | 27.73% | |||
| ReverseAdaptive ( ) | 1533.13 | 40.53% | |||
| FFA-LoRA | 983.13 | 61.87% |
| Method | MB | Final loss | Held-out | Gap (Dolly) | Gap (Alpaca) |
|---|---|---|---|---|---|
| FLoRA | 2578.13 | – | – | ||
| Two-Phase | 1863.13 | 0.0043 | 0.0034 | ||
| ReverseAdaptive ( ) | 1533.13 | 0.0061 | 0.0063 |
| Method | ARC-Easy | BoolQ | HellaSwag | vs. base (BoolQ) |
|---|---|---|---|---|
| Base (no adapter) | 0.274 | 0.626 | 0.448 | – |
| FLoRA | 0.246 | 0.562 | 0.420 | pp |
| Two-Phase | 0.248 | 0.560 | 0.420 | pp |
| ReverseAdaptive | 0.252 | 0.552 | 0.420 | pp |
| Spread (max–min) | 0.006 | 0.010 | 0.000 | – |
| Method | MMLU | ARC-Easy | BoolQ | HellaSwag |
|---|---|---|---|---|
| Base (no adapter) | 0.544 | 0.832 | 0.706 | 0.540 |
| FLoRA | ||||
| Two-Phase | ||||
| ReverseAdaptive |
| Switch round | Final loss | Total comm. (MB) | Savings | |
|---|---|---|---|---|
| 0.001 | 11 | 1.2638 | 2083.13 | 19.20% |
| 0.002 | 9 | 1.2675 | 1863.13 | 27.73% |
| 0.005–0.2 | 6 | 1.2736 | 1533.13 | 40.53% |
| Method | Final loss | MB | Savings |
|---|---|---|---|
| FLoRA | 2625.0 | 0% | |
| Two-Phase | 1942.5 | 26.0% | |
| ReverseAdaptive | % |
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | Seed | Final loss | Total MB | Switch / boundary |
|---|---|---|---|---|
| FLoRA | 42 | 1.261408 | 2578.13 | – |
| FLoRA | 123 | 1.260479 | 2578.13 | – |
| FLoRA | 456 | 1.260500 | 2578.13 | – |
| FedIT | 42 | 1.260767 | 2578.13 | – |
| FedIT | 123 | 1.258420 | 2578.13 | – |
| FedIT | 456 | 1.261503 | 2578.13 | – |
| Setting | Seed | Final loss | Total MB | Switch round | Active clients |
|---|---|---|---|---|---|
| ReverseAdaptive, | 42 | 1.285933 | 1533.13 | 6 | 10 |
| ReverseAdaptive, | 123 | 1.282625 | 1533.13 | 6 | 10 |
| ReverseAdaptive, | 456 | 1.252133 | 1533.13 | 6 | 10 |
| ReverseAdaptive, | 42 | 1.268353 | 1379.81 | 6 | 9 |
| ReverseAdaptive, | 123 | 1.288985 | 1533.13 | 6 | 10 |
| ReverseAdaptive, | 456 | 1.348083 | 1533.13 | 6 | 10 |
| Method | Seed | Final loss | Total MB | Switch |
|---|---|---|---|---|
| FLoRA | 42 | 1.371683 | 2625.0 | – |
| FLoRA | 123 | 1.373402 | 2625.0 | – |
| FLoRA | 456 | 1.371998 | 2625.0 | – |
| Two-Phase | 42 | 1.393776 | 1942.5 | |
| Two-Phase | 123 | 1.394877 | 1942.5 | |
| Two-Phase | 456 | 1.392772 | 1942.5 |
| Method | Seed | Final loss | Held-out | Total MB |
|---|---|---|---|---|
| FLoRA | 42 | 1.656762 | 2578.13 | |
| FLoRA | 123 | 1.652934 | 2578.13 | |
| FLoRA | 456 | 1.653426 | 2578.13 | |
| Two-Phase | 42 | 1.666616 | 1863.13 | |
| Two-Phase | 123 | 1.662685 | 1863.13 | |
| Two-Phase | 456 | 1.663475 | 1863.13 |
| Comparison | final loss | held-out | ||
|---|---|---|---|---|
| TinyLlama-1.1B, Alpaca, IID | ||||
| Two-Phase vs. FLoRA | ||||
| ReverseAdaptive vs. FLoRA | ||||
| ReverseAdaptive vs. Two-Phase | ||||
| FFA-LoRA vs. ReverseAdaptive | ||||
| FFA-LoRA vs. FLoRA | ||||
| Category | Runs |
|---|---|
| Agreed within setting-aware tolerance | 27 |
| Disagreed | 4 |
| Comparable pairs | 31 |
| No valid MPS reference (CUDA reported) | 3 |
| Total | 34 |
| Run | Config. | MPS | CUDA | Note |
|---|---|---|---|---|
| 11 | , IID | 2083.13 | 1973.13 | : 11 to 10 |
| 12 | , IID | 1863.13 | 1753.13 | : 9 to 8 |
| 16 | , 123 | 1533.13 | 1533.13 | spike rule |
| 17 | , 456 | 1533.13 | 1643.13 | : 6 to 7 |
Federated Fine-Tuning
Federated fine-tuning adapts pretrained models across decentralized data holders without pooling their private data.
See all 115 papers