DeepAFL: Deep Analytic Federated Learning
Organizations: Peking University, PRC · Key Lab of High Confidence Software Technologies (PKU), Ministry of Education, PRC · Central South University, PRC · University of Pennsylvania, USA ·
Abstract
Federated Learning (FL) is a popular distributed learning paradigm to break down data silo. Traditional FL approaches largely rely on gradient-based updates, facing significant issues about heterogeneity, scalability, convergence, and overhead, etc. Recently, some analytic-learning-based work has attempted to handle these issues by eliminating gradient-based updates via analytical (i.e., closed-form) solutions. Despite achieving superior invariance to data heterogeneity, these approaches are fundamentally limited by their single-layer linear model with a frozen pre-trained backbone. As a result, they can only achieve suboptimal performance due to their lack of representation learning capabilities. In this paper, to enable representable analytic models while preserving the ideal invariance to data heterogeneity for FL, we propose our Deep Analytic Federated Learning approach, named DeepAFL. Drawing inspiration from the great success of ResNet in gradient-based learning, we design gradient-free residual blocks in our DeepAFL with analytical solutions. We introduce an efficient layer-wise protocol for training our deep analytic models layer by layer in FL through least squares. Both theoretical analyses and empirical evaluations validate our DeepAFL's superior performance with its dual advantages in heterogeneity invariance and representation learning, outperforming state-of-the-art baselines by up to 5.68%-8.42% across three benchmark datasets.
Figures & tables
| Baseline | CIFAR-100 | Tiny-ImageNet | ||||||
|---|---|---|---|---|---|---|---|---|
| Non-IID-1 | Non-IID-2 | Non-IID-1 | Non-IID-2 | |||||
| FedAvg ( 2017 ) | ||||||||
| FedProx ( 2020 ) | ||||||||
| MOON ( 2021a ) | ||||||||
| FedGen ( 2021 ) | ||||||||
Appendix figures & tables34 assets
Supplementary material from the paper’s appendix.
Appendix
| Baseline | CIFAR-10 | |||
|---|---|---|---|---|
| Non-IID-1 | Non-IID-2 | |||
| FedAvg ( 2017 ) | ||||
| FedProx ( 2020 ) | ||||
| MOON ( 2021a ) | ||||
| FedGen ( 2021 ) | ||||
| Datasets | Layers | |||||||
|---|---|---|---|---|---|---|---|---|
| CIFAR-10 | ||||||||
| CIFAR-100 | ||||||||
| Datasets | Layers | |||||||
|---|---|---|---|---|---|---|---|---|
| CIFAR-10 | ||||||||
| CIFAR-100 | ||||||||
| Datasets | Layers | None | GELU | ReLU | LeakyReLU | Tanh | Hardwish | Softshrink |
|---|---|---|---|---|---|---|---|---|
| CIFAR-10 | ||||||||
| CIFAR-100 | ||||||||
| Datasets | Layers | |||||||
|---|---|---|---|---|---|---|---|---|
| CIFAR-10 | ||||||||
| CIFAR-100 | ||||||||
| Datasets | Layers | |||||||
|---|---|---|---|---|---|---|---|---|
| CIFAR-10 | ||||||||
| CIFAR-100 | ||||||||
| Datasets | Layers | |||||||
|---|---|---|---|---|---|---|---|---|
| CIFAR-10 | ||||||||
| CIFAR-100 | ||||||||
| Layers | Training Acc | Testing Acc | Time Cost | |||
|---|---|---|---|---|---|---|
| AFL | ||||||
| Layers | Training Acc | Testing Acc | Time Cost | |||
|---|---|---|---|---|---|---|
| AFL | ||||||
| Layers | Training Acc | Testing Acc | Time Cost | |||
|---|---|---|---|---|---|---|
| AFL | ||||||
| Layers | DeepAFL | Ablation (1) | Ablation (2) | Ablation (3) | Ablation (4) |
|---|---|---|---|---|---|
| {\color[rgb]{0,0,0}\textbf{85.35}_{\pm 0.02}} | |||||
| {\color[rgb]{0,0,0}\textbf{89.03}_{\pm 0.03}} | |||||
| {\color[rgb]{0,0,0}\textbf{90.61}_{\pm 0.05}} | |||||
| {\color[rgb]{0,0,0}\textbf{91.80}_{\pm 0.04}} | |||||
| {\color[rgb]{0,0,0}\textbf{92.74}_{\pm 0.03}} | |||||
| {\color[rgb]{0,0,0}\textbf{93.49}_{\pm 0.02}} |
| Layers | DeepAFL | Ablation (1) | Ablation (2) | Ablation (3) | Ablation (4) |
|---|---|---|---|---|---|
| Layers | DeepAFL | Ablation (1) | Ablation (2) | Ablation (3) | Ablation (4) |
|---|---|---|---|---|---|
| Layers | DeepAFL | Ablation (1) | Ablation (2) | Ablation (3) | Ablation (4) |
|---|---|---|---|---|---|
| Layers | DeepAFL | Ablation (1) | Ablation (2) | Ablation (3) | Ablation (4) |
|---|---|---|---|---|---|
| Layers | DeepAFL | Ablation (1) | Ablation (2) | Ablation (3) | Ablation (4) |
|---|---|---|---|---|---|
| Layers | DeepAFL | Strategy (1) | Strategy (2) | Strategy (3) | Strategy (4) |
|---|---|---|---|---|---|
| {\color[rgb]{0,0,0}\textbf{85.35}_{\pm 0.02}} | |||||
| {\color[rgb]{0,0,0}\textbf{89.03}_{\pm 0.03}} | |||||
| {\color[rgb]{0,0,0}\textbf{90.61}_{\pm 0.05}} | |||||
| {\color[rgb]{0,0,0}\textbf{91.80}_{\pm 0.04}} | |||||
| {\color[rgb]{0,0,0}\textbf{92.74}_{\pm 0.03}} | |||||
| {\color[rgb]{0,0,0}\textbf{93.49}_{\pm 0.02}} |
| Layers | DeepAFL | Strategy (1) | Strategy (2) | Strategy (3) | Strategy (4) |
|---|---|---|---|---|---|
| Layers | DeepAFL | Strategy (1) | Strategy (2) | Strategy (3) | Strategy (4) |
|---|---|---|---|---|---|
| Layers | DeepAFL | Strategy (1) | Strategy (2) | Strategy (3) | Strategy (4) |
|---|---|---|---|---|---|
| Layers | DeepAFL | Strategy (1) | Strategy (2) | Strategy (3) | Strategy (4) |
|---|---|---|---|---|---|
| Layers | DeepAFL | Strategy (1) | Strategy (2) | Strategy (3) | Strategy (4) |
|---|---|---|---|---|---|
| Baseline | CIFAR-10 | CIFAR-100 | Tiny-ImageNet | |||
|---|---|---|---|---|---|---|
| FedAWA ( 2025 ) | 79.42% | 76.37% | 58.94% | 50.48% | 59.07% | 52.33% |
| FedLPA ( 2024a ) | 49.57% | 42.56% | 40.04% | 22.55% | 39.48% | 26.67% |
| AFL ( 2025 ) | 80.75% | 80.75% | 58.56% | 58.56% | 54.67% | 54.67% |
| DeepAFL | 85.20% | 85.20% | 64.72% | 64.72% | 60.31% | 60.31% |
| DeepAFL | 85.93% | 85.93% | 65.96% | 65.96% | 61.37% | 61.37% |
| Layers | |||||
|---|---|---|---|---|---|
| FedAvg | 56.62% | 55.83% | 5% | 3% | 1% |
| FedDyn | 57.55% | 57.22% | 56.26% | 3% | 1% |
| Layers | Testing Acc | Time Cost | ||||
|---|---|---|---|---|---|---|
| AFL | 75.45% | 107.19s | ||||
| DeepAFL ( ) | 78.05% | 2.60% | 2.60% | 118.97s | 11.78s | 11.78s |
| DeepAFL ( ) | 79.01% | 0.96% | 3.56% | 126.55s | 7.58s | 19.37s |
| DeepAFL ( ) | 79.73% | 0.72% | 4.28% | 133.79s | 7.24s | 26.60s |
| DeepAFL ( ) | 80.13% | 0.40% | 4.68% | 141.02s | 7.23s | 33.83s |
| DeepAFL ( ) | 80.33% | 0.20% | 4.88% | 148.56s | 7.54s | 41.37s |
| Layers | Testing Acc | Time Cost | ||||
|---|---|---|---|---|---|---|
| AFL | 86.35% | 107.19s | ||||
| DeepAFL ( ) | 87.71% | 1.36% | 1.36% | 118.97s | 11.78s | 11.78s |
| DeepAFL ( ) | 88.15% | 0.44% | 1.80% | 126.55s | 7.58s | 19.37s |
| DeepAFL ( ) | 88.49% | 0.34% | 2.14% | 133.79s | 7.24s | 26.60s |
| DeepAFL ( ) | 88.80% | 0.31% | 2.45% | 141.02s | 7.23s | 33.83s |
| DeepAFL ( ) | 88.95% | 0.15% | 2.60% | 148.56s | 7.54s | 41.37s |
| Layers | Consistent Participation | Inconsistent Participation | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 64.72% | 64.53% | 64.38% | 63.92% | 62.85% | 61.54% | 64.72% | 64.21% | 64.03% | 64.05% | 62.19% | 61.36% | |
| 65.96% | 65.46% | 65.26% | 64.49% | 64.76% | 63.17% | 65.96% | 65.36% | 65.05% | 64.45% | 64.09% | 62.93% | |
| 66.56% | 66.39% | 66.20% | 65.38% | 65.17% | 65.08% | 66.56% | 66.22% | 66.17% | 65.64% | 64.60% | 64.98% | |
| 66.98% | 66.76% | 66.66% | 66.47% | 66.27% | 65.14% | 66.98% | 66.57% | 66.51% | 66.32% | 65.50% | 64.81% | |
| Layers | ||||||
|---|---|---|---|---|---|---|
| DeepAFL ( ) | 64.72% | 64.20% | 63.63% | 63.25% | 62.51% | 60.91% |
| DeepAFL ( ) | 65.96% | 65.43% | 65.09% | 63.96% | 62.98% | 60.99% |
| DeepAFL ( ) | 66.56% | 66.11% | 65.66% | 64.29% | 63.15% | 60.97% |
| DeepAFL ( ) | 66.98% | 66.38% | 66.11% | 64.69% | 63.35% | 60.45% |
| Layer | Training CSR | Testing CSR | Training IFS | Testing IFS | Training DM | Testing DM |
|---|---|---|---|---|---|---|
| DeepAFL ( ) | 1.583 | 1.529 | 3.25 | 3.13 | 3.21 | 3.09 |
| DeepAFL ( ) | 1.565 | 1.520 | 3.20 | 3.11 | 3.16 | 3.07 |
| DeepAFL ( ) | 1.555 | 1.516 | 3.17 | 3.10 | 3.14 | 3.06 |
| DeepAFL ( ) | 1.546 | 1.513 | 3.15 | 3.09 | 3.12 | 3.05 |
| DeepAFL ( ) | 1.539 | 1.510 | 3.13 | 3.08 | 3.11 | 3.04 |