FedLin (Federated Linearized Learning with Client Heterogeneity and Gradient Sparsification)
Algorithms AIMech:FedLinFederatedLinearizedLearningWithClientHeterogeneityAndGradientSparsification DRAFT
Aliases: FedLin, Federated Linearized Learning, FedLin with Server and Client Gradient Sparsification
A federated optimization algorithm (Mitra et al. 2021, Section 3, Algorithm 1, Equation 8; Section 4, Algorithms 2-3) that resolves the speed-accuracy conflict under simultaneous objective and systems heterogeneity (variable local steps tau_i >= 1) by updating each client i via the fresh-gradient linearized correction x_{i,l+1}^{(t)} = x_{i,l}^{(t)} - eta_i (nabla f_i(x_{i,l}^{(t)}) - nabla f_i(bar_x_t) + g_t) with client-specific step-size eta_i = bar_eta / tau_i and current global gradient g_t = (1/m) sum_{i=1}^m nabla f_i(bar_x_t) (or error-compensated TOP-k sparsified gradients in Algorithms 2-3).
Identity Notes & Concept Scope
Distinct from SCAFFOLD (AIMech:ScaffoldStochasticControlledAveragingForFederatedLearning, Karimireddy et al. 2020), which uses stale control variates c_i from round t-1 that violate the exact single-step fixed-point property at x^* and lacks client- inversely scaled step-sizes eta_i = bar_eta / tau_i for arbitrary systems heterogeneity, and from FedProx (AIMech:FedProxProximalHeterogeneousFederatedOptimization) and FedNova (AIMech:FedNovaNormalizedAveragingForHeterogeneousFederatedOptimization), which Propositions 1-2 prove converge to distorted surrogate objectives under constant step-sizes.
Claims & Grounded Evidence
4 Claims
Mechanism
#r433-surrogate-distortion-of-fedprox-fednova-and-fedlin-linear-convergence
On heterogeneous quadratics f_i(x) = (1/2)(x - c_i)^T A_i (x - c_i), Propositions 1 and 2 prove that FedProx and FedNova converge to distorted surrogate minimizers weighted by Q_i = sum_{l=0}^{tau_i-1} [I - eta(A_i + beta I)]^l (Eqs. 5, 7, 9, 10), whereas FedLin (Algorithm 1) with eta_i = bar_eta / tau_i satisfies the exact fixed-point property x_{i,l}^{(t)} = x^* whenever bar_x_t = x^* and achieves global linear convergence f(bar_x_{T+1}) - f(x^*) <= (1 - 1 / (6 kappa))^T (f(bar_x_1) - f(x^*)) under arbitrary objective and systems heterogeneity (Theorem 1), matching a 2-client lower bound proving that local steps H cannot eliminate the kappa condition-number dependence (Theorem 6, Appendix E).
Mechanism
#r433-asymmetric-server-versus-client-top-k-sparsification-theorems
Under contractive TOP-k gradient sparsification with parameter delta = d / k, FedLin reveals a fundamental asymmetry between server-side and client-side compression (Section 4): server-side sparsification converges linearly to the exact optimum x^* even without error feedback at rate 1 - mu / (2 (delta_s + sqrt(delta_s)) delta_s L) (Theorem 7) and improves to 1 - 1 / (96 kappa delta_s) with server error feedback (Theorem 8) because ||nabla f(bar_x_t)|| -> 0 at x^*, whereas client-side sparsification with error feedback converges linearly only to an O(delta_c D^2) neighborhood proportional to heterogeneity D^2 when nabla f_i(x^*) != 0 (Theorem 9, Corollary 1).
Limitation
#r433-two-communication-phases-or-stale-gradient-tradeoff-limitation
Computing the fresh global gradient g_t = (1/m) sum_{i=1}^m nabla f_i(bar_x_t) requires an extra gradient-aggregation round per epoch, and stochastic noise in local mini-batch gradients reintroduces an O(bar_eta sigma^2) variance floor (Theorem 4).
Application
#r433-heterogeneous-logistic-regression-and-cifar10-federated-classification-application
Applied to heterogeneous federated logistic regression and neural network classification on MNIST and CIFAR-10 under non-IID client partitions and random local iteration counts tau_i.
⚠️ Curated Limitations & Negative Results
- Client-side TOP-k sparsification cannot eliminate the O(delta_c D^2) heterogeneity neighborhood without decaying step-sizes or uncompressed periodic syncs.
- Local steps tau_i do not reduce the condition-number dependence kappa in the worst-case heterogeneous setting (Theorem 6).
Artificial Intelligence Ontology (AIO) Grounding
Mapped AIO Classes
| AIO Target Class | Mapping Relation | Mapping Rationale |
|---|---|---|
AIO:FederatedLearning
Federated Learning
|
narrow_match | FedLin is a federated optimization algorithm that achieves linear convergence under simultaneous objective and systems heterogeneity and gradient sparsification. |
AIO:SparseLearning
Sparse Learning
|
related_match | FedLin analyzes contractive TOP-k gradient sparsification with error feedback at both server and clients. |
Evidence Sources & Primary Literature
3 Sources| Source ID | Publication / Document | Type / Access | Reference & Checksum |
|---|---|---|---|
r433-primary-pdf
|
Aritra Mitra, Rayana H. Jaafar, George J. Pappas, Hamed Hassani
|
primary paper
full_text
|
ID: https://proceedings.neurips.cc/paper/2021/hash/7a6bda9ad6ffdac035c752743b7e9d0f-Abstract.html
sha256:c8b14664dbf5…
|
r433-arxiv-abs
|
Aritra Mitra, Rayana H. Jaafar, George J. Pappas, Hamed Hassani
|
primary paper
abstract_only
|
ID: https://arxiv.org/abs/2102.07053
sha256:0df343f2f1d4…
|
r433-metadata-cr
|
Aritra Mitra, Rayana H. Jaafar, George J. Pappas, Hamed Hassani
|
primary paper
abstract_only
|
ID: https://proceedings.neurips.cc/paper/2021/hash/7a6bda9ad6ffdac035c752743b7e9d0f-Abstract.html
sha256:ffd1c450f32a…
|
Ontology & Grounding Topology
Open in interactive network graph →Curation Provenance
Curator: Antigravity
Method: ai_assisted
Model: Gemini
Created: 2026-10-19T08:36:00Z
Updated: 2026-10-19T08:36:00Z
Explore across X-Mech
Related computational & biological Mechs:
Embedding Map
Network Graph
ProteinTraitsMech (ESM-2 pLM)
DUFMech (Domain pLM)
TraitMech
CultureMech
All 14 X-Mech Repositories