FedLBW: A Loss-Based Weighting Strategy for Federated Learning on Non-IID Data in Wireless Networks

arXiv cs.AI Papers
federated-learning non-iid-data wireless-networks loss-based-weighting arxiv

Summary

This paper proposes FedLBW, a federated learning aggregation strategy that weights client updates by inverse validation loss instead of dataset size, improving accuracy and robustness to non-IID data and client dropouts in wireless networks.

arXiv:2608.07007v1 Announce Type: new Abstract: Federated Learning (FL) enables collaborative machine learning (ML) across distributed clients while preserving privacy. However, efficient model convergence in FL remains challenging, especially in wireless networks where non-independent and identically distributed (non-IID) data and frequent client dropouts are common. Traditional FL algorithms, such as FedAvg, rely solely on dataset size to weight client updates. This introduces biases towards clients with larger datasets and makes the process sensitive to non-IID data, outliers, and client dropouts. To address these challenges, we propose Federated Learning with Loss-Based Weighting (FedLBW), a novel aggregation method that assigns each client's update a weight proportional to the inverse of its validation loss, computed using a small proxy dataset on the server, rather than its dataset size. This ensures that lower-loss models exert greater influence during aggregation, prioritizing the most reliable updates and boosting overall performance. Through extensive experiments across multiple datasets, including FashionMNIST (CNN), CIFAR-10 (ResNet-18), and CIFAR-100 (ResNet-34), we demonstrate that FedLBW achieves higher accuracy and faster convergence compared to baseline algorithms such as FedAvg, FedAvgM, FedProx, FedNova, FedLAW and FedDkw, with notable improvements of up to 7.6 % higher accuracy on CIFAR-10 in extreme non-IID cases. Moreover, FedLBW showcases exceptional resilience to increasing dropout probabilities, consistently maintaining significantly higher accuracy even in challenging conditions. These results establish FedLBW as an effective and resilient solution for FL in wireless network environments, offering marked improvements in model accuracy, convergence speed, and robustness to non-IID data and client dropouts.
Original Article
View Cached Full Text

Cached at: 08/10/26, 08:00 AM 

# FedLBW: A Loss-Based Weighting Strategy for Federated Learning on Non-IID Data in Wireless Networks
Source: [https://arxiv.org/abs/2608.07007](https://arxiv.org/abs/2608.07007)
[View PDF](https://arxiv.org/pdf/2608.07007)

> Abstract:Federated Learning \(FL\) enables collaborative machine learning \(ML\) across distributed clients while preserving privacy\. However, efficient model convergence in FL remains challenging, especially in wireless networks where non\-independent and identically distributed \(non\-IID\) data and frequent client dropouts are common\. Traditional FL algorithms, such as FedAvg, rely solely on dataset size to weight client updates\. This introduces biases towards clients with larger datasets and makes the process sensitive to non\-IID data, outliers, and client dropouts\. To address these challenges, we propose Federated Learning with Loss\-Based Weighting \(FedLBW\), a novel aggregation method that assigns each client's update a weight proportional to the inverse of its validation loss, computed using a small proxy dataset on the server, rather than its dataset size\. This ensures that lower\-loss models exert greater influence during aggregation, prioritizing the most reliable updates and boosting overall performance\. Through extensive experiments across multiple datasets, including FashionMNIST \(CNN\), CIFAR\-10 \(ResNet\-18\), and CIFAR\-100 \(ResNet\-34\), we demonstrate that FedLBW achieves higher accuracy and faster convergence compared to baseline algorithms such as FedAvg, FedAvgM, FedProx, FedNova, FedLAW and FedDkw, with notable improvements of up to 7\.6 % higher accuracy on CIFAR\-10 in extreme non\-IID cases\. Moreover, FedLBW showcases exceptional resilience to increasing dropout probabilities, consistently maintaining significantly higher accuracy even in challenging conditions\. These results establish FedLBW as an effective and resilient solution for FL in wireless network environments, offering marked improvements in model accuracy, convergence speed, and robustness to non\-IID data and client dropouts\.

## Submission history

From: Taehong Kim \[[view email](https://arxiv.org/show-email/bfe83162/2608.07007)\] **\[v1\]**Fri, 7 Aug 2026 09:21:37 UTC \(730 KB\)

Similar Articles