Inverse Probability Weighting and Age-of-Information Aggregation for Decentralized Federated Learning under Partial Reception

When focusing on the low-latency “fire-and-forget” regime, where retransmissions are infeasible, decentralized wireless networks face spatial incompleteness and temporal freshness issues. Model parameters are serialized into fixed-size chunks. Each chunk is dropped independently, so the receiving node receives an incomplete parameter vector. Critically, these two failure modes are independent in origin. Selection bias from spatial incompleteness arises from channel quality, while staleness arises from compute and communication heterogeneity and must be addressed jointly within a single aggregation rule to avoid compounding their effects.

Most existing production-level Federated Learning approaches assume reliable communication, even when they do not explicitly state so [10, 11, 8]. When it comes to DFL [12, 13, 14, 15, 16] and wireless DFL deployments focusing on vehicular systems [17, 18, 19, 20], most works still assume perfect communication. However, a lightweight solution that meets low-latency constraints is required to address the aforementioned challenge in DFL configurations, especially when operating in a wireless regime with no or minimal retransmissions. Systematic solutions in the literature for handling partial receptions without retransmissions either drop partially received updates or naively fill in the missing data from the local model [21, 22] or zeros or maintain a complex network coding to reconstruct missing data chunks [23]. These approaches lead either to wasted communication resources by ignoring updates, to selection bias due to naive partial acceptance, or to extra overhead in heavy computation. Reconstruction methods for lossy channels alone cannot withstand realistic conditions in a wireless DFL setting, as described earlier. Moreover, the integration of model freshness effects into partial model reception in configurations remains limited, even though it should be properly managed given the inherent asynchrony. This gap is particularly acute in IoT and sensor network deployments, where per-link channel quality is determined by physical distance and interference, yet existing gossip aggregation protocols are limited in their ability to estimate and correct per-link reception rates online.

We observe that selection bias is a statistical sampling problem where the inbox is a biased sample of the neighborhood because links with lower reception rates appear less frequently. The classical remedy is the Horvitz-Thompson estimator [24], which reweights each observed sample by the inverse of its inclusion probability. DFL-AA instantiates this principle in the gossip aggregation step with local-fill reconstruction for missing chunks: each model update(reconstructed) from neighbor $j$ is weighted by $1/\hat{q}_{ij}$, where $\hat{q}_{ij}$ is an online EWMA estimate of the observed link reception rate, up-weighting under-represented neighbors to restore a statistically unbiased sample of the neighborhood. This is combined with an AoI (Age of Information) [25]-based exponential decay $\exp(-\mathrm{AoI}_{ij}/\tau)$, computed from generation timestamps in received messages without a global clock, to discount temporally stale contributions. The combined weight addresses both failure modes independently and multiplicatively, where a high weight requires both good channel quality and temporal freshness.

1. C1

   Problem positioning. We provide the first systematic study of the joint effect of selection bias and update staleness in asynchronous DFL over fixed directed graphs in wireless networks, revealing drastic performance degradation of state-of-the-art methods.

2. C2

   Failure mode analysis. We prove that uniform gossip under local-fill reconstruction distorts each neighbor’s contribution by $(1-q_{ij})$ (Proposition 1), and that push-sum weight variables drain geometrically under chunk-level loss (Proposition 2), which is a fundamental study on directed graphs.

3. C3

   DFL-AA algorithm. We introduce DFL-AA (Decentralized Federated Learning with Adaptive AoI-weighted Aggregation), an IPW-corrected, AoI-aware gossip aggregation rule for asynchronous fixed directed graphs, with online EWMA channel estimation requiring no coordination, no global clock, and no additional communication overhead.

4. C4

   Unbiasedness guarantee. We prove that DFL-AA removes the $(1-q_{ij})$ coefficient distortion in expectation, weighting each neighbor purely by AoI freshness (Theorem 1).

5. C5

   Empirical validation. We evaluate on EMNIST and CIFAR-10 across $20$- and $80$-node fixed directed topologies using discrete-event priority queue simulation, demonstrating consistent improvement over all baselines across loss levels $10\%$–$50\%$, with additional validation under heterogeneous per-link channel conditions.

The remainder of the paper is structured as follows. Section II reviews related work and discusses the gaps identified with positioning our work. Section III introduces the system model, including the network model, decentralized learning, and missing value reconstruction. Section IV presents the DFL-AA solution with all formulations. Section V demonstrates the validation of design choices. Section VI presents the performance evaluation, and Section VII concludes the paper and outlines future directions.

