Learn More by Using Less: Distributed Learning with Energy-Constrained Devices
This work addresses energy limitations in FL for real-world deployments on battery-constrained devices, offering an incremental improvement to mitigate client dropout and extend device availability.
The paper tackled the problem of energy constraints in Federated Learning (FL) by proposing LeanFed, an energy-aware framework that optimizes client selection and training workloads, resulting in enhanced model accuracy and stability, with consistent improvements on CIFAR-10 and CIFAR-100 datasets under high data heterogeneity and limited battery conditions.
Federated Learning (FL) has emerged as a solution for distributed model training across decentralized, privacy-preserving devices, but the different energy capacities of participating devices (system heterogeneity) constrain real-world implementations. These energy limitations not only reduce model accuracy but also increase dropout rates, impacting on convergence in practical FL deployments. In this work, we propose LeanFed, an energy-aware FL framework designed to optimize client selection and training workloads on battery-constrained devices. LeanFed leverages adaptive data usage by dynamically adjusting the fraction of local data each device utilizes during training, thereby maximizing device participation across communication rounds while ensuring they do not run out of battery during the process. We rigorously evaluate LeanFed against traditional FedAvg on CIFAR-10 and CIFAR-100 datasets, simulating various levels of data heterogeneity and device participation rates. Results show that LeanFed consistently enhances model accuracy and stability, particularly in settings with high data heterogeneity and limited battery life, by mitigating client dropout and extending device availability. This approach demonstrates the potential of energy-efficient, privacy-preserving FL in real-world, large-scale applications, setting a foundation for robust and sustainable pervasive AI on resource-constrained networks.