Source-linked AI summary

On the Convergence of Federated Averaging with Cyclic Client Participation

Yae Jee Cho, Pranay Sharma, Gauri Joshi, Zheng Xu, Satyen Kale, Tong Zhang

arXiv:2302.03109v1cs.LGcs.DC

TL;DR

Prior analyses of FedAvg assume full participation or uniform client sampling, although practical cross-device availability can follow a cyclic pattern. This paper analyzes FedAvg under cyclic participation and finds that it can converge faster than standard FedAvg under suitable conditions.

  • Problem

    Prior FedAvg convergence analyses assume full client participation or uniform random sampling, while practical cross-device clients participate under local criteria and can follow cyclic availability patterns.

    Source: §1 · p. 2
  • Method

    The paper analyzes FedAvg with client groups becoming available cyclically, covering GD, SGD, and shuffled SGD as well as several FedAvg variants.

    Source: §1 · p. 2, §3 · p. 4
  • Results

    For global PL objectives and suitable conditions, cyclic participation can achieve an eO(1/T^2) error-convergence rate, faster than FedAvg’s standard O(1/T) rate.

    Source: §1 · p. 2
  • Takeaways & Limitations

    The framework characterizes how local procedures, client-group heterogeneity, and cycle length affect convergence, and identifies conditions where cyclic participation converges faster than standard FedAvg.

    Source: §6 · p. 12

    The local shuffled-SGD analysis assumes bounded intra-client component heterogeneity, with a uniform bound ν across clients, data components, and model parameters.

    Source: §4.4 · p. 9

Abstract

from arXiv · show

Federated Averaging (FedAvg) and its variants are the most popular optimization algorithms in federated learning (FL). Previous convergence analyses of FedAvg either assume full client participation or partial client participation where the clients can be uniformly sampled. However, in practical cross-device FL systems, only a subset of clients that satisfy local criteria such as battery status, network connectivity, and maximum participation frequency requirements (to ensure privacy) are available for training at a given time. As a result, client availability follows a natural cyclic pattern. We provide (to our knowledge) the first theoretical framework to analyze the convergence of FedAvg with cyclic client participation with several different client optimizers such as GD, SGD, and shuffled SGD. Our analysis discovers that cyclic client participation can achieve a faster asymptotic convergence rate than vanilla FedAvg with uniform client participation under suitable conditions, providing valuable insights into the design of client sampling protocols.

1 Introduction

Prior convergence analyses of FedAvg assume full participation or uniform client sampling, while practical cross-device systems exhibit cyclic availability. This paper analyzes FedAvg under cyclic participation and shows faster convergence under suitable conditions.

1 Introduction

  • Practical client availability is cyclic because participation depends on local criteria, observed diurnal patterns, and limits on how frequently clients may participate.These criteria include device idleness, charging, and unmetered network access.Source: §1 · p. 2
  • Under a global PL objective, cyclic client participation can achieve eO(1/T^2) error convergence under suitable conditions, faster than FedAvg’s standard O(1/T) rate.Here T is the number of communication rounds.Source: §1 · p. 2
  • The framework covers client subsampling within available groups, group count and data heterogeneity effects, and local GD, SGD, and shuffled SGD.It includes several FedAvg variants as special cases and matches prior bounds in those cases.Source: §1 · p. 2
  • Preliminary experiments show better test accuracy and training-loss convergence for cyclic participation than standard FedAvg.The passage reports this as preliminary experimental evidence.Source: §1 · p. 2

2 Related Work

Prior work studied partial and arbitrary client participation, privacy, communication efficiency, and shuffling-based optimization, but did not fully characterize convergence under realistic cyclic group availability. The paper relates cyclic participation to incremental-gradient and shuffling analyses while identifying gaps in earlier participation models.

2.1 Client Participation in FL.

  • Earlier FedAvg work focused on unbiased partial participation with O(1/T) convergence; cyclic participation had not been considered in that line of work.Related work also considered flexible or loss-dependent biased participation.Source: §2.1 · p. 2
  • An analysis under the regularized participation class does not capture how client ordering or the number of client groups within a cycle affects convergence; asynchronous client-specific cycles also fail to simulate groups sequentially becoming available to the server.Source: §2.1 · pp. 2–3
  • Prior cyclic-participation work examined privacy or communication efficiency, but one speedup bound had an unclear rate because its heterogeneity-dependent variance constant lacked bounds.The paper states that it gives a clear eO(1/T^2) speedup rate and identifies conditions for it.Source: §2.1 · p. 3