To overcome the bottlenecks and single point of failure (SPoF) of client-server FL, decentralized federated learning (DFL) has emerged as a scalable and robust alternative [1, 2]. Existing DFL research has explored heterogeneity-aware asynchronous learning that employed a coordinator that predicts the runtime configuration for the next round based on past data and probability distributions [12], adaptive topology and communication optimization [13, 15], asynchronous model synchronization to mitigate stragglers [14], and adaptive neighbor weighting strategies [16]. While these approaches improve efficiency and scalability, they generally assume reliable communication and do not explicitly address partial model reception and update staleness in lossy wireless environments.

Staleness naturally arises when nodes train and communicate at different rates. FedAsync and FedBuff address this in centralized settings by weighting or buffering delayed updates, but staleness is typically defined in round- or iteration-based terms (i.e., how many server/global steps have elapsed since an update was computed) rather than by wall-clock time [31, 26]. AD-PSGD [29] allows fully decentralized asynchronous operation under a bounded-delay assumption. SWIFT [30] removes the bounded-delay requirement, achieving $O(1/\!\sqrt{T})$ convergence guarantees. FedASMU and BLADE [32, 33] follow similar principles, using dynamic, staleness-aware updates. However, all of these methods assume lossless delivery. They are designed for reliable networks and lack a mechanism to handle incomplete model receptions. However, in lossy wireless communication, model updates may be received only partially, resulting in incomplete information used during aggregation and, consequently, degrading learning performance.

A smaller body of work explicitly addresses the reception of partial models. Soft-DSGD [21] adopts a UDP-based fire-and-forget model and fills missing parameters with local values. AMA [22] uses a similar local-fill strategy with link-quality thresholding. Kwon et al. [23] reconstruct missing parameters via an SVD-based approximation combined with systematic network coding, but report 172% packet overhead and operate in a centralized server setting that is not applicable to peer-to-peer topologies [23]. DFL-AA requires no additional communication overhead beyond the model transmission itself. Crucially, none of these methods corrects the selection bias introduced by partial reception. Furthermore, none of these methods handle staleness as they assume synchronous or near-synchronous operation, which does not hold in genuinely asynchronous wireless deployments.

Communication-efficient approaches such as knowledge distillation [34, 35], adaptive sparse training [36] and RL-based clustering [37] reduce bandwidth consumption but assume reliable delivery and do not address the partial-reception regime.

The partial reception problem is particularly acute in wireless deployments, where per-link channel quality varies with distance, interference, and node heterogeneity [38]. Wireless FL research has explored vehicular networks [17, 18, 19], UAV systems [20], and mobility-aware optimization [39], but these works focus on scheduling and resource allocation under reliable communication rather than correcting aggregation bias from packet loss.

In IoT and sensor network settings, early work on in-network aggregation [40] established the importance of handling partial data collection in sensor networks to reduce communication overhead. However, this work was designed to reduce aggregation overhead but does not explicitly consider the stochastically received model aggregation. FL for IoT under communication constraints [41, 42] typically assumes reliable delivery or relies on centralized gateway aggregation, neither of which applies to our setting.

Table I summarizes how DFL-AA relates to existing methods. No prior work jointly addresses partial reception and staleness in asynchronous wireless DFL: approaches that handle partial updates typically ignore staleness, while asynchronous methods assume lossless communication. In addition, round-based staleness metrics are poorly suited, where heterogeneous computation and communication delays create irregular round gaps that fail to reflect true model freshness; Age-of-Information (AoI) provides a more appropriate measure in such environments. In contrast, DFL-AA is the only method that jointly corrects selection bias due to lossy links and captures temporal staleness via AoI, operating on directed graphs without requiring synchronized rounds or lossless delivery assumptions, which are properties particularly valuable for IoT and sensor network deployments in a wireless setting where per-link channel quality is heterogeneous, and retransmission is infeasible.

We consider $n$ nodes connected by a fixed directed graph $\mathcal{G}=(\mathcal{V},\mathcal{E})$ that remains fixed over time. A directed edge $(j,i)\in\mathcal{E}$ indicates that node $j$ can transmit to node $i$. Let $\mathcal{N}(i)=\{j:(j,i)\in\mathcal{E}\}$ denote the in-neighborhood of node $i$. In contrast, $\mathcal{N}_{out}(i)$ denote the out-neighborhood. We assume $\mathcal{G}$ is strongly connected.

