DROD: Dual-Ratio Ordered Dropout for Federated Continual Learning on Heterogeneous Devices
Organizations: Middle East Technical University · Stanford University
Abstract
Federated learning for capacity-heterogeneous clients and federated continual learning are each well studied on their own, but their intersection -- clients holding differently sized models while every client's data stream also drifts over time -- remains underexplored. We study this setting directly, training a ResNet-18 fleet from scratch on Split-CIFAR-100 (10 tasks, class-incremental), and introduce DROD (Dual-Ratio Ordered Dropout), which builds on FjORD's Ordered Dropout by using two independent reduction axes to extract sub-models: a shortcut ratio and a branch ratio. On a shared parameter/FLOP budget ladder, DROD achieves the highest mean accuracy among the heterogeneous-model federated learning strategies we compare. Its advantage over the two strongest baseline geometries persists across every replay-buffer size we test and under ER, DER++, and BiC; under EWC, all three geometries collapse alike.
Figures & tables
| class-IL | Task-IL | % saved, mid / low | ||||
|---|---|---|---|---|---|---|
| reduction aggregation | ACC | forg. | ACC | forg. | params | FLOPs |
| DROD | 74.9 / 93.6 | 74.8 / 93.7 | ||||
| DROD | 75.0 / 93.5 | 75.4 / 94.0 | ||||
| DROD | 74.8 / 93.7 | 74.8 / 93.7 | ||||
| Width FjORD | 74.9 / 93.6 | 74.9 / 93.7 | ||||
| DROD | 74.7 / 93.6 | 74.7 / 93.8 | ||||
| CL objective | DROD | Width FjORD | Shortcut FjORD |
|---|---|---|---|
| ER | |||
| DER++ | |||
| BiC | |||
| EWC |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| setting | value | |
| Data | dataset | CIFAR-100, 10 tasks 10 classes |
| class order | fixed (identity), not shuffled across seeds | |
| augmentation | random crop , pad (black fill) horizontal flip | |
| batch size | 32 (train), 512 (eval) | |
| client partition | Dirichlet , redrawn per task FL | |
| Model | architecture | ResNet-18, CIFAR stem, from scratch |
| cell | fleet class-IL ACC | high (t100) | mid (t025) | low (t006) |
|---|---|---|---|---|
| DROD =0.50 † | 0.3125 0.0160 | 0.3076 | 0.3161 | 0.3137 |
| DROD =1.00 † | 0.2812 0.0048 | 0.2795 | 0.2840 | 0.2801 |
| DROD =0.25 † | 0.2760 0.0091 | 0.2709 | 0.2785 | 0.2784 |
| Width FjORD (kd=false) † | 0.2741 0.0084 | 0.2701 | 0.2815 | 0.2708 |
| DROD =0.75 † | 0.2723 0.0083 | 0.2680 | 0.2766 | 0.2723 |
| Width FjORD (kd=true) † | 0.2719 0.0157 | 0.2493 | 0.2896 | 0.2770 |
| cell | fleet class-IL ACC | high (t100) | mid (t025) | low (t006) |
|---|---|---|---|---|
| DROD =0.50 † | 0.3125 0.0160 | 0.3076 | 0.3161 | 0.3137 |
| DROD =1.00 † | 0.2812 0.0048 | 0.2795 | 0.2840 | 0.2801 |
| DROD =0.25 † | 0.2760 0.0091 | 0.2709 | 0.2785 | 0.2784 |
| Width FjORD (kd=false) † | 0.2741 0.0084 | 0.2701 | 0.2815 | 0.2708 |
| DROD =0.75 † | 0.2723 0.0083 | 0.2680 | 0.2766 | 0.2723 |
| Width FjORD (kd=true) † | 0.2719 0.0157 | 0.2493 | 0.2896 | 0.2770 |
| cell | fleet class-IL ACC | high (t100) | mid (t025) | low (t006) |
|---|---|---|---|---|
| DROD =0.50 † | 0.0863 0.0044 | 0.0832 | 0.0870 | 0.0886 |
| DROD =0.25 † | 0.0845 0.0032 | 0.0828 | 0.0848 | 0.0861 |
| DROD =1.00 † | 0.0845 0.0002 | 0.0833 | 0.0849 | 0.0854 |
| DROD =0.75 † | 0.0822 0.0029 | 0.0806 | 0.0832 | 0.0829 |
| Width FjORD (kd=true) † | 0.0818 0.0036 | 0.0790 | 0.0819 | 0.0845 |
| Shortcut FIARSE | 0.0798 0.0016 | 0.0804 | 0.0810 | 0.0779 |
| cell | fleet class-IL ACC | high (t100) | mid (t025) | low (t006) |
|---|---|---|---|---|
| DROD =0.50 † | 0.0863 0.0044 | 0.0832 | 0.0870 | 0.0886 |
| DROD =0.25 † | 0.0845 0.0032 | 0.0828 | 0.0848 | 0.0861 |
| DROD =1.00 † | 0.0845 0.0002 | 0.0833 | 0.0849 | 0.0854 |
| DROD =0.75 † | 0.0822 0.0029 | 0.0806 | 0.0832 | 0.0829 |
| Width FjORD (kd=true) † | 0.0818 0.0036 | 0.0790 | 0.0819 | 0.0845 |
| Width FjORD (kd=false) † | 0.0797 0.0001 | 0.0778 | 0.0794 | 0.0819 |
| CL objective | fleet | class-IL ACC | high | mid | low | class-IL forg. |
|---|---|---|---|---|---|---|
| ER | DROD | 0.3076 | 0.3161 | 0.3137 | 0.251 | |
| Width | 0.2701 | 0.2815 | 0.2708 | 0.263 | ||
| Shortcut | 0.2530 | 0.2459 | 0.2564 | 0.291 | ||
| DER++ | DROD | 0.2947 | 0.2975 | 0.2952 | 0.322 | |
| Width | 0.2805 | 0.2835 | 0.2596 | 0.326 | ||
| Shortcut | 0.2460 | 0.2501 | 0.2555 | 0.324 |
| buffer | DROD | Width FjORD | Shortcut FjORD |
|---|---|---|---|
| 3,200 | |||
| 6,400 | |||
| 12,800 |
| fleet | buffer | fleet | high | mid | low |
|---|---|---|---|---|---|
| DROD | 3,200 | 0.2341 | 0.2423 | 0.2478 | |
| 6,400 | 0.3076 | 0.3161 | 0.3137 | ||
| 12,800 | 0.3516 | 0.3603 | 0.3523 | ||
| Width FjORD | 3,200 | 0.2080 | 0.2170 | 0.2170 | |
| 6,400 | 0.2701 | 0.2815 | 0.2708 | ||
| 12,800 | 0.3354 | 0.3472 | 0.3175 |
Federated Continual Learning
Federated continual learning enables distributed clients to adapt models to changing data while retaining past knowledge and keeping raw data local.
See all 37 papers