2.2 Shuffling-based methods.

  • Shuffling-based methods use each component once per epoch, making within-epoch progress approximate full-batch gradient descent; CyCP resembles incremental gradient at the server level.In this analogy, each client is interpreted as a sample.Source: §2.2 · p. 3
  • Shuffling analyses distinguish random reshuffling, shuffle once, and incremental gradient, with improved dependence on epoch count requiring a sufficiently large epoch regime.The cited bounds state that the improved dependence requires K larger than O(κ^a), where a is in [1, 2].Source: §2.2 · p. 3

3 Problem Formulation

The formulation divides clients into fixed groups that become available in a predetermined cycle, with a random subset sampled from each available group. It defines local update choices and includes several familiar FedAvg and shuffling methods as special cases.

3 Problem Formulation

  • The global objective is the average of clients’ empirical losses, and the server seeks a model minimizing that objective.The system has M clients, each with a local training dataset and loss function.Source: §3 · p. 3
  • In CyCP, M clients are split into K fixed, non-overlapping groups, traversed in a predetermined order, and N clients are sampled uniformly without replacement from each available group.Once selected, a client cannot participate for at least the next K−1 rounds.Source: §3 · pp. 4–5
  • CyCP captures participation limits and patterns associated with privacy requirements, time zones, and device-charging preferences.A timer can enforce that clients participate again only after a period corresponding to K.Source: §3 · p. 4
  • A cycle-epoch is the interval in which the server visits all client groups once, comprising K communication rounds.At each round, the server sends the current global model to the selected clients in the currently available group.Source: §3 · p. 4
  • The framework considers local GD, SGD, and shuffled SGD; selected clients compute updates that the server aggregates into the global update.Local SGD samples from a client’s dataset, while local SSGD processes disjoint components in a randomly permuted order.Source: §3 · pp. 4–5
  • With K = 1, CyCP becomes standard FedAvg with partial participation; with K = 1 and N = M, it also includes minibatch RR or local RR for the stated local-update settings.The paper states that its theoretical results match the bounds for these special cases.Source: §3 · p. 5

4 Convergence Analysis

This section derives convergence bounds for cyclic client participation under smoothness, a global PL objective, and assumptions that distinguish intra-group from inter-group data heterogeneity. Across GD, SGD, and shuffled SGD, faster convergence or lower total cost depends on the optimizer and system conditions.

4.1 Assumptions

  • The assumptions distinguish within-group heterogeneity γ from across-group heterogeneity α, yielding tighter bounds that separate their effects.The analysis assumes a global PL objective and notes that its treatment is restricted to PL functions for brevity and comparison with prior work.Source: §4.1 · p. 6

4.2 Convergence for CyCP with Local GD

  • For CyCP+GD, intra-group heterogeneity γ governs the dominant convergence term; when γ is close to zero, the rate can reach ˜O(1/T²).The section states that intra-group heterogeneity has a more significant contribution to error than inter-group heterogeneity α.Source: §4.2 · p. 7
  • For CyCP+GD, K = M/N is necessary for lower cost than Standard FedAvg; middle-range 1 < K < M/N costs more.The analysis also says the choice of N can be adjusted when K is naturally fixed.Source: §4.2 · p. 7

4.3 Convergence for CyCP with Local SGD

  • For both local SGD and SSGD, increasing K has no adverse impact on convergence because T = KK in the analysis.The SSGD analysis also finds the same dependency on K as CyCP+GD.Source: §4.2 · p. 7, §4.3 · p. 8, §4.4 · p. 9
  • For CyCP+SGD, the best convergence rate remains ˜O(1/T) when stochastic gradient variance is positive, even with K = M/N or γ close to zero.A lower cost than Standard FedAvg is possible when N = M/K and γ² ≥ Mσ²/Nτ.Source: §4.3 · p. 8