Each node $i$ holds private dataset $\mathcal{D}_{i}$ drawn from local distribution $\mathcal{P}_{i}$. Each node maintains a local model $\mathbf{w}_{i}\in\mathbb{R}^{d}$ cooperating to minimize a global objective:

where $f_{i}(\mathbf{w})=\mathbb{E}_{\xi\sim\mathcal{D}_{i}}[\ell(\mathbf{w};\xi)]$ is the local loss for node $i$ over its private data distribution $\mathcal{D}_{i}$, and $\ell(\cdot)$ is the loss function. We consider the non-IID setting $\mathcal{P}_{i}\neq\mathcal{P}_{j}$, modeled by Dirichlet-$\alpha$ label partitioning.

We adopt a fire-and-forget UDP-like communication model. Each node serializes its model $\mathbf{w}_{i}\in\mathbb{R}^{d}$ into $K$ fixed-size chunks, each containing $\lfloor d/K\rfloor$ parameters. Chunk $k$ on link $(j\to i)$ is received independently with probability $q_{ij}\in(0,1]$.

Nodes operate without a global synchronization barrier. Each node $i$ maintains:

• •

  Local model $\mathbf{w}_{i}\in\mathbb{R}^{d}$.

• •

  Inbox $\mathcal{B}_{i}$ storing the most recent $(\tilde{\mathbf{w}}_{j},c_{ij},t_{\mathrm{gen}}^{j})$ from each in-neighbor, where $t_{\mathrm{gen}}^{j}$ is generation timestamp. Older entries are overwritten on the new reception.

• •

  EWMA estimate $\hat{q}_{ij}$ for each in-neighbor, updated on each reception.

DFL-AA addresses two independent failure modes with two independent components:

Spatial: IPW corrects selection bias. Under partial reception, low-quality links (low $q_{ij}$) contribute fewer messages, leading to systematic under-representation. This creates selection bias, where the received models are not a uniform sample of neighbors. DFL-AA uses the Horvitz–Thompson [24] correction by weighting each received update with $1/\hat{q}_{ij}$, so weaker links are up-weighted to compensate for their lower visibility.

Temporal: AoI decay discounts staleness. DFL-AA addresses the staleness issue by applying an exponential decay $\exp(-\mathrm{AoI}_{ij}/\tau)$, reducing the influence of outdated updates relative to fresh ones. This operates independently of IPW, where the first corrects spatial sampling bias, while the second accounts for temporal staleness.

Table II summarizes the notation used throughout Algorithm 1, which presents DFL-AA at a single node $i$, and Figure 3 illustrates the overall process of the proposed method.

Since the true reception rate $q_{ij}$ is unknown a priori and may vary with distance and interference, DFL-AA estimates it online via EWMA:

where $c_{ij}$ is the observed completeness of the most recent transmission from $j$ and $\beta\in(0,1)$ is the EWMA forgetting factor. The estimate is initialized to the first observed $c_{ij}$ and floored at $q_{\mathrm{floor}}>0$ to prevent $1/\hat{q}_{ij}$ from becoming numerically degenerate: $\hat{q}_{ij}\leftarrow\max(\hat{q}_{ij},q_{\mathrm{floor}})$.

The aggregation step at node $i$ is a normalized weighted average:

where $\mathcal{B}_{i}^{+}=\{j\in\mathcal{B}_{i}:c_{ij}\geq c_{\min}\}$ excludes transmissions below the minimum completeness threshold $c_{\min}$ and $\hat{\mathbf{w}}_{j}$ is local-fill reconstructed model parameters according to Equation (3) and $a_{ij}^{(t)}=(1/{\hat{q}_{ij}})\cdot\exp(-{\mathrm{AoI}_{ij}^{(t)}}/{\tau})$ with $\tau$ as hyperparameter for decay.

Push-sum [28] is the canonical method for gossip-based SGD on directed graphs and is the algorithmic family closest to DFL-AA’s setting. However, we show that it is fundamentally incompatible with chunk-level packet loss, making it unsuitable for directed DFL under lossy communication.

