May 24, 2026
Asynchronous decentralized federated learning (ADFL) eliminates central coordination and global synchronization, making it attractive for large-scale and heterogeneous systems. However, frequent peer-to-peer communication, asynchronous updates on directed topologies, and non-IID data jointly lead to excessive communication overhead, biased aggregation and severe model drift. We propose PushCen-ADFL, a communication-efficient ADFL framework that enables stable training under asymmetric communication and delayed client participation. PushCen-ADFL couples communication, aggregation, and local stabilization in a shared centroid representation space, forming a closed loop between compression and optimization. Clients exchange centroid-form messages, apply average-preserving push-sum mixing to correct aggregation bias, and use a lightweight centroid regularization anchored in the same centroid space to mitigate drift under heterogeneity and staleness. A bounded, sender-deduplicated buffer further improves robustness under irregular asynchronous arrivals. Experiments on vision datasets demonstrate that PushCen-ADFL improves accuracy under data heterogeneity by up to 6% while reducing per-push communication cost by more than 80%, achieving a favorable accuracy-communication trade-off.
<ccs2012> <concept> <concept_id>10010147.10010257.10010293.10010294</concept_id> <concept_desc>Computing methodologies Neural networks</concept_desc> <concept_significance>300</concept_significance> </concept> <concept> <concept_id>10010147.10010257.10010321.10010337</concept_id> <concept_desc>Computing methodologies Regularization</concept_desc> <concept_significance>300</concept_significance> </concept> <concept> <concept_id>10010147.10010169.10010170.10010174</concept_id> <concept_desc>Computing methodologies Massively parallel algorithms</concept_desc> <concept_significance>500</concept_significance> </concept> <concept> <concept_id>10010147.10010169.10010170.10003824</concept_id> <concept_desc>Computing methodologies Self-organization</concept_desc> <concept_significance>300</concept_significance> </concept> <concept> <concept_id>10010147.10010178.10010219.10010222</concept_id> <concept_desc>Computing methodologies Mobile agents</concept_desc> <concept_significance>100</concept_significance> </concept> </ccs2012>
Federated Learning (FL) has emerged as a privacy-preserving distributed machine learning paradigm that enables collaborative model training when data cannot be centrally shared [1]–[3]. By performing training on local devices and exchanging only model updates, FL allows distributed data resources to be effectively utilized without direct access to raw data and has demonstrated significant potential in applications such as mobile intelligence, the Internet of Things and cross-organizational collaboration [4], [5].
However, most existing federated learning frameworks adopt a centralized and synchronous training paradigm, where a central server collects model updates from all clients and aggregates them in each training round. While such designs perform well under ideal network conditions, they face multiple challenges in real-world systems. First, the central server introduces a single point of bottleneck and failure, which limits scalability due to constraints on communication bandwidth, computation capacity and reliability in large-scale deployments [6]. Second, synchronous training requires strict temporal alignment of clients at round boundaries, making the system vulnerable to device heterogeneity and system variability and leading to the well-known straggler problem that significantly degrades training efficiency [7], [8]. Finally, centralized architectures are often difficult to deploy in cross-organizational or cross-domain scenarios and conflict with the growing demand for trustless and autonomous systems [6], [9].
To overcome these limitations, Decentralized Federated Learning (DFL) has attracted increasing attention. In DFL, clients exchange model information only with their neighboring nodes without relying on a central server, thereby improving system robustness and scalability [9], [10]. Nevertheless, most existing decentralized approaches still rely on synchronous update mechanisms, assuming that clients remain aligned across logical training rounds, an assumption that is difficult to satisfy in dynamic network environments [11], [12].
Consequently, Asynchronous Decentralized Federated Learning (ADFL) has been recognized as a more practical training paradigm for real-world deployments [12], [13]. ADFL allows clients to perform local updates and communications at different times without global synchronization, while supporting model collaboration solely through neighbor-to-neighbor communication [14], [15]. This setting naturally accommodates device heterogeneity, network variability and partial participation and enables clients to join the training process at arbitrary times [16], [17]. As a result, ADFL is well suited for edge computing, peer-to-peer networks and large-scale distributed systems, offering significant advantages in terms of system throughput, fault tolerance and deployment flexibility.
Despite these advantages, such a highly flexible training paradigm also introduces new challenges. Specifically, ADFL systems often face three fundamental difficulties simultaneously. First, since clients communicate only with their neighbors and model exchanges may occur frequently, directly transmitting full model parameters incurs prohibitive communication overhead, which is difficult to sustain in bandwidth-constrained or large-scale systems [1], [18]. Compared to centralized settings, decentralized architectures lack unified communication scheduling and aggregation, making communication efficiency a more prominent concern [9], [19]. Second, asynchronous updates combined with asymmetric communication topologies can lead to biased aggregation [20], [21]. In practice, clients may push models at different times and communication links can be unbalanced, under which naive neighbor averaging fails to guarantee unbiased consensus [15], [22]. Finally, the widely observed statistical heterogeneity (non-IID data) across clients is further amplified in asynchronous decentralized environments [12], [14], [23]. Local updates computed on non-identically distributed data tend to induce significant model drift, while asynchronous communication may delay or mismatch these updates, exacerbating training instability and degrading final performance [24], [25].
These challenges are not independent. Communication constraints necessitate compression. Compression can amplify client drift under non-IID data. Asynchrony further delays and skews these updates. Together, these effects exacerbate aggregation bias in decentralized training. Accordingly, the central research question addressed in this work is:
How can we jointly design compression, de-biased aggregation, and local-update stabilization for accurate and stable ADFL under asymmetric communication and non-IID data?
To address this research question, we propose a new ADFL framework, termed PushCen-ADFL (Push-Sum Centroid Asynchronous Decentralized Federated Learning). The proposed framework is built around lightweight centroid structures and integrates centroid-based compression with push-sum aggregation, where both aggregation and local optimization are constrained in the same centroid representation space, enabling communication-efficient and stable federated training without relying on a central server or global synchronization.
Our main contributions are summarized as follows:
We propose PushCen-ADFL, an asynchronous framework that unifies centroid-based communication, push-sum de-biasing and buffered neighbor aggregation into a single decentralized training pipeline, where both aggregation and local optimization operate in the same centroid representation space.
We design an average-preserving push-sum aggregation with mass splitting and a deduplicated bounded buffer, mitigating bias from asymmetric information flow and preventing stale messages from dominating aggregation.
We develop a centroid-aligned proximal regularizer that stabilizes local updates in the same compressed centroid space used for communication. This alignment mitigates non-IID data distributions by contracting local trajectories toward a shared compressed reference.
We provide a convergence analysis for PushCen-ADFL and validate it on CIFAR-10, CIFAR-100 and Tiny-ImageNet, improving accuracy by up to 6% over communication-efficient baselines, while reducing per-push transmitted payload by more than 80% less than full-model communication.
A representative direction is to design wait-free or fully asynchronous communication mechanisms to mitigate stragglers. SWIFT proposes a wait-free decentralized FL algorithm that allows each client to proceed without synchronization barriers, while maintaining standard iteration complexity guarantees [11]. Complementarily, A\(^2\)CiD\(^2\) studies asynchronous decentralized deep learning from an optimization perspective and introduces a continuous local momentum to accelerate randomized gossip-style communication [26]. Beyond communication primitives, a second line of work explicitly addresses the staleness and heterogeneity issues that arise in ADFL. Ma et al. propose dynamic staleness control for ADFL in decentralized topologies, aiming to balance training efficiency with model quality by regulating the tolerated staleness degree and deriving scheduling/control policies with theoretical support [23]. For heterogeneous devices, Liao et al. develop AsyDFL, which integrates neighbor selection and gradient pushing to reduce communication cost and completion time under non-IID data and system heterogeneity [27]. More recently, DSpodFL provides a unified viewpoint by modeling both local updates and pairwise exchanges as sporadic random events, thereby capturing time-varying computation/communication patterns [28]. In addition, AEDFL proposes an ADFL framework that leverages reinforcement learning for neighbor model selection in heterogeneous, serverless training settings [12]. Despite these advances, most existing ADFL methods focus on asynchronous scheduling or staleness control while assuming full-precision communication, and thus fail to jointly address frequent peer-to-peer exchanges, aggregation bias on asymmetric communication, and model drift under non-IID data, limiting their practicality in communication-constrained decentralized systems.
A prominent line of research leverages knowledge distillation to reduce the communication payload. FedKD adopts a local teacher-student scheme and transmits only a compact student (or distilled knowledge), substantially lowering uplink/downlink costs while preserving accuracy via distillation [29]. Another direction exploits sparsity to shrink communicated updates: SpaFL induces structured sparsity with low overhead through trainable thresholds, reducing both communication and computation [30]. In addition, low-precision methods cut the bit-width of local training and transmitted updates; Li et al. show that low-precision local training can already achieve competitive accuracy in FL [31]. Communication efficiency has also been explored under asynchronous decentralized settings: DivShare exchanges model fragments per interaction to better tolerate stragglers and bandwidth heterogeneity, improving wall-clock convergence in peer-to-peer training [32]. Overall, prior communication-efficient FL methods reduce communication cost but are not designed to handle the coupled challenges of aggregation bias and training instability in fully asynchronous decentralized environments.
We consider an ADFL system with \(N\) clients, denoted by \(\mathcal{V}=\{1,2,\dots,N\}\). Client communications are modeled by a directed graph \(\mathcal{G}=(\mathcal{V},\mathcal{E})\) induced by asynchronous gossip, where \((j \rightarrow i)\in\mathcal{E}\) indicates that client \(i\) can receive information from client \(j\). The communication topology can be asymmetric and unbalanced and there is no central server; clients communicate only with their neighbors. We denote the in-neighbors and out-neighbors of client \(i\) as \(\mathcal{N}_i^-\) and \(\mathcal{N}_i^+\), respectively.
Each client \(i\) holds a private local dataset \(\mathcal{D}_i\) with potentially heterogeneous (non-IID) data distributions, and maintains a local model copy \(w_i\). Let \(f_i(w)\) denote the empirical risk on \(\mathcal{D}_i\). We adopt a uniform client-level objective and formulate decentralized learning as a consensus optimization problem: \[\min_{\{w_i\}_{i=1}^{N}}\;\frac{1}{N}\sum_{i=1}^{N} f_i(w_i) \quad \text{s.t.}\quad w_i = w_j,\;\forall (j\rightarrow i)\in\mathcal{E}.\]
where the constraints \(w_i = w_j\) specify the desired agreement among local models at convergence, while local models may differ during asynchronous training. The system operates in a fully asynchronous manner without global synchronization. Clients perform local computation and communication at arbitrary times and received messages may correspond to stale model states due to heterogeneous computation speeds and network delays. Moreover, clients may become active at arbitrary times, and only active clients generate updates and participate in communication.
Under directed and potentially unbalanced topologies, achieving the above consensus objective requires decentralized aggregation mechanisms that preserve the uniform client-level average. Accordingly, the problem addressed in this work is to design a communication-efficient ADFL framework that enables stable and robust training under asymmetric communication and non-IID data distributions.
The overall workflow of PushCen-ADFL is illustrated in Figure 1 and summarized in Algorithm 2.
Clients exchange information only with their neighbors over a peer-to-peer communication, without relying on a central coordinator or round-based synchronization.
PushCen-ADFL is built on three coupled principles. First, each push transmits a centroid message \((V_i, A_i)\) instead of dense parameters, so the communication payload is controlled by the centroid budget. Second, aggregation follows a mass split push-sum update, using the mass push-sum \(s_i\) to correct bias and preserve the averaging under directed asynchronous mixing. Third, local optimization is stabilized by a centroid proximal term anchored at the current dictionary estimate \(G_i\), so that both communication and stabilization operate in the same centroid representation space. In addition, a deduplicated bounded buffer retains only the most recent message per sender, preventing stale or repeated messages from dominating the event-driven aggregation.
Concretely, each client \(i\) maintains a local model \(w_i\), a push-sum mass \(s_i\) that tracks the mixing weight for de-biased aggregation, and a local centroid dictionary estimate \(G_i\) that defines the shared compressed representation space, together with a bounded buffer \(\mathcal{B}_i\) that stores recent neighbor messages (line 1). When a neighbor message \((V_j, A_j, \sigma_{j\rightarrow i}, j)\) arrives, client \(i\) updates \(\mathcal{B}_i\) via BufferUpdate (lines 3–5), which replaces older entries from the same sender and enforces a fixed buffer capacity; see Section 4.5. Upon a local compute event, client \(i\) aggregates buffered information using PushSumAgg (line 6), which jointly updates \((w_i, G_i, s_i)\) in an average-preserving manner under directed and potentially unbalanced communication; see Section 4.4. Client \(i\) then performs LocalUpdate on \(\mathcal{D}_i\) (line 7). For communication, the model is encoded in a centroid form \((V_i, A_i)\) (Section 4.3); for optimization, the update is anchored to a centroid-based reference constructed from \(G_i\), yielding a centroid-aligned proximal effect that contracts local trajectories and mitigates non-IID drift; see Section 4.2. Finally, client \(i\) splits its push-sum mass (line 8) and sends \((V_i, A_i, \sigma_i, i)\) to all out-neighbors (line 9), ensuring that subsequent asynchronous aggregations remain average-preserving.
Non-IID data can cause client drift, which is further amplified by asynchronous execution. To mitigate this drift, PushCen-ADFL introduces a lightweight centroid regularization in the local update step (Algorithm 3). Client \(i\) first applies WCP to initialize the assignments \(A_i\) and pruning mask \(M_i\) (line 1), and then constructs a centroid anchor from its current centroid dictionary estimate \(G_i\) (line 2), where \(A_i\) specifies how centroids are indexed to form a layer-wise reference. We instantiate the centroid anchor \(\tilde{w}_i\) by decoding the assignment map \(A_i\) with the centroid dictionary \(G_i\) through an element-wise lookup: \[\tilde{w}_{i,l} \triangleq G_{i,l}[A_{i,l}], \label{eq:anchor-def}\tag{1}\] where each entry of \(\tilde{w}_{i,l}\) takes the centroid value indexed by the corresponding entry in \(A_{i,l}\), yielding the same shape as \(w_{i,l}\).
Client \(i\) then performs local optimization with a proximal-style regularizer while enforcing pruning: \[\min_{w_i}\;f_i(w_i;\mathcal{D}_i) + \lambda \|w_i - \tilde{w}_i\|_2^2, \label{eq:cen-reg}\tag{2}\] where \(M_i\) is a binary mask and masked coordinates are kept inactive via \(w_i \leftarrow w_i \odot M_i\) (lines 3–5). Here, \(\lambda\) controls the regularization strength and \(\tilde{w}_i\) is fixed during the local update. Unlike conventional proximal terms that pull toward a global model, \(\tilde{w}_i\) is decoded from the current centroid dictionary, keeping \(w_i\) close to a centroid-structured reference and reducing re-encoding distortion before communication. After local training, client \(i\) applies Weight Clustering Pruning (WCP) to refresh the centroid representation \((V_i,A_i)\) for subsequent communication (line 6) and returns \((w_i,V_i,A_i)\) (line 7).
The motivation is to stabilize asynchronous local optimization without decoupling it from centroid-based compression. In fully asynchronous decentralized training, a client can take many local steps before receiving fresh information; under non-IID data, this can move \(w_i\) into regions poorly represented by the current centroid dictionary, leading to unstable trajectories and larger distortion when the model is re-encoded. The proximal term \(\lambda\|w_i-\tilde{w}_i\|_2^2\) provides a local contraction toward \(\tilde{w}_i\), curbing drift while keeping the iterate close to the dictionary-induced quantized manifold. Figure 4 provides a visualization of centroid regularization. Because \(\tilde{w}_i\) is constructed from \((G_i,A_i)\), the regularizer directly ties the optimization path to the centroid structure used for communication, yielding more stable local updates under asynchrony.
PushCen-ADFL compresses communication by encoding each weight layer in a centroid form rather than transmitting dense parameters. For layer \(l\), client \(i\) vectorizes weights \(\theta=\mathrm{vec}(w_{i,l})\in\mathbb{R}^{d_l}\) and clusters its entries into \(K\) scalar centroids, producing a centroid table \(V_{i,l}\in\mathbb{R}^{K}\) and an assignment map \(A_{i,l}\in\{1,\dots,K\}^{d_l}\). Reconstruction uses an element-wise lookup \(w_{i,l}\approx V_{i,l}[A_{i,l}]\), so the payload scales with \(K\) (typically \(K\ll d_l\)) rather than \(d_l\).
Algorithm 5 implements this encoding via WCP. When available, WCP warm-starts centroids from the local dictionary estimate \(G_{i,l}\in\mathbb{R}^{K}\) by setting \(\mu=[0;\,G_{i,l}[1{:}K\!-\!1]]\), fixing one centroid at \(0\) and refining the others by alternating nearest-centroid assignment and cluster-mean updates. It returns \((V_{i,l},A_{i,l})\) and a binary pruning mask \(M_{i,l}\), where SortRemap reorders centroid indices into a canonical order and remaps \(A_{i,l}\) accordingly to ensure consistent encoding. Full cost details are deferred to Appendix 8.
Asynchronous decentralized training is inherently imbalanced: fast clients (due to higher compute speed or lower latency) can generate and disseminate updates more frequently, which may cause their local trajectories to dominate aggregation when using naive neighbor averaging. This issue is compounded on directed or unbalanced graphs, where information flow is asymmetric. PushCen-ADFL addresses both effects with an average-preserving push-sum style aggregation that tracks a scalar mass and uses it to normalize heterogeneous message rates and topology asymmetry.
Each client \(i\) maintains a local model \(w_i\), a push-sum mass \(s_i>0\), a local estimate of the centroid dictionary \(G_i\) and a buffer \(\mathcal{B}_i\) that stores neighbor messages received. A buffered message from neighbor \(j\) is denoted by \((V_j, A_j, \sigma_{j\rightarrow i}, j)\), where \((V_j,A_j)\) is the centroid representation and \(\sigma_{j\rightarrow i}\) is the mass share sent to \(i\). Upon aggregation, client \(i\) implicitly reconstructs a quantized neighbor model \(\hat{w}_j=\mathcal{R}(V_j,A_j)\) (Algorithm 6, line 4). It then computes the total mass \[\begin{align} S \leftarrow s_i + \sum_{(V_j,A_j,\sigma_{j\rightarrow i},j)\in\mathcal{B}_i}\sigma_{j\rightarrow i}, \label{eq:pushsum-mass} \end{align}\tag{3}\] which serves as the normalization constant for mass-conserving averaging. Client \(i\) updates its model via \[\begin{align} w_i \leftarrow \frac{s_i}{S}w_i + \sum_{(V_j,A_j,\sigma_{j\rightarrow i},j)\in\mathcal{B}_i}\frac{\sigma_{j\rightarrow i}}{S}\,\hat{w}_j, \label{eq:pushsum-model} \end{align}\tag{4}\]
so that a sender’s influence is determined by conserved mass rather than raw message frequency. The same weights are applied to update the centroid dictionary estimate using the received centroid tables. \[\begin{align} G_i \leftarrow \frac{s_i}{S}G_i + \sum_{(V_j,A_j,\sigma_{j\rightarrow i},j)\in\mathcal{B}_i}\frac{\sigma_{j\rightarrow i}}{S}\,V_j. \label{eq:pushsum-pool} \end{align}\tag{5}\] aligning the shared centroid structure with the aggregated model trajectory. After aggregation, client \(i\) sets \(s_i\leftarrow S\) and clears \(\mathcal{B}_i\).
After local training, client \(i\) performs mass splitting before communication. Let \(d_i^{+}=|\mathcal{N}_i^{+}|\) be the out-degree; client \(i\) sends \(\sigma_i=s_i/(d_i^{+}+1)\) to each out-neighbor and updates \(s_i\leftarrow \sigma_i\). This mass-splitting rule prevents fast clients from dominating the network dynamics under asynchrony.
In a fully asynchronous decentralized setting, message arrivals are decoupled from local compute events. A client may receive neighbor messages at arbitrary times, possibly in bursts, and some messages can be stale by the time aggregation is triggered. PushCen-ADFL therefore maintains a local buffer \(\mathcal{B}_i\) at each client \(i\) to cache received neighbor information until the next compute event, when aggregation is performed.
The buffer is designed to be sender-deduplicated and capacity-bounded. When a new message from neighbor \(j\) arrives, client \(i\) removes any existing buffered entry from \(j\) and keeps only the latest one. This enforces the principle that each sender contributes at most one message per aggregation event, preventing bursty arrivals from being counted multiple times and distorting the intended neighbor weighting, which per-message push-sum normalization alone does not prevent.
Empirically, removing sender deduplication consistently degrades performance (Section 5.2.4), indicating that discarding older messages from the same neighbor is beneficial in practice. Finally, to control memory and limit the influence of long-delayed messages under high communication rates, we impose a buffer size limit and drop the oldest entries upon overflow. Together, these rules provide a lightweight interface between asynchronous communication and push-sum aggregation, as summarized in Algorithm 7.
We summarize the assumptions used in our convergence analysis; full definitions, update rules and proofs are deferred to Appendix 9.
Each \(f_i\) is \(L\)-smooth and \(F(w)\triangleq \frac{1}{N}\sum_{i=1}^N f_i(w)\) is lower bounded by \(F^\star\).
\(\mathbb{E}[g_i(w;\xi)\mid w]=\nabla f_i(w)\) and \(\mathbb{E}[\|g_i(w;\xi)-\nabla f_i(w)\|_2^2\mid w]\le \sigma^2\).
For all \(i\) and \(w\), \(\|\nabla f_i(w)-\nabla F(w)\|_2^2\le \zeta^2\).
Any message used at event \(t\) is generated within the last \(\tau\) events.
The time-varying directed communication is \(B\)-window strongly connected and push-sum masses stay positive.
\(\|\mathcal{R}(\mathcal{C}(w)) - w\|_2 \le \varepsilon_c\) for any transmitted model \(w\).
To simplify constants, we further assume bounded push-sum masses, bounded iterates, and bounded gradient second moments. These are given in Assumptions 9.6.1.1 in Appendix 9.
On directed and potentially unbalanced graphs, the arithmetic mean is not preserved. We therefore analyze the de-biased push-sum reference \(\bar w^t \triangleq X_{\mathrm{tot}}^t/Y_{\mathrm{tot}}^t\) (defined via system-level accounting that includes buffered/in-flight messages; Appendix 9) and measure convergence by \(\frac{1}{T}\sum_{t=0}^{T-1}\mathbb{E}\|\nabla F(\bar w^t)\|_2^2\).
Theorem 1 (Stationarity of PushCen-ADFL). Suppose Assumptions1 - 6 and Assumptions 9.6.1.1 hold. Let the stepsize satisfy \(\eta \le \min\Big\{\frac{1}{8LE},\;\frac{1}{4\lambda}\Big\}\). Then \[\label{eq:main-body} \begin{align} \frac{1}{T}\sum_{t=0}^{T-1}\mathbb{E}\big[\|\nabla F(\bar w^t)\|_2^2\big] \;&\le\; \frac{2\big(F(\bar w^0)-F^\star\big)}{\eta E\,\underline{\gamma}\,T} + 2(\sigma^2+\zeta^2) + 2L\,\overline{c}^{\,2}\varepsilon_c^2 \\ &\quad+\; \frac{4L^2 N}{\underline{\gamma}} \left( \frac{\mathbb{E}[\mathcal{E}_{\mathrm{con}}^0]}{T(1-\rho)} + \frac{B_{\mathrm{con}}}{1-\rho} \right) + \frac{L}{\underline{\gamma}}\,D_\Delta . \end{align}\qquad{(1)}\]
where \(\underline{\gamma}=\frac{y_{\min}}{Ny_{\max}}\), \(\rho\in(0,1)\) is the contraction factor in the directed push-sum consensus recursion (Lemma 3 in Appendix 9) and \[\overline{c} \triangleq \max_t \frac{d_{i_t}^+}{d_{i_t}^+ + 1}\cdot \frac{y_{i_t}^{t+\frac{1}{3}}}{Y_{\mathrm{tot}}} \le \frac{y_{\max}}{Ny_{\min}}. \label{eq:cbar-body}\qquad{(2)}\] Moreover, \[\begin{align} D_\Delta &\triangleq 2E^2\eta^2\Big(G^2 + 16\lambda^2W^2\Big), \label{eq:Ddelta-body} \\ B_{\mathrm{con}} &\triangleq \big(C_1 + C_2\tau^2\big)D_\Delta + C_3\varepsilon_c^2, \label{eq:Bcon-body} \end{align}\] {#eq: sublabel=eq:eq:Ddelta-body,eq:eq:Bcon-body} with \((C_1,C_2,C_3)\) from Lemma 3.
4pt
| Method | CIFAR-10 | CIFAR-100 | Tiny-ImageNet | ||||||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 2-7(lr)8-13(lr)14-19 | \(\alpha=0.1\) | \(\alpha=0.4\) | \(\alpha=1.0\) | \(\alpha=0.1\) | \(\alpha=0.4\) | \(\alpha=1.0\) | \(\alpha=0.1\) | \(\alpha=0.4\) | \(\alpha=1.0\) | ||||||||||||
| 2-3(lr)4-5(lr)6-7 (lr)8-9(lr)10-11(lr)12-13 (lr)14-15(lr)16-17(lr)18-19 | Acc. \(\uparrow\) | SD \(\downarrow\) | Acc. \(\uparrow\) | SD \(\downarrow\) | Acc. \(\uparrow\) | SD \(\downarrow\) | Acc. \(\uparrow\) | SD \(\downarrow\) | Acc. \(\uparrow\) | SD \(\downarrow\) | Acc. \(\uparrow\) | SD \(\downarrow\) | Acc. \(\uparrow\) | SD \(\downarrow\) | Acc. \(\uparrow\) | SD \(\downarrow\) | Acc. \(\uparrow\) | SD \(\downarrow\) | |||
| Independent | 43.85 | 6.55 | 54.06 | 8.94 | 68.01 | 10.80 | 22.79 | 3.64 | 33.09 | 5.06 | 52.86 | 6.17 | 21.69 | 3.01 | 32.37 | 3.50 | 53.56 | 4.38 | |||
| Full-communication | |||||||||||||||||||||
| Async-DFedAvg | 57.72 | 6.84 | 63.05 | 8.70 | 74.22 | 9.55 | 41.14 | 4.28 | 46.60 | 3.75 | 60.06 | 7.54 | 43.79 | 2.68 | 50.97 | 3.00 | 63.56 | 3.99 | |||
| SWIFT | 56.93 | 6.46 | 64.11 | 8.11 | 72.59 | 9.69 | 41.92 | 4.36 | 47.59 | 4.07 | 50.96 | 7.75 | 43.68 | 2.68 | 50.65 | 3.33 | 63.75 | 3.85 | |||
| Communication-efficient | |||||||||||||||||||||
| DivShare | 51.96 | 6.62 | 60.25 | 8.53 | 69.46 | 9.88 | 32.93 | 3.44 | 40.64 | 4.39 | 56.67 | 6.52 | 30.52 | 3.03 | 38.75 | 3.49 | 56.05 | 3.99 | |||
| PushCen-ADFL | 58.30 | 6.32 | 66.22 | 7.82 | 73.89 | 9.28 | 35.15 | 3.59 | 44.21 | 4.25 | 57.23 | 7.58 | 36.70 | 2.69 | 44.62 | 3.43 | 58.04 | 3.99 | |||
5pt
| Method | CIFAR-10 | CIFAR-100 | Tiny-ImageNet | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 2-3(lr)4-5(lr)6-7 | Comm. (MB) \(\downarrow\) | \(\times\) Async \(\downarrow\) | Comm. (MB) \(\downarrow\) | \(\times\) Async \(\downarrow\) | Comm. (MB) \(\downarrow\) | \(\times\) Async \(\downarrow\) | |||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| Async-DFedAvg | 10.27 | 6.33 | 449.11 | 5.88 | 451.16 | 5.89 | |||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| DivShare | 2.06 | 1.27 | 89.82 | 1.18 | 90.23 | 1.18 | |||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| SWIFT | 10.27 | 6.33 | 449.11 | 5.88 | 451.16 | 5.89 | |||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| PushCen-ADFL | 1.622 | 1.00 | 76.33 | 1.00 | 76.65 | 1.00 | |||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
We conduct experiments on a Slurm-managed cluster with two compute nodes. Node A has an Intel Core i9-13900K CPU, 64GB RAM and one NVIDIA GeForce RTX 4090 GPU; Node B has an Intel Xeon Silver 4309Y CPU, 128GB RAM and two NVIDIA A40 GPUs. All methods are implemented in Python. Our method and all baselines are implemented in PyTorch 1 and trained/inferred on GPUs. To emulate client asynchrony and message arrivals in asynchronous decentralized FL, we implement an event-driven simulator that schedules local computation and communication events independently for each client, allowing updates and message deliveries to occur at arbitrary times. Unless otherwise stated, all experiments use a fixed random seed, set \(K=32\), and cap the FIFO buffer at \(L=16\) to bound memory under bursty asynchronous arrivals. Each client pushes to 10 random neighbors via a gossip protocol. The code was released at 2.
We evaluate our method on three standard vision classification benchmarks: CIFAR-10, CIFAR-100 and Tiny-ImageNet. Following common practice, we pair datasets with architectures of appropriate capacity: LeNet for CIFAR-10 and ResNet-18 for CIFAR-100 and Tiny-ImageNet [38]–[41]. All methods are trained under the same data preprocessing and training pipeline to ensure fair comparisons.
To emulate the statistical heterogeneity (non-IID data) commonly observed in real-world federated learning, we partition each dataset using a Dirichlet distribution with concentration parameters \(\alpha\in\{0.1,0.4,1.0\}\), where smaller \(\alpha\) induces more skewed client data distributions. CIFAR-10 is partitioned across 100 clients, while CIFAR-100 and Tiny-ImageNet are each distributed among 50 clients to avoid overly small per-client datasets under Dirichlet splits, which can make training unstable. For evaluation under each client’s local distribution, we further split each client’s data into local training and test sets and report performance on the corresponding local test data.
To evaluate delayed client participation in asynchronous decentralized federated learning, we adopt a round-free (pseudo-time–based) evaluation protocol consistent with ADFL. Training proceeds over a fixed global pseudo-time horizon, uniformly divided into 60 evaluation intervals, at which the average Top-1 accuracy of all online clients is reported. For each dataset, 10% of clients are randomly designated as delayed clients prior to training, while all remaining clients participate from the beginning. Each delayed client joins the system at a randomly sampled time during training. To ensure a fair comparison, both the identities of delayed clients and their join times are fixed and shared across all compared methods.
We compare against four representative baselines. Async-DFedAvg serves as a canonical benchmark, capturing a standard asynchronous decentralized adaptation of the FedAvg-style aggregation paradigm [1]. DivShare represents a state-of-the-art communication-efficient approach in asynchronous decentralized settings and is included to benchmark communication efficiency [32]. Independent performs purely local training without any model exchange, providing a non-collaborative lower bound. Finally, SWIFT is a representative asynchronous decentralized FL algorithm and is commonly used as a baseline for handling statistical heterogeneity under asynchrony [11].
Table 1 reports the Top-1 accuracy together with the corresponding standard deviation (SD) under different levels of non-IID data. In addition to mean performance, the SD provides a sensitivity analysis that characterizes the variability of model accuracy across clients and training trajectories in asynchronous decentralized settings. On CIFAR-10, PushCen-ADFL achieves the best accuracy under high heterogeneity, with particularly strong gains as data become more non-IID. At \(\alpha=0.4\) and \(\alpha=0.1\), PushCen-ADFL reaches 66.22% and 58.30%, respectively, outperforming the communication-efficient baseline DivShare by clear margins, while exhibiting SDs that are comparable to and often lower than, those of competing methods on CIFAR-10. When heterogeneity is mild (\(\alpha=1.0\)), PushCen-ADFL remains highly competitive with full-communication baselines and shows similar variability. On more challenging datasets (CIFAR-100 and Tiny-ImageNet), PushCen-ADFL consistently outperforms DivShare across all heterogeneity levels, with more pronounced improvements under stronger non-IID distributions. For example, at \(\alpha=0.1\), PushCen-ADFL improves accuracy by 2.22% on CIFAR-100 and 6.18% on Tiny-ImageNet, while maintaining overall variability comparable to the baselines across heterogeneity levels. PushCen-ADFL achieves competitive mean accuracy and comparable SDs relative to full-communication methods, demonstrating that its centroid-based compression and regularization enable strong, robust performance in heterogeneous and asynchronous decentralized training without increasing sensitivity to updates.
Table 2 reports the communication cost of a single client push event, which is the total payload transmitted when a client broadcasts its message to 10 neighbors. This metric isolates the per-push communication footprint from the overall pseudo-time horizon and thus directly reflects the bandwidth burden induced by one communication action in our asynchronous decentralized protocol. We focus on the per-push cost because total system communication scales with the per-push payload times the number of push events. The event count depends on asynchrony, client delays, and scheduling, rather than the message format itself. As expected, full-precision methods that transmit dense model information (Async-DFedAvg and SWIFT) incur substantially larger per-push traffic, reaching 10.27 MB on CIFAR-10 and about 449 to 451 MB on CIFAR-100 and Tiny-ImageNet. In contrast, communication-efficient baselines significantly reduce the payload size. DivShare lowers the per-push cost to 2.06 MB (CIFAR-10) and 89.82 to 90.23 MB (CIFAR-100 and Tiny-ImageNet), while PushCen-ADFL achieves the smallest per-push footprint across all datasets, requiring only 1.62 MB on CIFAR-10 and 76.33 to 76.65 MB on CIFAR-100 and Tiny-ImageNet.
We further normalize the per-push communication volume to PushCen-ADFL, where PushCen-ADFL is 1.00 by definition. Under this normalization, DivShare requires 1.27\(\times\) more bandwidth per push on CIFAR-10 and 1.18\(\times\) on CIFAR-100 and Tiny-ImageNet, whereas Async-DFedAvg and SWIFT require 6.33\(\times\) on CIFAR-10 and about 5.88 to 5.89\(\times\) on CIFAR-100 and Tiny-ImageNet. Overall, these results demonstrate that PushCen-ADFL substantially reduces the communication footprint of each decentralized push, which is particularly important in asynchronous settings where communication are frequent and uncoordinated.
8pt
| Method | CIFAR-10 | CIFAR-100 | Tiny-ImageNet |
|---|---|---|---|
| Async-DFedAvg | 4.15 | 10.70 | 42.44 |
| DivShare | 4.15 | 10.61 | 42.35 |
| SWIFT | 4.16 | 10.82 | 42.56 |
| PushCen-ADFL | 4.03 | 11.93 | 42.41 |
Table 3 reports the computational cost (GFLOPs) of different methods on three datasets. Overall, PushCen-ADFL maintains a lightweight computation footprint comparable to baselines. On CIFAR-10, PushCen-ADFL slightly reduces the cost to 4.03 GFLOPs, marginally lower than Async-DFedAvg, DivShare and SWIFT. On Tiny-ImageNet, it achieves 42.41 GFLOPs, essentially matching the baseline range. On CIFAR-100, PushCen-ADFL reaches 11.93 GFLOPs versus 10.61–10.82 GFLOPs for other methods, remaining within a manageable range.
We observe a slightly higher computation cost on CIFAR-100, mainly due to WCP-related operations (centroid clustering and re-encoding) and the centroid-based anchor construction during local updates. This effect is more visible with \(K=32\), which yields less aggressive sparsity and thus provides limited training-time computation reduction while still incurring clustering overhead.
4pt
| Method | CIFAR-10 | CIFAR-100 | Tiny-ImageNet |
|---|---|---|---|
| PushCen-ADFL (No Reg.) | 65.30% | 42.82% | 43.93% |
| PushCen-ADFL (No Buffer.) | 65.04% | 32.01% | 43.85% |
| PushCen-ADFL | 66.22% | 44.21% | 44.62% |
Table 4 evaluates the contribution of two key components in PushCen-ADFL: the regularization term and the buffer-based message handling mechanism. Removing the regularization (No Reg) leads to a consistent decrease in accuracy across all datasets, from 66.22% to 65.30% on CIFAR-10, from 44.21% to 42.82% on CIFAR-100 and from 44.62% to 43.93% on Tiny-ImageNet. This indicates that the regularization is beneficial for stabilizing training and improving generalization under heterogeneous and asynchronous decentralized updates. Disabling the buffer mechanism (No Buffer) also degrades performance, with the most pronounced effect observed on CIFAR-100 (44.21% to 32.01%), suggesting that buffer-based handling of incoming messages plays an important role in mitigating adverse effects caused by asynchronous arrivals (e.g., redundant or highly stale information) on more challenging tasks. Overall, both components contribute to the final performance and the full PushCen-ADFL achieves the best results across all datasets.
5pt
| Method | CIFAR-10 | CIFAR-100 | Tiny-ImageNet | |||
|---|---|---|---|---|---|---|
| 2-3(lr)4-5(lr)6-7 | Acc. \(\uparrow\) | SD \(\downarrow\) | Acc. \(\uparrow\) | SD \(\downarrow\) | Acc. \(\uparrow\) | SD \(\downarrow\) |
| Independent | 60.99 | 0.944 | 33.94 | 0.301 | 31.06 | 0.037 |
| Full-communication | ||||||
| Async-DFedAvg | 68.54 | 0.436 | 49.43 | 0.289 | 50.32 | 0.029 |
| Swift | 68.65 | 0.502 | 50.37 | 0.337 | 49.66 | 0.059 |
| Communication-efficient Methods | ||||||
| DivShare | 67.47 | 0.474 | 44.48 | 0.182 | 37.71 | 0.080 |
| PushCen-ADFL | 72.77 | 0.356 | 46.50 | 0.321 | 43.08 | 0.034 |
Table 5 reports delayed-client performance measured by the mean of each delayed client’s maximum test accuracy during training, together with the corresponding standard deviation (SD). On CIFAR-10, PushCen-ADFL reaches an accuracy of 72.77%, exceeding both the full-communication baselines and the communication-efficient. On CIFAR-100 and Tiny-ImageNet, PushCen-ADFL attains 46.50% and 43.08%, respectively, consistently improving over DivShare while remaining competitive with full-communication methods. These results indicate that PushCen-ADFL is particularly effective at incorporating delayed clients and enabling them to achieve strong personalized performance under their local distributions. Beyond the mean accuracy, SD shows that PushCen-ADFL also yields low variability across delayed clients, achieving the smallest SD on CIFAR-10 (0.356) and comparable to full-communication baselines on CIFAR-100 and Tiny-ImageNet. Detailed training curves for delayed clients are provided in Appendix 10.1.2.
Table 6 shows that increasing \(K\) improves accuracy at the cost of higher communication and computation. Even with \(K = 32\), PushCen-ADFL keeps per-push cost and GFLOPs strictly below all baselines, confirming that \(K\) provides a practical knob to trade accuracy for efficiency. We use \(K=32\) by default as it yields the best accuracy across all datasets.
2pt
| \(K\) | CIFAR-10 | CIFAR-100 | Tiny-ImageNet | |||
|---|---|---|---|---|---|---|
| 2-3(lr)4-5(lr)6-7 | Acc. (%) | Comm. (MB) | Acc. (%) | Comm. (MB) | Acc. (%) | Comm. (MB) |
| 8 | 52.45 | 0.976 | 25.51 | 48.70 | 15.83 | 48.89 |
| 16 | 62.17 | 1.298 | 36.67 | 62.51 | 35.96 | 62.77 |
| 32 | 66.22 | 1.622 | 44.21 | 76.33 | 44.62 | 76.65 |
Table 7 shows that enabling centroid regularization (\(\lambda>0\)) consistently outperforms \(\lambda=0\) across all datasets, confirming its role in stabilizing local updates under heterogeneity and asynchrony. On CIFAR-10, performance is stable for \(\lambda\in[0.01,0.10]\) (\({\approx}66\%\)). On CIFAR-100 and Tiny-ImageNet, smaller \(\lambda\) is slightly preferred, yet gains persist across a broad range. We adopt \(\lambda=0.1\) as a simple default.
6pt
| \(\lambda\) | CIFAR-10 | CIFAR-100 | Tiny-ImageNet |
|---|---|---|---|
| 0.10 | 66.22 | 44.21 | 44.62 |
| 0.05 | 66.19 | 43.72 | 44.92 |
| 0.01 | 66.40 | 43.35 | 44.59 |
| 0.005 | 65.51 | 43.16 | 44.55 |
| 0.001 | 65.68 | 44.57 | 45.40 |
| 0 (No Reg.) | 65.30 | 42.82 | 43.93 |
We provide additional learning-dynamics analyses to complement the main results. The global accuracy curves show that PushCen-ADFL follows stable convergence trajectories across all datasets, supporting the final averaged accuracies reported in Table 1. The delayed-client accuracy curves on a pseudo-time axis further show that late-joining clients improve rapidly after entering the system and reach performance comparable to online clients within a short pseudo-time window, corroborating the delayed-client results summarized in Table 5. Together, these results reinforce the convergence stability and robustness of PushCen-ADFL under asynchronous and heterogeneous decentralized settings.
This work investigated asynchronous decentralized federated learning, where frequent peer-to-peer, asymmetric and unbalanced communication and the statistical heterogeneity jointly induce high bandwidth cost, aggregation bias and model drift. We proposed PushCen-ADFL, an asynchronous decentralized framework that couples communication, de-biased aggregation, and local stabilization in a shared centroid representation space, forming a closed loop between compression and optimization. It integrates centroid-based model representation, average-preserving push-sum aggregation, centroid regularization, and bounded sender-deduplicated buffering to enable stable and robust training with delayed client participation. Extensive experiments demonstrate that PushCen-ADFL consistently improves performance over existing asynchronous decentralized baselines while substantially reducing per-push communication cost, achieving a favorable accuracy-communication trade-off under dynamic settings.
The appendix provides complementary analyses and additional experimental results that support the main paper:
Communication and computation properties of WCP. Appendix 8 revisits Weight Clustering Pruning (WCP) and derives its per-push communication cost as the sum of centroid transmission and index assignment, yielding an interpretable compression ratio mainly controlled by the number of centroids \(K\). It further compares the clustering overhead with standard training cost and shows that, under typical settings, WCP adds only a small fraction of extra computation.
Detailed convergence analysis of PushCen-ADFL. Appendix 9 presents a full convergence analysis under an event-driven asynchronous model on directed graphs. It formalizes push-sum aggregation with buffered (possibly stale) messages, centroid compression/reconstruction error, and the centroid proximal regularization used in local updates. The analysis establishes key lemmas on system-level mass conservation, numerator perturbation induced by lossy re-encoding, a consensus-error recursion under directed mixing with staleness and compression, and bounds showing how the centroid proximal term suppresses local deviation from its anchor. These components are combined into a constant-closed stationarity guarantee that makes the impacts of heterogeneity, staleness, directed mixing, and compression distortion explicit.
Supplementary experiments and ablations. Appendix 10 reports additional empirical evidence beyond the main tables, including global test-accuracy trajectories and delayed-client accuracy curves to visualize learning dynamics under asynchrony. It also provides ablations on the number of centroids \(K\) and the regularization weight \(\lambda\), quantifying the trade-offs between accuracy, per-push communication cost, and computational overhead, and validating the effectiveness and robustness of centroid regularization.
In this section, we provide a unified analysis of the communication and computational properties of Weight Clustering Pruning (WCP). Although WCP was originally introduced in our previous work, it remains a core component of the proposed method and plays an important role in reducing both communication cost and practical system overhead. For completeness, we revisit its theoretical properties and highlight why WCP remains efficient and well-suited for asynchronous federated learning.
We first analyze how WCP reduces communication overhead compared to standard federated learning. In conventional federated optimization, each client transmits a full model parameter vector \(\theta \in \mathbb{R}^N\) at every communication event. Assuming each parameter is represented using \(B\) bits, the communication cost per client per round is \[C_{\text{full}} = N \times B \quad \text{bits}.\]
Under WCP, model parameters are represented using a clustered form. Specifically, weights are quantized into \(k\) clusters and represented by a centroid set and an index assignment. One centroid is fixed at zero to naturally encode pruned weights and therefore does not need to be transmitted.
The total communication cost consists of two components. The first is the transmission of centroid values. Only \(k-1\) non-zero centroids need to be communicated, leading to \[C_{\text{centroid}} = (k-1) \times B \quad \text{bits}.\]
The second component corresponds to the index sequence that maps each weight to its associated centroid. Each index requires \(\lceil \log_2 k \rceil\) bits, resulting in \[C_{\text{index}} = N \times \lceil \log_2 k \rceil \quad \text{bits}.\]
Combining the two terms, the total communication cost of WCP is \[C_{\text{WCP}} = (k-1) \times B + N \times \lceil \log_2 k \rceil.\]
To quantify the communication reduction, we define the compression ratio \[\rho = \frac{C_{\text{WCP}}}{C_{\text{full}}} = \frac{(k-1)B + N \lceil \log_2 k \rceil}{N B}.\]
Since modern neural networks typically satisfy \(N \gg k\), the centroid transmission term becomes negligible. Thus, the compression ratio can be approximated as \[\rho \approx \frac{\lceil \log_2 k \rceil}{B}.\]
This result shows that communication efficiency is primarily controlled by the number of clusters \(k\). A moderate choice of \(k\) yields substantial bandwidth savings while maintaining sufficient representational capacity, which is essential for preserving model accuracy.
We now examine the computational cost introduced by WCP and compare it to the standard training cost in federated learning.
Consider a neural network with \(L\) layers, where the parameter matrix of layer \(l\) contains \(P^l = n_{l-1} n_l\) parameters. The total number of parameters is \[P_{\text{tot}} = \sum_{l=1}^{L} P^l.\]
For each training sample, the forward pass, backward pass and parameter update all incur costs proportional to the number of parameters. Thus, the per-sample training cost satisfies \[C_{\text{train}} = \mathcal{O}\!\left(\sum_{l=1}^{L} P^l \right).\] For a local dataset of size \(N\) and \(E\) training epochs, the total training cost becomes \[C_{\text{total\_train}} = \mathcal{O}\!\left(N E P_{\text{tot}}\right).\]
WCP applies clustering independently to each layer. For a layer with \(P^l\) parameters, a K-means iteration consists of an assignment step with complexity \(\mathcal{O}(P^l k)\) and a centroid update step with complexity \(\mathcal{O}(P^l)\). Assuming \(T\) clustering iterations, the cost for layer \(l\) is \[C_{\text{cluster}}^{l} = \mathcal{O}(T k P^l).\] Summing over all layers yields \[C_{\text{total\_cluster}} = \mathcal{O}(T k P_{\text{tot}}).\]
The ratio between training cost and clustering cost is therefore \[\frac{C_{\text{total\_train}}}{C_{\text{total\_cluster}}} = \frac{N E}{T k}.\]
Under typical experimental settings (e.g., \(E=1\), \(k=32\) and small \(T\)), this ratio is substantially larger than one. This indicates that the computational overhead introduced by WCP is dominated by the standard training process and remains a small fraction of the overall computation.
Overall, WCP introduces negligible additional computation while providing significant communication savings. Its computational cost scales linearly with the number of parameters and clusters, ensuring good scalability. These properties make WCP particularly suitable for asynchronous and decentralized federated learning, where both communication efficiency and lightweight local computation are critical.
We consider \(N\) clients indexed by \(\mathcal{V}=\{1,\dots,N\}\) connected by a directed graph \(\mathcal{G}=(\mathcal{V},\mathcal{E})\). For client \(i\), denote its in-neighbors and out-neighbors by \(\mathcal{N}_i^{-}\) and \(\mathcal{N}_i^{+}\) and let \(d_i^{+}\triangleq|\mathcal{N}_i^{+}|\). Clients exchange messages only along directed edges; there is no central server.
Training evolves on a global event timeline \(t=0,1,2,\dots\). At each event \(t\), either (i) a message arrives at some client and is appended into its local buffer, or (ii) a single client is activated and performs an aggregate \(\rightarrow\) local update \(\rightarrow\) broadcast cycle. Let \(i_t\in\mathcal{V}\) denote the activated client at event \(t\) (if any). No global synchronization is assumed and different clients may be activated at different rates.
Each client \(i\) maintains the following states at event \(t\):
Model \(w_i^t\in\mathbb{R}^d\), the local trainable parameters (vectorized for analysis).
Push-sum mass \(y_i^t>0\), corresponding to ps_mass in the implementation.
centroid dictionary \(G_i^t\), the layer-wise centroid tables maintained/updated by aggregation.
Buffer \(\mathcal{B}_i^t\), storing the most recent neighbor messages received since the last compute event.
Remark (local-only parameters). In the implementation, BatchNorm-related parameters/statistics are local-only (FedBN-style): they are not aggregated and remain purely local. Our analysis focuses on the shared trainable vector \(w_i^t\) (excluding such local-only components).
When client \(j\) transmits, it sends a centroid-compressed payload \(\mathcal{C}(w_j^t) \triangleq (V_j^t, A_j^t, U_j^t),\) where \(V_j^t\) is the centroid table, \(A_j^t\) is the assignment map and \(U_j^t\) contains uncompressed shared tensors that are communicated without clustering (e.g., biases or non-compressible layers). Local-only parameters (e.g., BatchNorm statistics) are not transmitted. Upon reception, client \(i\) reconstructs an approximate model \(\hat{w}_j^t = \mathcal{R}(\mathcal{C}(w_j^t)),\) where \(\mathcal{R}\) indexes \(V_j^t\) with \(A_j^t\) and copies \(U_j^t\). We define the reconstruction (compression) error: \[e_j^t \triangleq \hat{w}_j^t - w_j^t. \label{eq:compression-error-def}\tag{6}\]
A message sent from \(j\) to \(i\) is represented as \(m_{j\to i}^t = (\hat{w}_j^t, V_j^t, y_{j\to i}^t, j),\) where \(y_{j\to
i}^t>0\) is the push-sum mass share attached to the message (field ps_mass_share). At event \(t\), client \(i\)’s buffer is \(\mathcal{B}_i^t=\{m_{j\to i}^t: j\in\mathcal{S}_i^t\},\) where \(\mathcal{S}_i^t\subseteq\mathcal{N}_i^{-}\) indexes the set of senders currently stored. The implementation keeps only the newest
message per sender (sender deduplication) and may enforce a capacity bound; in analysis we treat \(\mathcal{B}_i^t\) as an arbitrary finite set of most-recent neighbor messages.
To make average-preservation explicit under directed, potentially unbalanced communication, we introduce the standard push-sum numerator variable \[x_i^t \triangleq y_i^t\, w_i^t \in \mathbb{R}^d. \label{eq:pushsum-x-def}\tag{7}\] The de-biased local estimate is always recovered as \(w_i^t = x_i^t / y_i^t\).
After finishing a local update, client \(i\) splits its current mass uniformly among its out-neighbors and itself: \[y_{i\to k}^t \;=\; \frac{y_i^t}{d_i^{+}+1},\quad \forall k\in\mathcal{N}_i^{+}, \qquad y_i^{t+} \;=\; \frac{y_i^t}{d_i^{+}+1}. \label{eq:mass-splitting}\tag{8}\] Each outgoing message to \(k\) carries \(y_{i\to k}^t\) and client \(i\) keeps the same share locally. Hence, the sender-side conservation holds exactly: \(d_i^{+}\cdot \frac{y_i^t}{d_i^{+}+1} + 1\cdot \frac{y_i^t}{d_i^{+}+1} = y_i^t.\) Analogously, the transmitted numerator share is \(x_{i\to k}^t \triangleq y_{i\to k}^t \hat{w}_i^t\) and the locally retained numerator is \(x_i^{t+}\triangleq y_i^{t+} w_i^{t+}\).
Under asynchronous message passing, a portion of mass may reside in buffers (or be in transit) between send and aggregation events. Let \(\mathcal{M}^t\) denote the set of all messages that are currently buffered or in transit in the system at event \(t\). For each message \(m\in\mathcal{M}^t\), let \(y_m\) be its mass share and \(x_m\) its numerator share. We define the total system mass and total system numerator: \[Y_{\mathrm{tot}}^t \triangleq \sum_{i=1}^{N} y_i^t + \sum_{m\in\mathcal{M}^t} y_m, \qquad X_{\mathrm{tot}}^t \triangleq \sum_{i=1}^{N} x_i^t + \sum_{m\in\mathcal{M}^t} x_m . \label{eq:total-mass-def}\tag{9}\] The mass-splitting rule 8 and the aggregation rule (which transfers buffered shares into the local states) together imply that \((Y_{\mathrm{tot}}^t, X_{\mathrm{tot}}^t)\) are invariant over time; this will be formalized in the mass-conservation lemma.
Using the conserved system totals, we define the canonical de-biased global reference model: \[\bar{w}^t \triangleq \frac{X_{\mathrm{tot}}^t}{Y_{\mathrm{tot}}^t}. \label{eq:global-reference}\tag{10}\] This sequence is well-defined on asymmetric and unbalanced graphs and serves as the main reference trajectory for optimality analysis (while \(w_i^t\) are local de-biased estimates with a consensus error).
Client \(i\) holds a private dataset \(\mathcal{D}_i\) and local objective \(f_i(w)\triangleq \mathbb{E}_{\xi\sim\mathcal{D}_i}[\ell(w;\xi)].\) Before each compute event, client \(i\) constructs a proximal target \(\tilde{w}_i^t\) from its current centroid dictionary estimate \(G_i^t\) and its current assignment map (obtained by clustering the current local model) and performs local stochastic optimization on \[f_i(w) + \lambda \|w-\tilde{w}_i^t\|_2^2, \label{eq:local-prox-objective}\tag{11}\] where \(\lambda\ge 0\) is the regularization weight. The construction of \(\tilde{w}_i^t\) is implementation-aligned: it is produced by indexing centroid tables using the assignment map so that \(\tilde{w}_i^t\) matches the shape of \(w_i^t\).
We list the standard assumptions required to establish the convergence of PushCen-ADFL. These assumptions cover stochastic optimization with heterogeneous data distributions, asynchronous communication with message delays and centroid-based lossy communication over directed graphs. All random variables are defined on a common probability space induced by data sampling, client activations and message arrivals.
Each local objective \(f_i:\mathbb{R}^d\rightarrow\mathbb{R}\) is \(L\)-smooth: \[\|\nabla f_i(u)-\nabla f_i(v)\|_2 \le L\|u-v\|_2,\quad \forall u,v\in\mathbb{R}^d, \label{eq:assump-smooth}\tag{12}\] and the global objective \(F(w)\triangleq \frac{1}{N}\sum_{i=1}^N f_i(w)\) is lower bounded, i.e., \(F(w)\ge F^\star\) for some finite \(F^\star\).
At client \(i\), a stochastic gradient computed from a mini-batch \(\xi\sim\mathcal{D}_i\) satisfies \[\mathbb{E}\!\left[g_i(w;\xi)\mid w\right] = \nabla f_i(w), \qquad \mathbb{E}\!\left[\|g_i(w;\xi)-\nabla f_i(w)\|_2^2 \mid w\right] \le \sigma^2, \label{eq:assump-sgd}\tag{13}\] for some \(\sigma^2<\infty\). When centroid proximal regularization is applied, we equivalently view the client as optimizing the regularized objective \(f_i^\lambda(w)\triangleq f_i(w)+\lambda\|w-\tilde{w}_i\|_2^2\), and the stochastic gradient is taken w.r.t.\(f_i^\lambda\).
There exists a constant \(\zeta^2<\infty\) such that for all \(w\in\mathbb{R}^d\) and all clients \(i\in[N]\), \[\|\nabla f_i(w)-\nabla F(w)\|_2^2 \le \zeta^2, \qquad\text{where}\qquad F(w)\triangleq \frac{1}{N}\sum_{j=1}^{N} f_j(w). \label{eq:A3-uniform}\tag{14}\]
Remark. Assumption A3 is a standard strengthening of the mean-heterogeneity condition and is commonly used in non-convex FL analyses to control one-step descent for the activated client under asynchrony.
Messages aggregated at event \(t\) may correspond to stale model states due to heterogeneous computation/communication. We assume there exists an integer \(\tau\ge 0\) such that any message used by an activated client at event \(t\) was generated within the last \(\tau\) activations/events. Equivalently, if a buffered message \(m_{j\to i}^t\) carries a reconstruction \(\hat{w}_j^{\kappa}\), then its generation time \(\kappa\) satisfies \[t-\kappa \le \tau. \label{eq:assump-stale}\tag{15}\]
The directed communication graph is strongly connected and the sequence of effective communication events is sufficiently mixing. Concretely, there exists an integer \(B\ge 1\) such that for any window of \(B\) consecutive events, the union of directed edges along which messages are delivered is strongly connected. Moreover, the push-sum masses are uniformly bounded away from zero: there exists \(y_{\min}>0\) such that \[y_i^t \ge y_{\min},\quad \forall i\in\mathcal{V},\;\forall t. \label{eq:assump-mass-lb}\tag{16}\] This ensures the de-biased ratios \(w_i^t = x_i^t/y_i^t\) and the global reference \(\bar w^t\) are well-defined.
The centroid compression-reconstruction operator \(\mathcal{R}\circ\mathcal{C}\) induces a bounded absolute error: there exists \(\varepsilon_c\ge 0\) such that for any communicated model \(w\), \[\|\mathcal{R}(\mathcal{C}(w)) - w\|_2 \le \varepsilon_c. \label{eq:assump-compress-C1}\tag{17}\] Equivalently, with \(e_j^t\) defined in 6 , we have \(\|e_j^t\|_2 \le \varepsilon_c\) for all transmitted messages. This assumption captures the deterministic quantization/clustering distortion introduced by WCP with a fixed number of centroids \(K\).
Remark. Assumptions (A1)-(A3) are standard in stochastic non-convex optimization and federated learning. Assumptions (A4)-(A5) formalize bounded staleness and directed-graph mixing needed for asynchronous push-sum. Assumption (A6) matches our implementation, where WCP-based centroid encoding introduces deterministic reconstruction distortion controlled by the clustering resolution (e.g., \(K=32\)).
We formalize the event-driven update rules of PushCen-ADFL (ADFLCenReg) by separating (i) asynchronous push-sum aggregation over buffered neighbor messages and (ii) local stochastic optimization with centroid proximal regularization. Throughout, we use the push-sum numerator-denominator representation \(x_i^t = y_i^t w_i^t\) introduced in 7 .
When client \(j\) is activated and broadcasts to an out-neighbor \(i\in\mathcal{N}_j^{+}\), it transmits a tuple \[m_{j\to i}^{(\kappa)} \;=\; \big(\mathcal{C}(w_j^{\kappa}),\; y_{j\to i}^{(\kappa)},\; j \big), \label{eq:message-format}\tag{18}\] where \(\kappa\) denotes the generation event index at which the message is produced, \(\mathcal{C}(w_j^{\kappa})=(V_j^{\kappa},A_j^{\kappa},U_j^{\kappa})\) is the centroid-compressed payload and \(y_{j\to i}^{(\kappa)}>0\) is the attached mass share. Upon reception at some later event \(t\ge \kappa\), client \(i\) reconstructs the (possibly stale) neighbor model \[\hat{w}_{j\to i}^t \;\triangleq\; \mathcal{R}\!\left(\mathcal{C}(w_j^{\kappa})\right), \label{eq:stale-recon}\tag{19}\] and buffers the tuple \((\hat{w}_{j\to i}^t, V_j^{\kappa}, y_{j\to i}^{(\kappa)}, j)\). 3 At the time client \(i\) is activated at event \(t\), let \(\mathcal{B}_i^t\) denote its buffer, i.e., the set of currently stored messages to be aggregated (sender-deduplicated and possibly capacity-bounded).
When client \(i=i_t\) is activated at event \(t\), it aggregates all buffered neighbor messages \(\mathcal{B}_i^t = \{(\hat{w}_{j\to i}^t, V_j^{\kappa}, y_{j\to i}^{(\kappa)}, j)\}\). Define the post-buffer total mass at \(i\): \[Y_i^t \;\triangleq\; y_i^t + \sum_{(\cdot,\cdot,y_{j\to i}^{(\kappa)},\cdot)\in\mathcal{B}_i^t} y_{j\to i}^{(\kappa)}. \label{eq:agg-total-mass}\tag{20}\] The de-biased model update is \[w_i^{t+\frac{1}{3}} \;=\; \frac{y_i^t}{Y_i^t}\, w_i^t \;+\; \sum_{(\hat{w}_{j\to i}^t,\cdot,y_{j\to i}^{(\kappa)},\cdot)\in\mathcal{B}_i^t} \frac{y_{j\to i}^{(\kappa)}}{Y_i^t}\, \hat{w}_{j\to i}^t, \label{eq:agg-model}\tag{21}\] with the mass update and buffer clearing \[y_i^{t+\frac{1}{3}} \;=\; Y_i^t, \qquad \mathcal{B}_i^{t+\frac{1}{3}} \;=\; \emptyset. \label{eq:agg-mass}\tag{22}\] Equivalently, in numerator-denominator form, letting \(x_i^t=y_i^t w_i^t\) and \(x_{j\to i}^{(\kappa)}\triangleq y_{j\to i}^{(\kappa)} \hat{w}_{j\to i}^t\), the aggregation is additive: \[x_i^{t+\frac{1}{3}} \;=\; x_i^t + \sum_{(\cdot,\cdot,y_{j\to i}^{(\kappa)},\cdot)\in\mathcal{B}_i^t} x_{j\to i}^{(\kappa)}, \qquad y_i^{t+\frac{1}{3}} \;=\; y_i^t + \sum_{(\cdot,\cdot,y_{j\to i}^{(\kappa)},\cdot)\in\mathcal{B}_i^t} y_{j\to i}^{(\kappa)}. \label{eq:agg-xy}\tag{23}\] Finally \(w_i^{t+\frac{1}{3}} = x_i^{t+\frac{1}{3}} / y_i^{t+\frac{1}{3}}\).
Client \(i\) updates its centroid dictionary estimate using the same push-sum weights: \[G_i^{t+\frac{1}{3}} \;=\; \frac{y_i^t}{Y_i^t}\, G_i^t \;+\; \sum_{(\cdot,V_j^{\kappa},y_{j\to i}^{(\kappa)},\cdot)\in\mathcal{B}_i^t} \frac{y_{j\to i}^{(\kappa)}}{Y_i^t}\, V_j^{\kappa}. \label{eq:agg-pool}\tag{24}\] (We treat \(G_i^t\) as an auxiliary state whose role is to form the proximal target in the local update.)
After aggregation, client \(i\) performs \(E\) local epochs/steps of SGD on \[\Phi_i^t(w) \;\triangleq\; f_i(w) + \lambda \|w - \tilde{w}_i^t\|_2^2, \label{eq:local-phi}\tag{25}\] where \(\tilde{w}_i^t\) is the centroid-based proximal target constructed from the current centroid dictionary \(G_i^{t+\frac{1}{3}}\) and the current assignment map (obtained by applying WCP to \(w_i^{t+\frac{1}{3}}\)). Let \((V_i^t,A_i^t,M_i^t)\) be the centroid table, assignment map and pruning mask produced by WCP at the beginning of the local update. For \(e=0,\dots,E-1\), we write the masked proximal-SGD step as
\[\begin{align} w_{i,e+1} &= \Pi_{M_i^t}\!\Bigl( w_{i,e} - \eta\, g_i\!\left(w_{i,e};\xi_{i,e}\right) \\ &\qquad\quad - 2\eta\lambda\,(w_{i,e}-\tilde{w}_i^t) \Bigr), \qquad w_{i,0} = w_i^{t+\frac{1}{3}} . \end{align} \label{eq:local-sgd}\tag{26}\]
where \(\xi_{i,e}\) is a mini-batch sampled from \(\mathcal{D}_i\) and \(g_i(\cdot;\xi)\) is the stochastic gradient of \(\ell(\cdot;\xi)\). The masking operator \(\Pi_{M_i^t}(\cdot)\) enforces pruning by element-wise multiplication, \[\Pi_{M_i^t}(u) \triangleq u \odot M_i^t,
\label{eq:mask-proj}\tag{27}\] matching the implementation (w\(\leftarrow\)w * M). We denote the post-local-update model by \[w_i^{t+\frac{2}{3}} \;\triangleq\; w_{i,E}.
\label{eq:local-output}\tag{28}\]
After local optimization, client \(i\) re-encodes its model via WCP to obtain an updated centroid representation \(\mathcal{C}(w_i^{t+\frac{2}{3}})=(V_i^{t+\frac{2}{3}},A_i^{t+\frac{2}{3}},U_i^{t+\frac{2}{3}})\) to be transmitted. It then performs push-sum mass splitting: \[y_{i\to k}^{t+\frac{2}{3}} \;=\; \frac{y_i^{t+\frac{1}{3}}}{d_i^{+}+1},\quad \forall k\in\mathcal{N}_i^{+}, \qquad y_i^{t+1} \;=\; \frac{y_i^{t+\frac{1}{3}}}{d_i^{+}+1}, \label{eq:broadcast-mass}\tag{29}\] and sends \((\mathcal{C}(w_i^{t+\frac{2}{3}}), y_{i\to k}^{t+\frac{2}{3}}, i)\) to each \(k\in\mathcal{N}_i^{+}\). Finally, \[w_i^{t+1} \;\triangleq\; w_i^{t+\frac{2}{3}}, \qquad x_i^{t+1} \;\triangleq\; y_i^{t+1} w_i^{t+1}. \label{eq:end-of-event}\tag{30}\] For any client \(\ell\neq i_t\) that is not activated at event \(t\), we keep its local state unchanged: \((w_\ell^{t+1},x_\ell^{t+1},y_\ell^{t+1},G_\ell^{t+1})=(w_\ell^{t},x_\ell^{t},y_\ell^{t},G_\ell^{t})\).
To analyze PushCen-ADFL under directed asynchronous communication, we introduce a de-biased global reference sequence together with two key error measures: an optimality metric that tracks progress toward stationary points and a consensus metric that tracks network disagreement.
Recall the total system numerator and mass defined in 9 : \(X_{\mathrm{tot}}^t\) and \(Y_{\mathrm{tot}}^t\). We define the (virtual) de-biased global reference model as \[\bar{w}^t \triangleq \frac{X_{\mathrm{tot}}^t}{Y_{\mathrm{tot}}^t}. \label{eq:key-wbar}\tag{31}\] Unlike the naive arithmetic mean \(\frac{1}{N}\sum_{i=1}^N w_i^t\), \(\bar{w}^t\) is the correct average-preserving reference on directed and potentially unbalanced graphs under push-sum dynamics.
Each client maintains a local push-sum numerator \(x_i^t\) and mass \(y_i^t\). The local de-biased model held at node \(i\) is \[w_i^t \triangleq \frac{x_i^t}{y_i^t}. \label{eq:key-local-debias}\tag{32}\] Important: We strictly distinguish the local model \(w_i^t\) from the centroid proximal anchor \(\tilde{w}_i^t\) defined in 25 , where \(\tilde{w}_i^t\) serves as the regularization target during local updates.
We quantify network disagreement by the mean-squared deviation of local models from the de-biased global reference: \[\mathcal{E}_{\mathrm{con}}^t \triangleq \frac{1}{N}\sum_{i=1}^{N}\left\|w_i^t - \bar{w}^t\right\|_2^2. \label{eq:key-consensus}\tag{33}\] This term captures the combined effects of directed mixing, asynchronous staleness and lossy reconstruction on consensus.
For the global objective \(F(w)\triangleq \frac{1}{N}\sum_{i=1}^{N} f_i(w)\), we use the standard non-convex stationarity metric evaluated at the de-biased global reference sequence: \[\mathcal{E}_{\mathrm{opt}}^t \triangleq \mathbb{E}\left[\left\|\nabla F(\bar{w}^t)\right\|_2^2\right]. \label{eq:key-optimality}\tag{34}\] Bounding \(\frac{1}{T}\sum_{t=0}^{T-1}\mathcal{E}_{\mathrm{opt}}^t\) implies convergence to a stationary neighborhood.
Remark. Our analysis proceeds by (i) deriving a recursion for \(\mathcal{E}_{\mathrm{con}}^t\) under push-sum mixing with perturbations (compression and staleness) and (ii) proving a descent inequality for \(F(\bar{w}^t)\) whose error terms are controlled by \(\mathcal{E}_{\mathrm{con}}^t\). Combining the two yields a bound on the averaged stationarity measure in 34 .
Lemma 1 (System-level mass conservation). Recall the set of in-flight/buffered messages \(\mathcal{M}^t\) and the total system mass \[Y_{\mathrm{tot}}^t \triangleq \sum_{i=1}^{N} y_i^t + \sum_{m\in\mathcal{M}^t} y_m, \label{eq:total-mass-recall}\qquad{(3)}\] where \(y_i^t\) is the local push-sum mass at node \(i\) and \(y_m\) is the mass share carried by message \(m\). Under the update rules in Section 9.3, the total system mass is invariant: \[Y_{\mathrm{tot}}^{t+1} = Y_{\mathrm{tot}}^{t},\qquad \forall t\ge 0. \label{eq:mass-conservation-only}\qquad{(4)}\] Consequently, the de-biased global reference \(\bar w^t = X_{\mathrm{tot}}^t / Y_{\mathrm{tot}}^t\) is well-defined for all \(t\) since \(Y_{\mathrm{tot}}^t\) is constant and strictly positive (Assumption 16 ).
Proof. We show that \(Y_{\mathrm{tot}}\) does not change under either type of event: (i) message generation/sending and (ii) message aggregation/consumption. All other operations (e.g., local SGD on \(w_i\)) do not modify \(y_i\) and thus do not affect \(Y_{\mathrm{tot}}\).
Suppose client \(i\) is activated and performs mass splitting as in 29 . Let \(d_i^{+}=|\mathcal{N}_i^{+}|\) and denote the pre-splitting local mass by \(y_i\). The splitting rule sets \[y_i' \;=\; \frac{y_i}{d_i^{+}+1}, \qquad y_{i\to k} \;=\; \frac{y_i}{d_i^{+}+1},\;\forall k\in\mathcal{N}_i^{+},\] and creates \(d_i^{+}\) new outgoing messages, each carrying mass \(y_{i\to k}\). Let \(\mathcal{M}\) be the message set immediately before splitting and \(\mathcal{M}'\) the set immediately after splitting. Then \(\mathcal{M}' = \mathcal{M}\cup \{m_{i\to k}:k\in\mathcal{N}_i^{+}\}\) and we have \[\begin{align} \Big(y_i' + \sum_{m\in\mathcal{M}'} y_m\Big) - \Big(y_i + \sum_{m\in\mathcal{M}} y_m\Big) &= (y_i' - y_i) + \sum_{k\in\mathcal{N}_i^{+}} y_{i\to k} \nonumber\\ &= \left(\frac{y_i}{d_i^{+}+1}-y_i\right) + d_i^{+}\cdot \frac{y_i}{d_i^{+}+1} \nonumber\\ &= 0. \label{eq:mass-split-diff} \end{align}\tag{35}\] All other nodes \(\ell\neq i\) keep \(y_\ell\) unchanged during this event, hence \(Y_{\mathrm{tot}}\) is invariant.
When an activated client \(i\) aggregates its buffer (Eq. 22 ), it consumes a subset of messages \(\mathcal{B}_i \subseteq \mathcal{M}\) and transfers their mass shares into the local state: \[y_i' \;=\; y_i + \sum_{m\in\mathcal{B}_i} y_m, \qquad \mathcal{M}' \;=\; \mathcal{M}\setminus \mathcal{B}_i.\] Therefore, \[\begin{align} \Big(y_i' + \sum_{m\in\mathcal{M}'} y_m\Big) - \Big(y_i + \sum_{m\in\mathcal{M}} y_m\Big) &= \sum_{m\in\mathcal{B}_i} y_m \;-\; \sum_{m\in\mathcal{B}_i} y_m = 0. \label{eq:mass-agg-diff} \end{align}\tag{36}\] Again, all other nodes keep their masses unchanged, so \(Y_{\mathrm{tot}}\) remains invariant.
Since \(Y_{\mathrm{tot}}\) is unchanged by both message splitting and message aggregation, we obtain \(Y_{\mathrm{tot}}^{t+1}=Y_{\mathrm{tot}}^{t}\) for all events \(t\). ◻
Remark. In contrast to the mass variable, the system numerator \(X_{\mathrm{tot}}^t\) may be perturbed by lossy compression: in the implementation, a sender retains the full-precision local model \(w_i\) while transmitting a compressed representation that reconstructs to \(\hat{w}_i\neq w_i\) at receivers. This perturbation will be explicitly handled in subsequent lemmas when deriving consensus and optimality recursions under Assumption 17 .
Lemma 2 (Numerator perturbation induced by lossy compression). Recall the total system mass \(Y_{\mathrm{tot}}^t\) in ?? , which is invariant by Lemma 1. Let \[X_{\mathrm{tot}}^t \triangleq \sum_{i=1}^{N} x_i^t + \sum_{m\in\mathcal{M}^t} x_m \label{eq:total-numerator-recall}\qquad{(5)}\] be the total system numerator, where \(x_i^t \triangleq y_i^t w_i^t\) and each message \(m\) carries numerator share \(x_m \triangleq y_m \hat{w}_m\) with reconstructed content \(\hat{w}_m\).
Consider an activation event \(t\) at client \(i=i_t\). Let \(w_i^{t+\frac{2}{3}}\) be the post-local-update model before broadcasting (Section 9.3, C2) and define the message model* produced by centroid re-encoding as \[\check w_i^{t} \triangleq \mathcal{R}\!\left(\mathcal{C}(w_i^{t+\frac{2}{3}})\right), \qquad e_i^{t} \triangleq \check w_i^{t} - w_i^{t+\frac{2}{3}}. \label{eq:sender-quant-error}\tag{37}\] Then the only communication operation that changes \(X_{\mathrm{tot}}\) is the mass splitting / broadcast step, and the induced numerator change satisfies the exact identity \[X_{\mathrm{tot}}^{t+1} - X_{\mathrm{tot}}^{t+\frac{2}{3}} \;=\; \sum_{k\in\mathcal{N}_i^{+}} y_{i\to k}^{t+\frac{2}{3}}\, e_i^{t}. \label{eq:Xtot-delta-exact}\tag{38}\] Under Assumption (A6) (bounded absolute compression error), i.e., \(\|e_i^{t}\|_2\le \varepsilon_c\), we further have \[\big\|X_{\mathrm{tot}}^{t+1} - X_{\mathrm{tot}}^{t+\frac{2}{3}}\big\|_2 \;\le\; \Big(\sum_{k\in\mathcal{N}_i^{+}} y_{i\to k}^{t+\frac{2}{3}}\Big)\,\varepsilon_c \;=\; \frac{d_i^{+}}{d_i^{+}+1}\,y_i^{t+\frac{1}{3}}\,\varepsilon_c, \label{eq:Xtot-delta-bound}\tag{39}\] where the last equality uses the mass splitting rule 29 . Consequently, since \(Y_{\mathrm{tot}}\) is invariant, the de-biased global reference \(\bar w^t = X_{\mathrm{tot}}^t/Y_{\mathrm{tot}}^t\) satisfies \[\big\|\bar w^{t+1} - \bar w^{t+\frac{2}{3}}\big\|_2 \;=\; \frac{\big\|X_{\mathrm{tot}}^{t+1} - X_{\mathrm{tot}}^{t+\frac{2}{3}}\big\|_2}{Y_{\mathrm{tot}}} \;\le\; \frac{d_i^{+}}{d_i^{+}+1}\cdot \frac{y_i^{t+\frac{1}{3}}}{Y_{\mathrm{tot}}}\,\varepsilon_c. \label{eq:wbar-delta-bound}\tag{40}\] *
Proof. We focus on the broadcast (mass splitting) step, since: (i) message arrivals only move a message into some buffer and do not change \(\mathcal{M}^t\) as a multiset of in-flight shares and (ii) buffer aggregation transfers message shares from \(\mathcal{M}^t\) into a local state without changing the total numerator (it is a pure re-allocation of existing shares). Therefore, the only communication-induced change to \(X_{\mathrm{tot}}\) arises from creating new outgoing messages while scaling down the sender’s local mass.
Fix event \(t\) where client \(i\) broadcasts after local training. Let the local mass right after aggregation be \(y_i^{t+\frac{1}{3}}\) (Eq. 22 ), which is the mass to be split. Before splitting, the sender contributes local numerator \(x_i^{\mathrm{pre}} = y_i^{t+\frac{1}{3}}\, w_i^{t+\frac{2}{3}}\) to the system total and the message set does not contain the newly created outgoing shares. After splitting (Eq. 29 ), the sender retains mass \(y_i^{t+1} = \frac{y_i^{t+\frac{1}{3}}}{d_i^{+}+1}\) while creating \(d_i^{+}\) outgoing messages each with mass \(y_{i\to k}^{t+\frac{2}{3}}=\frac{y_i^{t+\frac{1}{3}}}{d_i^{+}+1}.\) Crucially, the implementation sends a centroid-compressed payload, whose receiver-side reconstruction equals the message model \(\check w_i^{t}\) in 37 . Hence, immediately after splitting:
the sender’s retained local numerator becomes \(x_i^{\mathrm{post}} = y_i^{t+1}\, w_i^{t+\frac{2}{3}},\) because the local model is not replaced by its compressed reconstruction;
the newly created message set contributes additional numerator \(\sum_{k\in\mathcal{N}_i^{+}} y_{i\to k}^{t+\frac{2}{3}}\, \check w_i^{t}.\)
Therefore, the net change in the system total numerator caused by splitting is \[\begin{align} X_{\mathrm{tot}}^{t+1} - X_{\mathrm{tot}}^{t+\frac{2}{3}} &= \Big(x_i^{\mathrm{post}} + \sum_{k\in\mathcal{N}_i^{+}} y_{i\to k}^{t+\frac{2}{3}}\,\check w_i^{t}\Big) - x_i^{\mathrm{pre}} \nonumber\\ &= \Big(\tfrac{y_i^{t+\frac{1}{3}}}{d_i^{+}+1} w_i^{t+\frac{2}{3}} + d_i^{+}\cdot \tfrac{y_i^{t+\frac{1}{3}}}{d_i^{+}+1}\check w_i^{t}\Big) - y_i^{t+\frac{1}{3}} w_i^{t+\frac{2}{3}} \nonumber\\ &= \frac{d_i^{+}}{d_i^{+}+1}\, y_i^{t+\frac{1}{3}}\big(\check w_i^{t}-w_i^{t+\frac{2}{3}}\big) \nonumber\\ &= \sum_{k\in\mathcal{N}_i^{+}} y_{i\to k}^{t+\frac{2}{3}}\, e_i^{t}, \label{eq:deltaX-derivation} \end{align}\tag{41}\] which is exactly 38 .
Taking norms and using \(\|e_i^{t}\|_2\le \varepsilon_c\) yields \[\big\|X_{\mathrm{tot}}^{t+1} - X_{\mathrm{tot}}^{t+\frac{2}{3}}\big\|_2 \le \sum_{k\in\mathcal{N}_i^{+}} y_{i\to k}^{t+\frac{2}{3}}\, \|e_i^{t}\|_2 \le \Big(\sum_{k\in\mathcal{N}_i^{+}} y_{i\to k}^{t+\frac{2}{3}}\Big)\varepsilon_c,\] and substituting \(\sum_{k} y_{i\to k}^{t+\frac{2}{3}} = \frac{d_i^{+}}{d_i^{+}+1}y_i^{t+\frac{1}{3}}\) gives 39 . Finally, Lemma 1 implies \(Y_{\mathrm{tot}}^{t+1}=Y_{\mathrm{tot}}^{t}=Y_{\mathrm{tot}}\), so dividing by the constant \(Y_{\mathrm{tot}}\) yields 40 . ◻
Lemma 3 (Consensus error recursion under asynchronous push-sum with perturbations). Recall the consensus error \(\mathcal{E}_{\mathrm{con}}^t = \frac{1}{N}\sum_{i=1}^N \|w_i^t-\bar w^t\|_2^2\) in 33 , where \(w_i^t=x_i^t/y_i^t\) and \(\bar w^t=X_{\mathrm{tot}}^t/Y_{\mathrm{tot}}^t\). Consider an activation event \(t\) at client \(i=i_t\).
Define the activated-client drift \[\Delta_i^t \triangleq w_i^{t+\frac{2}{3}} - w_i^{t+\frac{1}{3}}, \label{eq:drift-def-lem3}\qquad{(6)}\] and for any buffered message from sender \(j\) generated at event \(\kappa\) and available at client \(i\) at event \(t\), define the receiver-side reconstruction error \[e_{j\to i}^{(\kappa)} \triangleq \hat{w}_{j\to i}^{t} - w_j^{\kappa}, \qquad \|e_{j\to i}^{(\kappa)}\|_2 \le \varepsilon_c \;\;\text{by Assumption (A6)}. \label{eq:recv-recon-error}\qquad{(7)}\]
Under Assumptions (A4)-(A6) and the directed mixing Assumption (A5), there exist constants \(\rho\in(0,1)\) and \(C_1,C_2,C_3>0\) such that \[\begin{align} \mathbb{E}\!\left[\mathcal{E}_{\mathrm{con}}^{t+1}\right] \;\le\; &\rho\,\mathbb{E}\!\left[\mathcal{E}_{\mathrm{con}}^{t}\right] + C_1\,\mathbb{E}\!\left[\|\Delta_{i_t}^t\|_2^2\right] + C_2\,\tau \sum_{r=t-\tau}^{t-1} \mathbb{E}\!\left[\|\Delta_{i_r}^r\|_2^2\right] + C_3\,\varepsilon_c^2 . \end{align} \label{eq:consensus-recursion}\qquad{(8)}\]
Proof. We bound the deviation \(w-\bar w\) across one event by separating (i) buffer aggregation (mixing) with stale/quantized inputs, (ii) client drift \(\Delta_i^t\) and (iii) the reference drift of \(\bar w\) due to lossy broadcast (Lemma 2).
At activation event \(t\), client \(i=i_t\) aggregates its buffer via 21 . Let \(Y_i^t\) be defined in 20 and define normalized weights \[\alpha_{i}^t \triangleq \frac{y_i^t}{Y_i^t}, \qquad \alpha_{j\to i}^t \triangleq \frac{y_{j\to i}^{(\kappa)}}{Y_i^t} \;\;\text{for each }(\hat{w}_{j\to i}^{t},\cdot,y_{j\to i}^{(\kappa)},j)\in\mathcal{B}_i^t, \label{eq:alpha-weights}\tag{42}\] so that \(\alpha_i^t+\sum_{(j\to i)\in\mathcal{B}_i^t}\alpha_{j\to i}^t=1\). Using staleness-aware notation, each buffered model satisfies \[\hat{w}_{j\to i}^{t} = w_j^{\kappa} + e_{j\to i}^{(\kappa)}, \qquad t-\kappa \le \tau \;\text{(Assumption (A4))}. \label{eq:stale-decomp}\tag{43}\] Thus \[w_i^{t+\frac{1}{3}} = \alpha_i^t w_i^t + \sum_{(j\to i)\in\mathcal{B}_i^t}\alpha_{j\to i}^t\, w_j^{\kappa} + \sum_{(j\to i)\in\mathcal{B}_i^t}\alpha_{j\to i}^t\, e_{j\to i}^{(\kappa)}. \label{eq:agg-decomp}\tag{44}\] Subtract \(\bar w^t\) and apply \(\|a+b\|^2\le 2\|a\|^2+2\|b\|^2\) and Jensen’s inequality: \[\begin{align} \|w_i^{t+\frac{1}{3}}-\bar w^t\|^2 \;\le\;& 2\Big\| \alpha_i^t (w_i^t-\bar w^t) + \sum_{(j\to i)\in\mathcal{B}_i^t}\alpha_{j\to i}^t (w_j^{\kappa}-\bar w^t) \Big\|^2 + 2\Big\| \sum_{(j\to i)\in\mathcal{B}_i^t}\alpha_{j\to i}^t e_{j\to i}^{(\kappa)} \Big\|^2 \nonumber\\ \;\le\;& 2\alpha_i^t \|w_i^t-\bar w^t\|^2 + 2\sum_{(j\to i)\in\mathcal{B}_i^t}\alpha_{j\to i}^t \|w_j^{\kappa}-\bar w^t\|^2 + 2\sum_{(j\to i)\in\mathcal{B}_i^t}\alpha_{j\to i}^t \|e_{j\to i}^{(\kappa)}\|^2 . \label{eq:agg-dev-bound-tight} \end{align}\tag{45}\]
By Assumption (A6), \(\|e_{j\to i}^{(\kappa)}\|\le \varepsilon_c\), hence \[2\sum_{(j\to i)\in\mathcal{B}_i^t}\alpha_{j\to i}^t \|e_{j\to i}^{(\kappa)}\|^2 \le 2\varepsilon_c^2. \label{eq:recv-error-term}\tag{46}\]
Between events \(\kappa\) and \(t\), at most \(\tau\) activations occur. Only the activated client changes its model at each event. Therefore, for any node \(j\) and any \(\kappa\) with \(t-\kappa\le\tau\), \[\|w_j^t - w_j^{\kappa}\| \le \sum_{r=\kappa}^{t-1}\|\Delta_{i_r}^r\|. \label{eq:stale-sum}\tag{47}\] Using \((\sum_{r=\kappa}^{t-1} a_r)^2 \le (t-\kappa)\sum_{r=\kappa}^{t-1}a_r^2 \le \tau \sum_{r=t-\tau}^{t-1} a_r^2\), we obtain \[\|w_j^t - w_j^{\kappa}\|^2 \le \tau \sum_{r=t-\tau}^{t-1}\|\Delta_{i_r}^r\|^2. \label{eq:stale-sum-sq}\tag{48}\] Consequently, \[\begin{align} \|w_j^{\kappa}-\bar w^t\|^2 &\le 2\|w_j^{t}-\bar w^t\|^2 + 2\|w_j^{t}-w_j^{\kappa}\|^2 \nonumber\\ &\le 2\|w_j^{t}-\bar w^t\|^2 + 2\tau \sum_{r=t-\tau}^{t-1}\|\Delta_{i_r}^r\|^2 . \label{eq:stale-to-current} \end{align}\tag{49}\] Substituting 49 and 46 into 45 yields \[\begin{align} \|w_i^{t+\frac{1}{3}}-\bar w^t\|^2 &\le 2\alpha_i^t \|w_i^t-\bar w^t\|^2 + 4\sum_{(j\to i)\in\mathcal{B}_i^t}\alpha_{j\to i}^t \|w_j^{t}-\bar w^t\|^2 + 4\tau \sum_{r=t-\tau}^{t-1}\|\Delta_{i_r}^r\|^2 + 2\varepsilon_c^2. \label{eq:agg-final-tight} \end{align}\tag{50}\] Remark (optional simplification). For cleaner constants, one may upper bound \(2\alpha_i^t\|w_i^t-\bar w^t\|^2 \le 4\alpha_i^t\|w_i^t-\bar w^t\|^2\) and combine the first two terms into a single factor \(4\); we keep 50 to avoid unnecessary slack.
After local SGD, \(w_i^{t+\frac{2}{3}} = w_i^{t+\frac{1}{3}} + \Delta_i^t\), hence \[\|w_i^{t+\frac{2}{3}}-\bar w^t\|^2 \le 2\|w_i^{t+\frac{1}{3}}-\bar w^t\|^2 + 2\|\Delta_i^t\|^2. \label{eq:add-drift}\tag{51}\] For all \(\ell\neq i\), \(w_\ell^{t+\frac{2}{3}}=w_\ell^{t}\).
At the end of event \(t\), the consensus error is measured w.r.t.\(\bar w^{t+1}\). For any node \(\ell\) (activated or not), \[\|w_\ell^{t+1}-\bar w^{t+1}\|^2 \le 2\|w_\ell^{t+1}-\bar w^{t}\|^2 + 2\|\bar w^{t+1}-\bar w^{t}\|^2. \label{eq:ref-drift-split}\tag{52}\] Lemma 2 implies \(\|\bar w^{t+1}-\bar w^{t}\|\le c_{\mathrm{ps}}\varepsilon_c\) for a constant \(c_{\mathrm{ps}}>0\) depending on \((d_i^+,y_i/Y_{\mathrm{tot}})\), thus the second term in 52 contributes an additive \(O(\varepsilon_c^2)\) term to the network average, absorbed into \(C_3\varepsilon_c^2\).
We formalize the mixing part of the event dynamics by a (random) column-stochastic matrix sequence. Let \(P^t\in\mathbb{R}^{N\times N}\) denote the effective push-sum mixing matrix at event \(t\) such that, in the no-perturbation case (i.e., ignoring local training drift, staleness and compression noise), the push-sum numerator/mass evolve as \[x^{t+1}=P^t x^t, \qquad y^{t+1}=P^t y^t, \label{eq:ps-linear}\tag{53}\] where \(x^t=[x_1^t;\dots;x_N^t]\) and \(y^t=[y_1^t;\dots;y_N^t]\). By construction of mass splitting and aggregation, each \(P^t\) is nonnegative and column-stochastic: \[P^t \ge 0, \qquad \mathbf{1}^\top P^t=\mathbf{1}^\top. \label{eq:col-stoch}\tag{54}\] Moreover, due to the fixed out-neighbor pushing rule and the mass-splitting scheme in the implementation, there exists a uniform constant \(\beta\in(0,1)\) such that for any event \(t\), \[(P^t)_{ij}>0 \;\Longrightarrow\;(P^t)_{ij}\ge \beta, \label{eq:weight-lb}\tag{55}\] and \((P^t)_{ij}>0\) only if either \(i=j\) (self-retention) or \(j\to i\) is an active directed communication in the underlying graph.
Let the \(B\)-step product be \(\Phi^{t:s}\triangleq P^{t}P^{t-1}\cdots P^{s}\) for \(t\ge s\). Under Assumption (A5) (bounded connectivity: the union of communication graphs over any window of length \(B\) is strongly connected), standard results on products of nonnegative column-stochastic matrices imply that every \(B\)-step product is scrambling (also called uniformly ergodic): there exists a constant \(\eta=\eta(N,B,\beta)\in(0,1)\) such that \[\tau\!\left(\Phi^{t+B-1:t}\right) \le 1-\eta,\qquad \forall t\ge 0, \label{eq:scrambling}\tag{56}\] where \(\tau(\cdot)\) denotes the (Hajnal/Dobrushin) coefficient of ergodicity for column-stochastic matrices, e.g., \(\tau(A)\triangleq \frac{1}{2}\max_{p,q}\sum_{k=1}^{N}|A_{kp}-A_{kq}|.\) It satisfies the submultiplicativity \(\tau(AB)\le \tau(A)\tau(B)\) for any column-stochastic \(A,B\). Consequently, \[\tau(\Phi^{t:0}) \le \prod_{r=0}^{\lfloor t/B\rfloor-1}\tau(\Phi^{(r+1)B-1:rB}) \le (1-\eta)^{\lfloor t/B\rfloor}. \label{eq:tau-geometric}\tag{57}\] (See, e.g., classical ergodicity theorems for stochastic matrix products and push-sum / ratio-consensus analyses.)
Now define the push-sum de-biased ratio \(w_i^t=x_i^t/y_i^t\) and \(\bar w^t=X_{\mathrm{tot}}^t/Y_{\mathrm{tot}}\) as in Section 9.4. Under Assumption 9.6.1.1 (mass bounded away from \(0\)), the ratio map is Lipschitz: \[\left\| \frac{x}{y}-\frac{x'}{y} \right\|_2 \le \frac{1}{y_{\min}}\|x-x'\|_2, \qquad \forall\, y\in[y_{\min},y_{\max}]^N, \label{eq:ratio-lipschitz}\tag{58}\] where division is element-wise. Combining 57 58 with the standard relation between \(\tau(\cdot)\) and disagreement, we obtain that the (unperturbed) push-sum dynamics contracts disagreement geometrically: there exists \(\rho\in(0,1)\) (e.g., \(\rho\triangleq (1-\eta)^{1/B}\)) such that \[\frac{1}{N}\sum_{i=1}^{N}\|w_i^{t+1}-\bar w^{t+1}\|_2^2 \le \rho\cdot \frac{1}{N}\sum_{i=1}^{N}\|w_i^{t}-\bar w^{t}\|_2^2, \label{eq:ps-contract}\tag{59}\] where \(\rho\) depends only on \((N,B,\beta)\) (and the ratio-Lipschitz constant via \(y_{\min}\)).
Finally, returning to the perturbed dynamics of our algorithm (local update drift, bounded staleness and compression), the contraction 59 applies to the mixing part, while the remaining effects enter as additive perturbations. This justifies the appearance of the term \(\rho\,\mathbb{E}[\mathcal{E}_{\mathrm{con}}^{t}]\) in the recursion and closes Step 5.
Combining Steps 1-5, averaging over nodes and taking expectations yields the recursion ?? , with constants \(C_1,C_2,C_3\) absorbing the fixed numerical factors from Steps 1-4 and the mixing constants from 57 . ◻
Lemma 4 (Bounding the local-update drift by regularized stochastic gradients). Fix an activation event \(t\) at client \(i=i_t\). Let \(w_{i,0}=w_i^{t+\frac{1}{3}}\) be the post-aggregation model and let \(\tilde{w}_i^t\) be the centroid proximal anchor used during the subsequent local update (Section 9.3, C2). Client \(i\) performs \(E\) masked proximal-SGD steps: \[\begin{align} w_{i,e+1} &= \Pi_{M_i^t}\!\Bigl( w_{i,e} - \eta\, g_i(w_{i,e};\xi_{i,e}) \\ &\qquad\quad - 2\eta\lambda\,(w_{i,e}-\tilde{w}_i^t) \Bigr), \qquad e=0,\dots,E-1 . \end{align} \label{eq:lemma4-local-sgd}\qquad{(9)}\]
and outputs \(w_i^{t+\frac{2}{3}} \triangleq w_{i,E}\). Define the client drift \(\Delta_i^t \triangleq w_i^{t+\frac{2}{3}}-w_i^{t+\frac{1}{3}} = w_{i,E}-w_{i,0}\).
Assume the mask operator is non-expansive: \[\|\Pi_{M_i^t}(u)-\Pi_{M_i^t}(v)\|_2 \le \|u-v\|_2,\quad \forall u,v, \label{eq:mask-nonexp}\qquad{(10)}\] (which holds for \(\Pi_{M}(u)=u\odot M\) with \(M\in\{0,1\}^d\) as in the implementation), and Assumption (A2) (unbiased stochastic gradients with variance \(\sigma^2\)). Then the drift satisfies the deterministic bound \[\|\Delta_i^t\|_2^2 \;\le\; E\,\eta^2 \sum_{e=0}^{E-1} \Big\| g_i(w_{i,e};\xi_{i,e}) + 2\lambda\,(w_{i,e}-\tilde{w}_i^t) \Big\|_2^2 . \label{eq:drift-det-bound}\qquad{(11)}\] Moreover, conditioning on the history up to the start of the local update (denote the filtration by \(\mathcal{F}_t\)), \[\begin{align} \mathbb{E}\!\left[\|\Delta_i^t\|_2^2 \mid \mathcal{F}_t\right] &\le 2E\,\eta^2 \sum_{e=0}^{E-1} \Big( \big\|\nabla f_i(w_{i,e}) + 2\lambda\,(w_{i,e}-\tilde{w}_i^t)\big\|_2^2 + \sigma^2 \Big) \label{eq:drift-exp-bound} \\ &\le 4E\,\eta^2 \sum_{e=0}^{E-1} \Big( \|\nabla f_i(w_{i,e})\|_2^2 + 4\lambda^2\|w_{i,e}-\tilde{w}_i^t\|_2^2 + \sigma^2 \Big). \label{eq:drift-exp-bound-split} \end{align}\] {#eq: sublabel=eq:eq:drift-exp-bound,eq:eq:drift-exp-bound-split} If additionally Assumption (A3) (bounded heterogeneity) holds, then for all \(w\), \(\|\nabla f_i(w)\|_2^2 \le 2\|\nabla F(w)\|_2^2 + 2\zeta^2\) and thus \[\mathbb{E}\!\left[\|\Delta_i^t\|_2^2 \mid \mathcal{F}_t\right] \le 8E\,\eta^2 \sum_{e=0}^{E-1} \Big( \|\nabla F(w_{i,e})\|_2^2 + \zeta^2 + 2\lambda^2\|w_{i,e}-\tilde{w}_i^t\|_2^2 + \tfrac{1}{2}\sigma^2 \Big). \label{eq:drift-exp-bound-hetero}\qquad{(12)}\]
Proof. We prove the drift bound by first controlling each single masked proximal-SGD step via non-expansiveness and then aggregating the \(E\) step increments using Cauchy-Schwarz. For the expectation bound, we apply the standard variance decomposition under unbiased stochastic gradients (Assumption (A2)) and finally separate the regularized gradient into the loss gradient and the proximal term.
We emphasize that the mask \(\Pi_{M_i^t}(u)=u\odot M_i^t\) does not increase distances (Eq. ?? ), hence it can only reduce the step length compared with the unmasked update; this is the only place where pruning enters the analysis.
Let \(u_{i,e} \triangleq w_{i,e} - \eta\, g_i(w_{i,e};\xi_{i,e}) - 2\eta\lambda\,(w_{i,e}-\tilde{w}_i^t).\) By the update rule ?? and non-expansiveness ?? , \[\begin{align} \|w_{i,e+1}-w_{i,e}\|_2 &= \|\Pi_{M_i^t}(u_{i,e}) - \Pi_{M_i^t}(w_{i,e})\|_2 \le \|u_{i,e}-w_{i,e}\|_2 \nonumber\\ &= \eta\,\| g_i(w_{i,e};\xi_{i,e}) + 2\lambda\,(w_{i,e}-\tilde{w}_i^t)\|_2 . \label{eq:one-step-inc} \end{align}\tag{60}\]
Since \(\Delta_i^t = \sum_{e=0}^{E-1}(w_{i,e+1}-w_{i,e})\), Cauchy-Schwarz gives \[\|\Delta_i^t\|_2^2 = \Big\|\sum_{e=0}^{E-1}(w_{i,e+1}-w_{i,e})\Big\|_2^2 \le E \sum_{e=0}^{E-1}\|w_{i,e+1}-w_{i,e}\|_2^2 . \label{eq:cs-sum}\tag{61}\] Substituting 60 into 61 yields ?? .
By Assumption (A2), \(\mathbb{E}[g_i(w_{i,e};\xi_{i,e})\mid w_{i,e}] = \nabla f_i(w_{i,e})\) and \(\mathbb{E}[\|g_i(w_{i,e};\xi_{i,e})-\nabla f_i(w_{i,e})\|_2^2\mid w_{i,e}]\le \sigma^2\). Let \(a_{i,e}\triangleq \nabla f_i(w_{i,e}) + 2\lambda(w_{i,e}-\tilde{w}_i^t)\) and \(\delta_{i,e}\triangleq g_i(w_{i,e};\xi_{i,e})-\nabla f_i(w_{i,e})\). Then \[g_i(w_{i,e};\xi_{i,e}) + 2\lambda(w_{i,e}-\tilde{w}_i^t) = a_{i,e} + \delta_{i,e}.\] Using \(\|a+\delta\|_2^2 \le 2\|a\|_2^2 + 2\|\delta\|_2^2\) and taking conditional expectation given \(\mathcal{F}_t\) (which fixes the trajectory up to \(w_{i,e}\)), we obtain \[\mathbb{E}\!\left[\|a_{i,e}+\delta_{i,e}\|_2^2 \mid \mathcal{F}_t\right] \le 2\|a_{i,e}\|_2^2 + 2\sigma^2.\] Plugging this into ?? yields ?? . Finally, applying \(\|u+v\|^2\le 2\|u\|^2+2\|v\|^2\) to \(a_{i,e}=\nabla f_i(w_{i,e}) + 2\lambda(w_{i,e}-\tilde{w}_i^t)\) gives ?? . Under Assumption (A3), the standard bound \(\|\nabla f_i(w)\|^2 \le 2\|\nabla F(w)\|^2 + 2\zeta^2\) yields ?? . ◻
Lemma 5 (Centroid proximal regularization suppresses local deviation from the anchor). Fix an activation event \(t\) at client \(i=i_t\) and consider the \(E\) masked proximal-SGD steps in ?? . Let the (time-\(t\)) centroid proximal anchor be \(\tilde{w}_i^t\) and define the anchor deviation \[u_{i,e} \triangleq w_{i,e}-\tilde{w}_i^t,\qquad e=0,\dots,E, \label{eq:u-def}\qquad{(13)}\] where \(w_{i,0}=w_i^{t+\frac{1}{3}}\) and \(w_{i,E}=w_i^{t+\frac{2}{3}}\). Assume (i) the masking operator is non-expansive as in ?? and (ii) the stepsize satisfies \[\eta\lambda \le \frac{1}{2}. \label{eq:eta-lambda-cond}\qquad{(14)}\] Then, for every local step \(e=0,\dots,E-1\), we have the one-step anchor-contraction inequality* \[\|u_{i,e+1}\|_2^2 \;\le\; (1-\eta\lambda)\,\|u_{i,e}\|_2^2 \;+\; \frac{\eta}{\lambda}\,\big\|g_i(w_{i,e};\xi_{i,e})\big\|_2^2. \label{eq:anchor-contraction}\tag{62}\] Moreover, taking conditional expectation given the filtration \(\mathcal{F}_t\) at the start of the local update and using Assumption (A2), we obtain \[\mathbb{E}\!\left[\|u_{i,e+1}\|_2^2 \mid \mathcal{F}_t\right] \;\le\; (1-\eta\lambda)\,\mathbb{E}\!\left[\|u_{i,e}\|_2^2 \mid \mathcal{F}_t\right] \;+\; \frac{\eta}{\lambda}\Big(\|\nabla f_i(w_{i,e})\|_2^2 + \sigma^2\Big). \label{eq:anchor-contraction-exp}\tag{63}\] Consequently, by unrolling the recursion, \[\begin{align} \mathbb{E}\!\left[\|u_{i,e}\|_2^2 \mid \mathcal{F}_t\right] \;\le\;& (1-\eta\lambda)^e \, \|u_{i,0}\|_2^2 \\ &\;+\; \frac{\eta}{\lambda} \sum_{r=0}^{e-1} (1-\eta\lambda)^{e-1-r} \Big( \|\nabla f_i(w_{i,r})\|_2^2 + \sigma^2 \Big) . \end{align} \label{eq:u-unroll}\tag{64}\] *
and the cumulative anchor deviation over the whole local update satisfies \[\sum_{e=0}^{E-1}\mathbb{E}\!\left[\|u_{i,e}\|_2^2 \mid \mathcal{F}_t\right] \;\le\; \frac{1}{\eta\lambda}\,\|u_{i,0}\|_2^2 + \frac{\eta}{\lambda^2}\sum_{e=0}^{E-1}\Big(\|\nabla f_i(w_{i,e})\|_2^2+\sigma^2\Big). \label{eq:u-sum-bound}\qquad{(15)}\]
Proof. The key idea is to track the deviation from the (fixed-in-this-event) centroid anchor \(u_{i,e}\triangleq w_{i,e}-\tilde{w}_i^t\). Rewriting the masked proximal-SGD step in terms of \(u_{i,e}\) reveals a linear contraction component induced by the quadratic regularizer and an additive forcing term driven by the stochastic gradient. We then use (i) non-expansiveness of the masking operator to drop the mask without loosening the bound in the wrong direction and (ii) a sharp Young/AM-GM inequality to control the cross term between \(u_{i,e}\) and the stochastic gradient. Finally, we take conditional expectation under Assumption (A2) and unroll the resulting recursion to obtain the cumulative deviation bound.
Define \(u_{i,e}=w_{i,e}-\tilde{w}_i^t\) and note that \(\tilde{w}_i^t\) is fixed during the \(E\) local steps at event \(t\). From ?? , \[w_{i,e+1} = \Pi_{M_i^t}\!\left( w_{i,e} - \eta\, g_i(w_{i,e};\xi_{i,e}) - 2\eta\lambda\,(w_{i,e}-\tilde{w}_i^t) \right).\] Subtract \(\tilde{w}_i^t\) from both sides and use \(\Pi_M(v)-\Pi_M(\tilde{w})=\Pi_M(v-\tilde{w})\) for \(\Pi_M(u)=u\odot M\): \[\begin{align} u_{i,e+1} &= \Pi_{M_i^t}\!\Bigl( u_{i,e} - \eta\, g_i(w_{i,e};\xi_{i,e}) - 2\eta\lambda\,u_{i,e} \Bigr) \nonumber\\ &= \Pi_{M_i^t}\!\Bigl( (1-2\eta\lambda)u_{i,e} - \eta\, g_i(w_{i,e};\xi_{i,e}) \Bigr). \label{eq:u-update} \end{align}\tag{65}\]
By non-expansiveness ?? , \[\|u_{i,e+1}\|_2^2 \le \|(1-2\eta\lambda)u_{i,e} - \eta\, g_i(w_{i,e};\xi_{i,e})\|_2^2. \label{eq:nonexp-apply}\tag{66}\] Let \(a=\eta\lambda\in(0,\frac{1}{2}]\) and \(g=g_i(w_{i,e};\xi_{i,e})\). Expanding the RHS gives \[\begin{align} \|(1-2a)u - \eta g\|_2^2 &= (1-2a)^2\|u\|_2^2 + \eta^2\|g\|_2^2 - 2\eta(1-2a)\langle u,g\rangle. \label{eq:expand} \end{align}\tag{67}\] Apply the inequality \(2\langle u,g\rangle \le a\|u\|_2^2 + \frac{1}{a}\|g\|_2^2\) to the cross term: \[-2\eta(1-2a)\langle u,g\rangle \le \eta(1-2a)\left(a\|u\|_2^2 + \frac{1}{a}\|g\|_2^2\right).\] Substituting into 67 yields \[\begin{align} \|(1-2a)u - \eta g\|_2^2 \;\le\;& \Big((1-2a)^2 + a(1-2a)\Big)\, \|u\|_2^2 + \Big(\eta^2 + \eta(1-2a)\tfrac{1}{a}\Big)\, \|g\|_2^2 \nonumber\\ =\;& (1-3a+2a^2)\, \|u\|_2^2 + \Big(\eta^2 + \eta\big(\tfrac{1}{a}-2\big)\Big)\, \|g\|_2^2 . \label{eq:coeffs} \end{align}\tag{68}\]
Since \(a\in(0,1]\) implies \(1-3a+2a^2 \le 1-a\) and \(\eta^2 + \eta(\tfrac{1}{a}-2)\le \eta\cdot\tfrac{1}{a}\) for \(a\le 1\), we obtain \[\|(1-2a)u - \eta g\|_2^2 \le (1-a)\|u\|_2^2 + \frac{\eta}{\lambda}\|g\|_2^2. \label{eq:key-ineq}\tag{69}\] Combining 66 and 69 gives 62 .
Taking conditional expectation of 62 given \(\mathcal{F}_t\) and using Assumption (A2), \(\mathbb{E}[\|g_i(w_{i,e};\xi_{i,e})\|_2^2 \mid \mathcal{F}_t] \le \|\nabla f_i(w_{i,e})\|_2^2 + \sigma^2,\) yields 63 . Unrolling the linear recursion gives 64 . Finally, summing 63 over \(e=0,\dots,E-1\) and using \(\sum_{e=0}^{E-1}(1-\eta\lambda)^e \le \frac{1}{\eta\lambda}\) and \(\sum_{e=r}^{E-1}(1-\eta\lambda)^{e-r} \le \frac{1}{\eta\lambda}\) yields ?? . ◻
Lemma 6 (One-event optimization descent of the de-biased global reference (tight, complete)). Let \(F(w)\triangleq \frac{1}{N}\sum_{i=1}^{N} f_i(w)\). Suppose Assumption (A1) (\(L\)-smoothness), Assumption (A2) (unbiased stochastic gradients with variance \(\sigma^2\)), Assumption (A3) (bounded heterogeneity with constant \(\zeta^2\)) and Assumption (A6) (bounded absolute compression error \(\varepsilon_c\)) hold.
Consider an event \(t\) where client \(i=i_t\) is activated. Let \(\bar w^t=X_{\mathrm{tot}}^t/Y_{\mathrm{tot}}\) be the de-biased global reference and define \[\gamma_t \triangleq \frac{y_i^{t+\frac{1}{3}}}{Y_{\mathrm{tot}}}\in(0,1]. \label{eq:gamma-def}\qquad{(16)}\] Let \(\{w_{i,e}\}_{e=0}^{E}\) denote the local trajectory during the subsequent local update, where \(w_{i,0}=w_i^{t+\frac{1}{3}}\) and \(w_{i,E}=w_i^{t+\frac{2}{3}}\). Define \(\Delta_i^t\triangleq w_i^{t+\frac{2}{3}}-w_i^{t+\frac{1}{3}}=w_{i,E}-w_{i,0}\).
Then \(\bar w^{t+1}\) admits the exact decomposition \[\bar w^{t+1}=\bar w^t+\gamma_t\Delta_i^t+\delta_t, \label{eq:wbar-update}\qquad{(17)}\] where the broadcast-induced perturbation \(\delta_t\) satisfies \[\|\delta_t\|_2 \le c_t\,\varepsilon_c,\qquad c_t \triangleq \frac{d_i^+}{d_i^+ + 1}\cdot \frac{y_i^{t+\frac{1}{3}}}{Y_{\mathrm{tot}}} \label{eq:delta-bound}\qquad{(18)}\] by Lemma 2.
Moreover, conditioning on the filtration \(\mathcal{F}_t\) at the start of the local update, the following descent inequality holds:
\[\begin{align} \mathbb{E}\!\left[F(\bar w^{t+1}) \mid \mathcal{F}_t\right] \;\le\;& F(\bar w^t) - \frac{\eta\gamma_t E}{2}\, \|\nabla F(\bar w^t)\|_2^2 \nonumber\\ &\;+\; \eta\gamma_t E(\zeta^2+\sigma^2) \nonumber\\ &\;+\; \eta\gamma_t L^2 \sum_{e=0}^{E-1} \mathbb{E}\!\left[ \|w_{i,e}-\bar w^t\|_2^2 \,\middle|\, \mathcal{F}_t \right] \nonumber\\ &\;+\; \frac{L}{2}\gamma_t^2\, \mathbb{E}\!\left[ \|\Delta_i^t\|_2^2 \,\middle|\, \mathcal{F}_t \right] \nonumber\\ &\;+\; 2L\|\delta_t\|_2^2 . \label{eq:descent-final} \end{align}\qquad{(19)}\]
Finally, the trajectory term in ?? admits the explicit tightening \[\begin{align} \sum_{e=0}^{E-1} \mathbb{E}\!\left[ \|w_{i,e}-\bar w^t\|_2^2 \,\middle|\, \mathcal{F}_t \right] \;\le\;& 2E\|w_i^{t+\frac{1}{3}}-\bar w^t\|_2^2 + 4E\|u_{i,0}\|_2^2 + 4\sum_{e=0}^{E-1} \mathbb{E}\!\left[ \|u_{i,e}\|_2^2 \,\middle|\, \mathcal{F}_t \right]. \label{eq:traj-tight-in-lem6} \end{align}\qquad{(20)}\]
where \(u_{i,e}\triangleq w_{i,e}-\tilde{w}_i^t\) and \(\tilde{w}_i^t\) is the centroid proximal anchor used at event \(t\).
Proof. We analyze a single event by separating buffer aggregation, local update and broadcast. Aggregation preserves the total push-sum mass and only redistributes existing numerators. The local update changes the system numerator through the client drift \(\Delta_i^t\) without affecting its mass, while the subsequent broadcast introduces a bounded perturbation due to lossy centroid re-encoding. This yields the exact decomposition \(\bar w^{t+1}=\bar w^t+\gamma_t\Delta_i^t+\delta_t\). Applying \(L\)-smoothness of \(F\) to this update and bounding the resulting terms along the local trajectory lead to the stated one-event descent inequality.
By Lemma 1, \(Y_{\mathrm{tot}}\) is invariant. Buffer aggregation only transfers existing mass/numerators from buffers to node \(i\) and hence does not change \(X_{\mathrm{tot}}\). During the local update, only node \(i\) changes its model from \(w_i^{t+\frac{1}{3}}\) to \(w_i^{t+\frac{2}{3}}\) while its mass remains \(y_i^{t+\frac{1}{3}}\) (mass splitting happens after local training). Thus, \[X_{\mathrm{tot}}^{t+\frac{2}{3}}-X_{\mathrm{tot}}^{t} = y_i^{t+\frac{1}{3}}\big(w_i^{t+\frac{2}{3}}-w_i^{t+\frac{1}{3}}\big) = y_i^{t+\frac{1}{3}}\Delta_i^t,\] and therefore \[\bar w^{t+\frac{2}{3}}-\bar w^t = \frac{X_{\mathrm{tot}}^{t+\frac{2}{3}}-X_{\mathrm{tot}}^{t}}{Y_{\mathrm{tot}}} = \gamma_t\Delta_i^t. \label{eq:wbar-mid}\tag{70}\] The subsequent broadcast step induces the perturbation \(\delta_t=\bar w^{t+1}-\bar w^{t+\frac{2}{3}}\), which satisfies ?? by Lemma 2. Combining gives ?? .
By \(L\)-smoothness of \(F\) (Assumption (A1)) and ?? ,
\[\begin{align} F(\bar w^{t+1}) \;\le\;& F(\bar w^t) + \left\langle \nabla F(\bar w^t),\; \gamma_t\Delta_i^t+\delta_t \right\rangle + \frac{L}{2} \|\gamma_t\Delta_i^t+\delta_t\|_2^2 \nonumber\\ \;\le\;& F(\bar w^t) + \gamma_t \left\langle \nabla F(\bar w^t),\; \Delta_i^t \right\rangle + \left\langle \nabla F(\bar w^t),\; \delta_t \right\rangle \nonumber\\ &\;+\; \frac{L}{2}\gamma_t^2 \|\Delta_i^t\|_2^2 + L\|\delta_t\|_2^2 . \label{eq:smooth-descent} \end{align}\tag{71}\]
where we used \(\|a+b\|^2\le 2\|a\|^2+2\|b\|^2\) to upper bound the quadratic term.
Next, bound the linear \(\delta_t\) term via \(2\langle a,b\rangle \le \|a\|^2+\|b\|^2\): \[\left\langle \nabla F(\bar w^t),\;\delta_t\right\rangle \le \frac{1}{4}\|\nabla F(\bar w^t)\|_2^2 + \|\delta_t\|_2^2. \label{eq:delta-linear}\tag{72}\] Substituting 72 into 71 yields
\[\begin{align} F(\bar w^{t+1}) \;\le\;& F(\bar w^t) + \gamma_t \left\langle \nabla F(\bar w^t),\; \Delta_i^t \right\rangle \nonumber\\ &\;+\; \frac{1}{4} \|\nabla F(\bar w^t)\|_2^2 + \frac{L}{2}\gamma_t^2 \|\Delta_i^t\|_2^2 + (L+1)\|\delta_t\|_2^2 . \label{eq:smooth-descent-2} \end{align}\tag{73}\]
Absorbing constants (replace \((L+1)\) by \(2L\) w.l.o.g.for \(L\ge 1\)) gives the \(\delta_t\) term in ?? .
Ignoring the non-expansive mask for notational simplicity (it can only reduce the step length), the local update satisfies \[w_{i,e+1}-w_{i,e} = -\eta\,g_i(w_{i,e};\xi_{i,e}) -2\eta\lambda\,(w_{i,e}-\tilde{w}_i^t).\] Summing over \(e=0,\dots,E-1\) gives \[\Delta_i^t = -\eta\sum_{e=0}^{E-1} g_i(w_{i,e};\xi_{i,e}) -2\eta\lambda\sum_{e=0}^{E-1}(w_{i,e}-\tilde{w}_i^t). \label{eq:Delta-expand}\tag{74}\] Taking the inner product with \(\nabla F(\bar w^t)\) and conditioning on \(\mathcal{F}_t\), we use \(\mathbb{E}[g_i(w_{i,e};\xi_{i,e})\mid\mathcal{F}_t]=\nabla f_i(w_{i,e})\) (Assumption (A2)) to obtain \[\begin{align} \mathbb{E}\!\left[ \left\langle \nabla F(\bar w^t),\Delta_i^t \right\rangle \mid \mathcal{F}_t \right] \;=\;& -\eta \sum_{e=0}^{E-1} \left\langle \nabla F(\bar w^t),\; \nabla f_i(w_{i,e}) \right\rangle - 2\eta\lambda \sum_{e=0}^{E-1} \left\langle \nabla F(\bar w^t),\; w_{i,e}-\tilde{w}_i^t \right\rangle . \label{eq:inner-delta} \end{align}\tag{75}\]
We now lower bound the first inner product and upper bound the second in absolute value.
Decompose \[\nabla f_i(w_{i,e}) = \nabla F(\bar w^t) + \big(\nabla f_i(w_{i,e})-\nabla f_i(\bar w^t)\big) + \big(\nabla f_i(\bar w^t)-\nabla F(\bar w^t)\big).\] Then
\[\begin{align} \left\langle \nabla F(\bar w^t),\nabla f_i(w_{i,e}) \right\rangle \;=\;& \|\nabla F(\bar w^t)\|_2^2 \;+\; \left\langle \nabla F(\bar w^t),\; \nabla f_i(w_{i,e})-\nabla f_i(\bar w^t) \right\rangle \;+\; \left\langle \nabla F(\bar w^t),\; \nabla f_i(\bar w^t)-\nabla F(\bar w^t) \right\rangle \nonumber\\ \;\ge\;& \frac{3}{4}\|\nabla F(\bar w^t)\|_2^2 \;-\;\|\nabla f_i(w_{i,e})-\nabla f_i(\bar w^t)\|_2^2 \;-\; \|\nabla f_i(\bar w^t)-\nabla F(\bar w^t)\|_2^2 \nonumber\\ \;\ge\;& \frac{3}{4}\|\nabla F(\bar w^t)\|_2^2 - L^2\|w_{i,e}-\bar w^t\|_2^2 - \zeta^2 . \label{eq:inner-lb-lem6} \end{align}\tag{76}\]
where we used \(L\)-smoothness (\(\|\nabla f_i(w)-\nabla f_i(v)\|\le L\|w-v\|\)) and Assumption (A3).
For the second term in 75 , use \(2\langle a,b\rangle \le \|a\|^2+\|b\|^2\) with \(a=\nabla F(\bar w^t)\) and \(b=2\lambda(w_{i,e}-\tilde{w}_i^t)\): \[-2\lambda\left\langle \nabla F(\bar w^t),\;w_{i,e}-\tilde{w}_i^t\right\rangle \le \frac{1}{4}\|\nabla F(\bar w^t)\|_2^2 + 4\lambda^2\|w_{i,e}-\tilde{w}_i^t\|_2^2. \label{eq:prox-cross}\tag{77}\]
Substitute 76 and 77 into 75 , sum over \(e=0,\dots,E-1\), and multiply by \(\gamma_t\). The \(\frac{1}{4}\|\nabla F(\bar w^t)\|^2\) terms in 73 and 77 are dominated by the leading negative term after summation (yielding the coefficient \(\frac{1}{2}\) in front of \(\eta\gamma_t E\|\nabla F(\bar w^t)\|^2\)). Collecting the remaining terms yields ?? after conditional expectation.
For each \(e\), \[w_{i,e}-\bar w^t=(w_i^{t+\frac{1}{3}}-\bar w^t)+(w_{i,e}-w_i^{t+\frac{1}{3}}),\] so \(\|w_{i,e}-\bar w^t\|^2\le 2\|w_i^{t+\frac{1}{3}}-\bar w^t\|^2+2\|w_{i,e}-w_i^{t+\frac{1}{3}}\|^2\). Moreover, \(w_{i,e}-w_i^{t+\frac{1}{3}}=(w_{i,e}-\tilde{w}_i^t)-(w_i^{t+\frac{1}{3}}-\tilde{w}_i^t)=u_{i,e}-u_{i,0}\), hence \(\|w_{i,e}-w_i^{t+\frac{1}{3}}\|^2\le 2\|u_{i,e}\|^2+2\|u_{i,0}\|^2\). Summing over \(e=0,\dots,E-1\) yields ?? . ◻
We now present a fully explicit (constant-closed) stationarity guarantee for PushCen-ADFL.
We use Assumptions (A1)-(A6) in Appendix 9.2 and additionally adopt the following standard technical conditions that close all constants.
There exist constants \(0<y_{\min}\le y_{\max}<\infty\) such that for all events \(t\) and all nodes \(i\), \(y_i^t \in [y_{\min},y_{\max}].\) Consequently, for the activated node \(i_t\), the normalized mass weight \[\gamma_t \triangleq \frac{y_{i_t}^{t+\frac{1}{3}}}{Y_{\mathrm{tot}}} \quad\text{satisfies}\quad \underline{\gamma}\triangleq \frac{y_{\min}}{Ny_{\max}} \le \gamma_t \le 1. \label{eq:gamma-lb}\tag{78}\]
There exists \(G^2<\infty\) such that for all \(i,w\) and any sample \(\xi\), \(\mathbb{E}\big[\|g_i(w;\xi)\|_2^2\big]\le G^2.\)
There exists \(W<\infty\) such that for all events \(t\) and all nodes \(i\), \[\|w_i^t\|_2 \le W \qquad\text{and}\qquad \|\tilde{w}_i^t\|_2 \le W . \label{eq:bounded-iterates}\tag{79}\] In particular, \(\|w_i^t-\tilde{w}_i^t\|_2^2 \le 4W^2\) for all \(i,t\).
Let \(F(w)\triangleq \frac{1}{N}\sum_{i=1}^N f_i(w)\) and \(F^\star\triangleq \inf_w F(w)\). Let \(\rho\in(0,1)\) and \(C_1,C_2,C_3>0\) be the constants in Lemma 3. Let \(\tau\) be the staleness bound (Assumption (A4)) and let \(\varepsilon_c\) be the bounded absolute compression error (Assumption (A6)). Let \(d_{\max}\triangleq \max_i d_i^+\). Define the uniform broadcast perturbation coefficient (Lemma 2) \[\overline{c} \triangleq \max_t \frac{d_{i_t}^+}{d_{i_t}^+ + 1}\cdot \frac{y_{i_t}^{t+\frac{1}{3}}}{Y_{\mathrm{tot}}} \le \frac{y_{\max}}{Ny_{\min}}. \label{eq:cbar}\tag{80}\]
Theorem 2 (Fully explicit stationarity bound (constant-closed)). Suppose Assumptions (A1)-(A6) and Assumptions 9.6.1.1 hold. Let the stepsize satisfy \[\eta \le \min\Big\{\frac{1}{8LE},\;\frac{1}{4\lambda}\Big\}. \label{eq:stepsize-cond-main}\qquad{(21)}\] Run PushCen-ADFL for \(T\) events. Then the averaged stationarity measure obeys \[\begin{align} \frac{1}{T}\sum_{t=0}^{T-1} \mathbb{E}\big[\|\nabla F(\bar w^t)\|_2^2\big] \;\le\;& \underbrace{ \frac{2\big(F(\bar w^0)-F^\star\big)}{\eta E\,\underline{\gamma}\,T} }_{\text{optimization term}} \nonumber\\ &\;+\; \underbrace{ 2(\sigma^2+\zeta^2) }_{\text{stochasticity \& heterogeneity}} \nonumber\\ &\;+\; \underbrace{ 2L\,\overline{c}^{\,2}\varepsilon_c^2 }_{\text{compression drift}} \label{eq:main-closed-1} \\ &\;+\; \underbrace{ \frac{4L^2 N}{\underline{\gamma}} \Bigg( \frac{\mathbb{E}[\mathcal{E}_{\mathrm{con}}^0]}{T(1-\rho)} + \frac{B_{\mathrm{con}}}{1-\rho} \Bigg) }_{\text{directed mixing / asynchrony}} \nonumber\\ &\;+\; \underbrace{ \frac{L}{\underline{\gamma}}\,D_\Delta }_{\text{client drift accumulation}} . \label{eq:main-closed-2} \end{align}\] {#eq: sublabel=eq:eq:main-closed-1,eq:eq:main-closed-2}
where the constants \(D_\Delta\) and \(B_{\mathrm{con}}\) are explicitly given by \[\begin{align} D_\Delta &\triangleq 2E^2\eta^2\Big(G^2 + 16\lambda^2W^2\Big), \label{eq:Ddelta-closed} \\ B_{\mathrm{con}} &\triangleq \big(C_1 + C_2\tau^2\big)D_\Delta + C_3\varepsilon_c^2. \label{eq:Bcon-closed} \end{align}\] {#eq: sublabel=eq:eq:Ddelta-closed,eq:eq:Bcon-closed} In particular, as \(T\to\infty\), the iterates converge to a stationary neighborhood whose radius is upper bounded by the right-hand side of ?? without the \(1/T\) terms.
Proof. We proceed by combining the descent inequality for \(F(\bar w^t)\) with the consensus recursion.
By Lemma 6 (tight form, after bounding the trajectory term inside its proof), there exist absolute constants \(\alpha_0,\alpha_1,\alpha_2,\alpha_3>0\) such that for each event \(t\) with activated node \(i_t\), \[\begin{align} \mathbb{E}\!\left[ F(\bar w^{t+1}) \mid \mathcal{F}_t \right] \;\le\;& F(\bar w^t) - \alpha_0\,\eta\gamma_t E\, \|\nabla F(\bar w^t)\|_2^2 \nonumber\\ &\;+\; \alpha_1\,\eta\gamma_t E\, (\sigma^2+\zeta^2) \nonumber\\ &\;+\; \alpha_2\,\eta\gamma_t L^2\,E\, \|w_{i_t}^{t+\frac{1}{3}}-\bar w^t\|_2^2 \nonumber\\ &\;+\; \frac{L}{2}\gamma_t^2\, \mathbb{E}\!\left[ \|\Delta_{i_t}^t\|_2^2 \mid\mathcal{F}_t \right] \nonumber\\ &\;+\; \alpha_3\,\varepsilon_c^2 . \label{eq:main-proof-descent} \end{align}\tag{81}\]
Moreover, Lemma 2 implies the broadcast perturbation contributes at most \(2L\overline{c}^{\,2}\varepsilon_c^2\) after taking expectations, which is absorbed into the \(\alpha_3\varepsilon_c^2\) term.
By Lemma 4 and Lemma 5, for each event \(t\), \[\mathbb{E}\big[\|\Delta_{i_t}^t\|_2^2\big] \le 2E^2\eta^2\Big(G^2 + 4\lambda^2\,\sup_{e}\mathbb{E}\|w_{i_t,e}-\tilde{w}_{i_t}^t\|_2^2\Big). \label{eq:drift-bound-mid}\tag{82}\] Under Assumption 9.6.1.3, \(\|w_{i_t,e}-\tilde{w}_{i_t}^t\|^2\le 4W^2\) for all \(e\), hence \[\mathbb{E}\big[\|\Delta_{i_t}^t\|_2^2\big] \le 2E^2\eta^2\Big(G^2 + 16\lambda^2W^2\Big) \;\triangleq\; D_\Delta . \label{eq:drift-closed}\tag{83}\] Using \(\gamma_t\le 1\) and taking full expectation in 81 yields \[\begin{align} \alpha_0\,\eta E\, \mathbb{E}\!\left[ \gamma_t\|\nabla F(\bar w^t)\|_2^2 \right] \;\le\;& \mathbb{E}\!\left[ F(\bar w^t)-F(\bar w^{t+1}) \right] \nonumber\\ &\;+\; \alpha_1\,\eta E\,(\sigma^2+\zeta^2) \nonumber\\ &\;+\; \alpha_2\,\eta L^2E\, \mathbb{E}\!\left[ \|w_{i_t}^{t+\frac{1}{3}}-\bar w^t\|_2^2 \right] \nonumber\\ &\;+\; \frac{L}{2}\,D_\Delta \;+\; \alpha_3\,\varepsilon_c^2 . \label{eq:descent-ready} \end{align}\tag{84}\]
Since \(\|w_{i_t}^{t+\frac{1}{3}}-\bar w^t\|^2 \le N\,\mathcal{E}_{\mathrm{con}}^t\) up to an absolute factor, it suffices to control \(\sum_t \mathbb{E}[\mathcal{E}_{\mathrm{con}}^t]\). Lemma 3 and 83 imply \[\mathbb{E}[\mathcal{E}_{\mathrm{con}}^{t+1}] \le \rho\,\mathbb{E}[\mathcal{E}_{\mathrm{con}}^{t}] + \Big(C_1 + C_2\tau^2\Big)D_\Delta + C_3\varepsilon_c^2 \;\triangleq\; \rho\,\mathbb{E}[\mathcal{E}_{\mathrm{con}}^{t}] + B_{\mathrm{con}} . \label{eq:con-rec-closed}\tag{85}\] Summing 85 over \(t=0,\dots,T-1\) and using \(\sum_{t=0}^{T-1}\rho^t\le \frac{1}{1-\rho}\) yields \[\sum_{t=0}^{T-1}\mathbb{E}[\mathcal{E}_{\mathrm{con}}^{t}] \le \frac{\mathbb{E}[\mathcal{E}_{\mathrm{con}}^{0}]}{1-\rho} + \frac{T\,B_{\mathrm{con}}}{1-\rho}. \label{eq:sum-con-closed}\tag{86}\]
Summing 84 over \(t=0,\dots,T-1\) telescopes the objective values: \[\sum_{t=0}^{T-1}\mathbb{E}\!\left[F(\bar w^t)-F(\bar w^{t+1})\right] = F(\bar w^0)-\mathbb{E}[F(\bar w^T)] \le F(\bar w^0)-F^\star.\] By Assumption 9.6.1.1, \(\gamma_t\ge \underline{\gamma}\), hence \[\sum_{t=0}^{T-1}\mathbb{E}\|\nabla F(\bar w^t)\|^2 \le \frac{1}{\underline{\gamma}} \sum_{t=0}^{T-1}\mathbb{E}\!\left[\gamma_t\|\nabla F(\bar w^t)\|^2\right].\] Plugging 86 into the sum of 84 , dividing by \(\alpha_0\eta E\,\underline{\gamma}\,T\) and absorbing fixed numerical factors into the explicit coefficients yields ?? ?? with \(D_\Delta\) and \(B_{\mathrm{con}}\) given in ?? ?? . ◻
To provide a more complete view of the learning dynamics beyond the final averaged accuracy, we report accuracy curves for all datasets. Specifically, we include (i) global test accuracy trajectories across communication rounds and (ii) accuracy trajectories of delayed clients measured on a pseudo-time axis that reflects asynchronous progress. These curves complement the main results by visualizing convergence speed, stability and robustness under system asynchrony.
This subsection reports the evolution of global test accuracy across communication rounds. Figures 8 (a)-8 (c) show the global accuracy curves on CIFAR-10, CIFAR-100 and Tiny-ImageNet, respectively. These curves illustrate the overall convergence behavior of different methods, including their convergence speed and final performance, under identical experimental settings.
Figure 8: Global test accuracy curves on CIFAR-10, CIFAR-100, and Tiny-ImageNet.. a — CIFAR-10, b — CIFAR-100, c — Tiny-ImageNet
This subsection presents accuracy curves for delayed clients, evaluated along a pseudo-time axis that reflects asynchronous local progress. Figures 9, 11 and 13 report the average test accuracy across delayed clients on CIFAR-10, CIFAR-100 and Tiny-ImageNet, respectively. In addition, Figures 10, 12 and 14 visualize the accuracy trajectories of representative individual delayed clients, providing a fine-grained view of training stability under client delays.