4.4 Convergence for CyCP with Local SSGD

  • For CyCP+SSGD, sufficiently large M and K = M/N make its total cost to reach ε error lower than local RR.The comparison favors smaller inter-group data heterogeneity and larger intra-group data heterogeneity.Source: §4.4 · pp. 10–11

5 Experimental Results

The experiments compare cyclic participation with Standard FedAvg across image-classification datasets, client-local procedures, and levels of data heterogeneity. Test accuracy generally improves with cyclic participation, while the reported improvement is larger in more heterogeneous settings.

Setup

  • The experiments use MLP image classifiers on FMNIST and 62-label EMNIST, with 100 or 500 clients and three partition seeds.They select 5 or 10 clients per round and partition data using Dirichlet distributions with α controlling heterogeneity.Source: §5 · p. 11

Effect of K and Data Heterogeneity

  • 5–10% higher test accuracy for FMNIST at α = 0.5 with CyCP (K > 1) than K = 1 (Standard FedAvg) across local procedures.The reported advantage also appears for K values between 1 and M/N.Source: Figure 2 · p. 11, §5 · pp. 11–12
  • 2–8% higher test accuracy for FMNIST at α = 2.0 with CyCP (K > 1) than Standard FedAvg, with a smaller gap than at α = 0.5.The text relates the smaller gap to lower data heterogeneity.Source: Figure 3 · p. 11, §5 · p. 12
  • CyCP has a larger test-accuracy improvement over Standard FedAvg on EMNIST at α = 0.05 than on FMNIST.The text attributes the larger gap to EMNIST's more heterogeneous partition.Source: Figure 4 · p. 12, §5 · p. 12

Difference Across Client Local Procedures

  • At the highest K, local SGD and SSGD test-accuracy curves oscillate as the server visits groups with increasing inter-group heterogeneity.The text reports similar oscillations in training-loss curves.Source: §5 · p. 12

6 Concluding Remarks

The concluding remarks frame cyclic participation as a practical pattern whose effect on FedAvg convergence had been unclear. The paper analyzes PL-objective convergence across GD, SGD, and shuffled SGD, identifies conditions for faster convergence than Standard FedAvg, and leaves broader objectives and changing or overlapping client groups for future work.

6 Concluding Remarks

  • The framework identifies conditions under which cyclic participation converges faster than Standard FedAvg and compares local procedures and special-case algorithms by cost to reach ε error.Future work includes non-PL objectives, general optimizers such as momentum, and client groups that are overlapping or not fixed during training.Source: §6 · p. 12

A Additional Experimental Details and Results

The appendix reports training-loss trends alongside experimental settings and proof details. It finds faster convergence at higher K, with larger gains under high data heterogeneity and oscillations in a specific high-K setting.

A Additional Experimental Details and Results

B Useful Inequalities

  • The appendix lists standard inequalities, including Young’s, Jensen’s, sum-of-squares, and without-replacement sampling results, for use in the proofs.The supplied passages name these results but do not show their full statements.Source: B Useful Inequalities · p. 18

C Proofs for the CyCP+Local GD Case

  • For the local-GD case, the model update is interpreted as noisy gradient descent, and the proof bounds its noise term.The effective learning rate is ηKN.Source: §C.1 · pp. 19–20

C.2 Intermediate Results

The intermediate results bound accumulated client-group gradients and the error induced by cyclic participation. The latter bound additionally depends on intra-group heterogeneity when clients are subsampled within groups.

C.2 Intermediate Results

  • The cyclic-participation error bound adds dependence on intra-group heterogeneity γ when clients are subsampled within each group.The result is stated alongside bounds on summed client-group gradients under Assumption 3.Source: §C.2 · p. 20

C.3 Proof of Theorem 1

The proof derives a convergence guarantee for cyclic participation with one local GD step under stated assumptions and a prescribed step size. It also identifies a condition under which choosing K > 1 improves the cost relative to K = 1.

C.3 Proof of Theorem 1

  • With the stated assumptions and step-size choice, the local-GD theorem gives a convergence-error bound when K ≥ 7κ log(MK2).The step size is η = log(MK2)/µNKK, and the bound suppresses logarithmic terms and numerical constants.Source: §C.3 · p. 21