The IPW and AoI factors must be combined multiplicatively rather than additively. An additive combination $a_{ij}=1/\hat{q}_{ij}+\exp(-\mathrm{AoI}_{ij}/\tau)$ would allow a very fresh but structurally under-represented neighbor to receive an inflated weight regardless of link quality, or vice versa. The multiplicative form enforces a both-must-be-good criterion: a neighbor receives high weight only if it is both well-represented across rounds (low $1/\hat{q}_{ij}$ correction needed) and temporally fresh (low AoI).

Table III reports test accuracy, loss, and AUC for all methods with $n=20$ nodes across all loss levels under a heterogeneous data distribution among nodes (Dirichlet $\alpha=0.1$). DFL-AA achieves the highest accuracy in every configuration across both datasets. As loss increases, the advantage of DFL-AA grows monotonically, reaching 73.19% on EMNIST and 49.39% on CIFAR-10 at $q=50\%$, outperforming the next-best method (SWIFT) by 5.91 pp and 12.18 pp respectively. This monotonic widening directly validates Proposition 1, the $(1-q_{ij})$ coefficient distortion grows with loss rate, and DFL-AA’s IPW correction becomes proportionally more valuable as the channel degrades.

Among the baselines, AD-PSGD and SWIFT perform second-best despite assuming full delivery, because we augment them with local-fill reconstruction to ensure a fair comparison. Soft-DSGD, the method designed for partial reception, underperforms on both EMNIST and CIFAR-10, validating that local-fill reconstruction alone cannot be robust in asynchronous environments. Strikingly, FedAvg, which simply drops partial updates, stays as a lower bound, limiting itself to training the local model due to its architectural design.

DFL-AA emerges as the best solution across different channel loss levels on every metric, including accuracy, loss, and AUC, thereby validating our unbiasedness Theorem 1 for DFL-AA aggregation over asynchronous lossy channels.

Figure 4 and  5 show accuracy and consensus distance over virtual time at loss levels 10–50% for both datasets. Three observations are consistent across all configurations.

First, DFL-AA converges faster and achieves higher accuracy than all baselines across all loss levels, and remains stable at higher loss rates. The convergence speed advantage is most pronounced under a high loss rate (at $q=50\%$) on EMNIST; DFL-AA reaches 70% accuracy, while all other baselines fail to reach that level at that loss level. FedAvg remains below 30% throughout the 1000 s runtime due to its local-only training as described in the main results section. Furthermore, the visible gap between all the baselines and DFL-AA widens on both datasets as channel loss increases $10\%\rightarrow 50\%$, again validating Proposition 1. The reduction in accuracy, along with an increase in channel loss, is minimal in DFL-AA compared to the best baseline. As loss increases from 10% to 50%, DFL-AA degrades by only 1.67 pp on EMNIST and 1.08 pp on CIFAR-10, while the next-best method SWIFT degrades by 4.75 pp and 4.5 pp, respectively. A relative degradation 2.84$\times$ and 4.17$\times$ larger than DFL-AA on EMNIST and CIFAR-10, respectively, confirming that IPW correction becomes proportionally more effective as channel quality worsens.

Second, Soft-DSGD exhibits a degradation pattern due to its inability to operate in asynchronous environments, despite being designed to handle partial model reception. This confirms that the local-fill reconstruction alone is insufficient in asynchronous wireless systems.

Third, Figure 5 shows that DFL-AA achieves strictly lower consensus distance than all baselines across all loss levels and both datasets. This confirms that DFL-AA not only improves individual node accuracy but also drives the network toward tighter model agreement, a property that follows from eliminating the directional distortion in Proposition 1, which would otherwise push different nodes’ aggregates in systematically different directions depending on their local link quality profiles.

Figure 6 shows accuracy and consensus distance for $n\in\{20,40,80\}$ nodes at fixed $10\%$ link loss. DFL-AA maintains its advantage across all node counts on both datasets. As $n$ increases, absolute accuracy decreases slightly across all methods due to sparser effective neighborhoods at a fixed average degree of 4 with fewer data samples, but DFL-AA’s relative advantage is preserved. Consensus distance for DFL-AA remains consistently below all baselines across all scales, confirming that the IPW correction scales correctly with network size and that DFL-AA is robust to scalability.

