Abstract:Federated learning (FL) has become one of the key methods for privacy-preserving collaborative learning, as it enables the transfer of models without requiring local data exchange. Within the FL framework, an aggregation algorithm is recognized as one of the most crucial components for ensuring the efficacy and security of the system. Existing average aggregation algorithms typically assume that all client-trained data holds equal value or that weights are based solely on the quantity of data contributed by each client. In contrast, alternative approaches involve training the model locally after aggregation to enhance adaptability. However, these approaches fundamentally ignore the inherent heterogeneity between different clients' data and the complexity of variations in data at the aggregation stage, which may lead to a suboptimal global model. To address these issues, this study proposes a novel dual-criterion weighted aggregation algorithm involving the quantity and quality of data from the client node. Specifically, we quantify the data used for training and perform multiple rounds of local model inference accuracy evaluation on a specialized dataset to assess the data quality of each client. These two factors are utilized as weights within the aggregation process, applied through a dynamically weighted summation of these two factors. This approach allows the algorithm to adaptively adjust the weights, ensuring that every client can contribute to the global model, regardless of their data's size or initial quality. Our experiments show that the proposed algorithm outperforms several existing state-of-the-art aggregation approaches on both a general-purpose open-source dataset, CIFAR-10, and a dataset specific to visual obstacle avoidance.

Adaptive Aggregation Weight Assignment for Federated Learning: A Deep Reinforcement Learning Approach

Understanding the Training Dynamics in Federated Deep Learning via Aggregation Weight Optimization

Revisiting Weighted Aggregation in Federated Learning with Neural Networks

An Element-Wise Weights Aggregation Method for Federated Learning

Optimizing Hierarchical Federated Learning: A Reinforcement Learning Approach

FedAA: A Reinforcement Learning Perspective on Adaptive Aggregation for Fair and Robust Federated Learning

FedSW: Federated learning with adaptive sample weights

AsyncFedED: Asynchronous Federated Learning with Euclidean Distance Based Adaptive Weight Aggregation

A Fair Federated Learning Framework with Reinforcement Learning.

FedAgg: Adaptive Federated Learning with Aggregated Gradients

Dual-Criterion Model Aggregation in Federated Learning: Balancing Data Quantity and Quality

Optimizing Federated Learning on Non-IID Data Using Local Shapley Value.

Hierarchical Federated Learning with Adaptive Clustering on Non-IID Data

An Aggregation-Free Federated Learning for Tackling Data Heterogeneity

DRL-empowered joint batch size and weighted aggregation adjustment mechanism for federated learning on non-IID data

Is Aggregation the Only Choice? Federated Learning via Layer-wise Model Recombination

FedDWA: Personalized Federated Learning with Dynamic Weight Adjustment

DWFed: A statistical- heterogeneity-based dynamic weighted model aggregation algorithm for federated learning

An Optimization Method for Non-IID Federated Learning Based on Deep Reinforcement Learning

Dynamic Fair Federated Learning Based on Reinforcement Learning

Improving Federated Learning with Quality-Aware User Incentive and Auto-Weighted Model Aggregation.