C.4 Proofs on Intermediate Lemmas

  • The proof bounds update error terms using intermediate lemmas and the assumptions, then unrolls the resulting inequality to obtain the theorem’s convergence bound.The derivation invokes smoothness and the step-size restriction η ≤ 1/(7LNK).Source: §C.3 · pp. 21–22, §C.4 · p. 24

C.3 Proof of Theorem 1

  • For any N > 1, the stated cost comparison favors K > 1 over K = 1 only when K = M/N, corresponding to full client participation in every group.The communication-round formulation sets T = KK and notes that increasing K requires increasing T accordingly.Source: §C.3 · pp. 21, 23

D Proofs for the CyCP+Local SGD Case

D.2 Intermediate Results

The appendix bounds errors from local SGD and cyclic participation, then uses these bounds in a convergence theorem. It also gives a within-cycle trajectory bound and outlines the shuffled-SGD update construction.

D.2 Intermediate Results

  • The local-SGD error bounds depend mainly on the number of local steps τ and intra- and inter-group heterogeneity γ and α.The cyclic-participation error bounds additionally include a variance term σ2 from stochastic gradients.Source: §D.2 · p. 28

D.3 Proof of Theorem 2

  • Under Assumptions 1–4 and the stated step size, the local-SGD theorem bounds convergence error for K ≥ 10κ log(MK2) communication rounds.The bound is expressed in terms of T = KK with logarithmic terms and constants suppressed.Source: §D.3 · p. 29

D.4 Proof for Corollary 2

  • Except when K = M/N, the passage states that for γ ≥ A1σ2/τ, CSGD|K>1 > CSGD|K=1 requires an additional condition.The passage does not provide that condition’s full right-hand side.Source: §D.4 · p. 32

D.5 Proofs on Intermediate Lemmas

  • The intermediate-lemma proofs use smoothness, stochastic-gradient independence across clients, and unbiasedness to derive the error bounds.They also apply the previously established client-group gradient and trajectory bounds.Source: §D.5 · pp. 33, 39, 40, 42

E Proofs for the CyCP+Shuffled SGD Case

  • In shuffled SGD, each selected client initializes from the global model and processes its local loss components sequentially in shuffled-SGD steps.The component losses sum to the client’s local loss.Source: E Proofs for the CyCP+Shuffled SGD Case · p. 45

E.1 Intermediate Results

Lemmas E.1 and E.2 bound the error due to CyCP, with bounds shaped by the learning-rate and cycle-length conditions and by heterogeneity.

  • Lemma E.1 bounds the norm of the error term arising from CyCP.Source: §E.1 · p. 46
  • Lemma E.2 bounds the squared norm of the error term arising from CyCP.Source: §E.1 · p. 47
  • The bounds depend on ν = γ + α and use η = log(MK2)/τµNKK with K ≥10κ log (MK2).The passage identifies ν as intra-component heterogeneity and γ and α as intra-group and inter-group heterogeneity.Source: §E.1 · p. 47

E.2 Proof for Theorem 3

The proof derives a convergence bound under stated assumptions and step-size and cycle-length conditions, then relates it to communication rounds. It also gives a cost-comparison condition and bounds intermediate error terms.

E.2 Proof for Theorem 3

  • Under Assumptions 1–3 and 5, the theorem bounds convergence error for η = log(MBK2)/µKBK and K ≥10κ log (MBK2).Here κ = L/µ.Source: §E.2 · p. 47
  • The corollary expresses the bound in terms of total communication rounds T = KK, while the lower bound on K means increasing K requires increasing T.Source: §E.2 · p. 47

E.3 Proof for Corollary 3

  • For CyCP+Shuffled SGD to cost less than LocalRR, the proof states a required condition and its equivalent form.Source: §E.3 · p. 50

E.4 Proofs on Intermediate Lemmas

  • The proof separately bounds all three components of r(k,0), uses the same techniques for additional terms, and bounds squared noise-term norms, ultimately obtaining a final bound.Source: §E.4 · pp. 50, 53, 54
Loading 2302.03109v1…