Notably, FedAvg’s accuracy collapses at 40 and 80 nodes on both datasets, because it limits its training on a local model as it drops partial receptions ($10\%$ loss), and due to increasing node counts, local data belonging to each node is limited, and it pushes the model to produce low accuracy proportional to node count on a test dataset. There is another notable performance degradation for Soft-DSGD as the node count increases, whereas AD-PSGD, SWIFT, and DFL-AA remain stable. This might be due to extreme heterogeneity and limited local data at each node, as well as directional bias introduced by local filling and averaging, which Soft-DSGD is designed to mitigate.

Table IV reports performance across three topologies at $10\%$ link loss for ring, fully connected, and Erdős-R’enyi random directed graph with average degree 4.

DFL-AA maintains competitive performance across all topologies. On the fully connected graph, Soft-DSGD achieves the second-highest accuracy on both datasets, only being defeated by DFL-AA. The gap is tiny, and it validates the Soft-DSGD design, as the authors demonstrate its robustness on a fully connected topology.

In the ring topology, all methods suffer significant accuracy degradation due to poor network connectivity, where each node has only two neighbors, thereby limiting information flow. DFL-AA maintains parity with SWIFT and AD-PSGD on this topology, confirming that the IPW correction is topology-agnostic, operating on per-link reception statistics available locally and not depending on global topology knowledge. Furthermore, our direct competitor, Soft-DSGD, performs well only in a fully connected regime, which its authors tested; we demonstrate its degradation under low information flow (random neighbors and ring topologies), while DFL-AA maintains stability across topologies, demonstrating its robustness.

Figure 8 illustrates the performance of each method on the CIFAR-10 dataset under $10\%$ link loss at two levels of heterogeneity ($\alpha=0.5$ and $\alpha=1.0$). As $\alpha$ increases, the heterogeneity of the data distribution decreases, helping each node achieve higher accuracy than at lower $\alpha$ values in the same configuration. Figure 8 clearly shows that the gap between DFL-AA and other baselines decreases with higher $\alpha$ values (lower data heterogeneity), but DFL-AA still manages to outperform other baselines by a clear margin, validating IPW-correctness under lossy channels along with staleness decay. However, the consensus distance between DFL-AA and the other best baselines (AD-PSGD and SWIFT) is not clearly visible at low levels of heterogeneity. These experimental results emphasize the importance of DFL-AA in wireless lossy network systems, where participating nodes mostly have heterogeneous data and operate asynchronously, and DFL-AA performs extremely well.

Table V reports DFL-AA accuracy on EMNIST under 10% loss for seven values of the AoI decay constant $\tau\in\{1,2,3,5,10,15,20\}$. Accuracy is stable across the range $\tau\in[3,15]$, with the peak at $\tau=10$ (74.89%). A very small $\tau=1$ slightly overpenalizes stale updates, reducing accuracy by 0.87 pp relative to the peak. A very large $\tau$ also degrades accuracy as the AoI weight approaches uniformity. The mean AoI in this experiment is 1.09 s with a maximum of 2.0 s, and the stable range is $\tau\in[3,15]$, suggesting a practical tuning rule that sets $\tau$ to 3–10$\times$ the expected mean AoI in the deployment.

Table VI shows DFL-AA accuracy across three staleness levels (mean AoI: 1.09, 1.38, 1.96 s) with fixed $\tau=5$. Accuracy degrades gracefully as staleness increases. A 0.87 s increase in mean AoI reduces accuracy by only 1.81 pp, confirming that the AoI decay term effectively discounts stale updates and maintains robust performance under increased asynchrony.

Figure 7 shows $\hat{q}_{ij}$ over virtual time for a tracked link with a true reception rate of $q_{ij}=0.8$, using a shared y-axis to make the asymptotic variance directly comparable across panels. All six values of $\beta$ yield estimates centered on the true rate throughout training, confirming that the online channel-estimation assumption underlying Theorem 1 holds in practice. The figure reveals a clear bias-variance trade-off, where $\beta=0.01$ yields an almost flat estimate with minimal variance, whereas $\beta=0.30$ converges at the same rate but exhibits roughly $5\times$ wider oscillations around the true value. The default $\beta=0.05$ maintains a tight, stable estimate. The residual bias $O(|\hat{q}_{ij}-q_{ij}|\cdot\|\mathbf{w}_{j}-\mathbf{w}_{i}\|)$ from Remark 5 is therefore negligible throughout training for all reasonable $\beta$ values.