FedNorm: Optimised Federated Learning Strategy Using Hybrid Regularisation Norm
摘要
Federated Learning (FL) has emerged as a game-changing method for training machine learning (ML) models on decentralized data sources without centralizing the data itself. Conventional FL algorithms frequently exhibit performance instability and inefficiency, especially when working with non-Independent and Identically Distributed (non-IID) data distributions. To overcome these difficulties, we present FedNorm, a new FL method that uses hybrid regularisation approach to improve model stability and accuracy by balancing sparsity and smoothness, to increase generalization across varied data distributions. Our experimental setup uses the MNIST dataset to train models in both Independent and Identically Distributed (IID) and non-IID situations. The findings show that FedNorm routinely outperforms classic algorithms like FedAvg, FedSGD, FedNova and FedSAM, with minimal test loss and high accuracy. FedNorm has an exceptional accuracy rating of 0.98 on both IID and non-IID data. FedNorm addresses the constraints of conventional FL algorithms, resulting in a strong and dependable solution for FL in diverse situations.