Performance Analysis and Optimization for a Multi-tasking Federated Learning Framework

doi:10.1007/978-3-032-10202-7_8

Return

Performance Analysis and Optimization for a Multi-tasking Federated Learning Framework

2026-01-01

PRE

Xuan-Phuc Phan Nguyen

Phu-Quoc Pham

Thao Nguyen-Thi-Thanh

Nga Ly-Tu *

DOI：10.1007/978-3-032-10202-7_8

Original

Original request for help

Save

Abstract

En 中文

Federated learning (FL) has been growing rapidly in recent years, thanks to its advantages in enhancing privacy and security while maintaining high accuracy. However, challenges such as overfitting, adaptive optimization, client heterogeneity, and communication efficiency impact real-world performance. Hence, optimizations for both the server and its clients play an important role in federated learning. This paper suggests FedSAP, a federated learning framework for training multiple models at once. To improve generalization and model convergence, we leverage enhancements such as warm-up rounds, pseudo-gradient, class weighting, and adaptive learning rates with dual cosine decay. Specifically, we conducted one simulated and two distributed experiments covering 7 optimizers, namely FedAvg, FedProx, SCAFFOLD, FedAvgM, FedOpt, FedAdam, and FjORD over 3 different datasets, including our RSSITag, VGG-Face2, and FERPlus, on various edge devices, from high-computing-power devices like RTX 3090 Ti, 4060 and 4070 to resource-constrained devices such as Jetson Orin Nano and Jetson Nano. Then, we present general criteria for non-IID data partitioning. Finally, our simulated results on the CIFAR-10 dataset achieve a 1.55% to 7.54% improvement in accuracy compared to the traditional FL framework.

Keywords:

Federated learning

UHF-RFID

Face Recognition

Facial Emotion Recognition

Attendance system