APPROACH
B. Physical Setup
FUTURE WORK
Federated Learning (FL) has emerged as a paramount approach for advancing artificial intelligence beyond centralized models, particularly due to its ability to harness individuals’ data while maintaining privacy. The largest hurdle toward evaluating FL algorithms is the difficulty of conducting real-world experiments. Most FL research results have been obtained through simulation, typically running on a single server, with few studies utilizing actual experimental testbeds. However, simulations often use simplified system models and cannot reliably assess certain performance aspects, such as individual device energy consumption or real-world behavior under various constraints like data, network, and device heterogeneity.
The limitations of simulation underscore the critical need for real-world testbeds that can account for the complexities of real-world FL environments. To address these challenges, our experiment will use NVFlare and our physical setup with edge devices to uncover previously unknown trade-offs and inefficiencies to provide a more holistic understanding of FL system performance.
INTRODUCTION
George Igwegbe, Martins Awojide, Adetola Adetunji
RESULTS
Real-World Federated Learning on Edge Devices with NVFLARE
C. Specifications and Parameters
Number of Admin/Server | 1/1 |
Number of Clients | 3 |
Rounds | 5 |
Epoch | 5 |
Device Specification | Server | Client |
CPU | Quad-core ARM Cortex-A57 | |
GPU | 128-core Maxwell GPU | |
GPU Memory | 2GB | 4GB |
Power Supply | 5V⎓3A (Micro-USB) | 5V⎓4A (Barrel Jack) |
Software | PyTorch v1.10 and NVFlare 2.6.0 | |
A. System Design
SERVER
CLIENTS
GPU Utilization (%) - SERVER
GPU Utilization (%) - CLIENT
CPU Utilization (%) - CLIENT
Temperature (OC) - CLIENT
Power Consumption(mW) - CLIENT
Accuracy over Rounds
When training starts at F, there is an uptick in CPU (D) and GPU (B) utilization of the clients, during the training portion of the rounds, which drops after each epoch and drops even further after each round (C, E). This phenomenon is also evident in the temperature rise and power drawn by the CPU and GPU which clearly delineate the epochs (H) and rounds (G). The server’s activity spike at A lines up with the end of each training round, when model averaging and transfer occurs.
The model accuracy across client is improved after each round as shown by the increase in the clients’ average accuracy and drop in variance of the clients’ accuracies between the last epoch of the prior round (I) and the first epoch of next round (J). The training time and power consumed for our experiment was 41 minutes [for 5 rounds with 5 epochs each] and 0.012 kilowatt-hour respectively.
H
F
A
C
B
E
D
I
G
J
AISaturdays Lagos
References: