AIMech AI

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

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 →
CONCEPT KIND Algorithms AIMech:FedLinFederatedLinearizedLearningWithClientHeterogeneityAndGradientSparsification FedLin (Federated... 4 claims · 3 sources AIO GROUNDING AIO:FederatedLearning (2)

Curation Provenance

Curator: Antigravity
Method: ai_assisted
Model: Gemini
Created: 2026-10-19T08:36:00Z
Updated: 2026-10-19T08:36:00Z
Source File: data/curated/FedLinFederatedLinearizedLearningWithClientHeterogeneityAndGradientSparsification.yaml · Schema: curated_concept.yaml

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