On Energy Optimization for Hierarchical Federated Learning With Delay Constraint Through Node Cooperation

Zhuo Li,Sailan Zou,Song Guo
DOI: https://doi.org/10.1109/jiot.2024.3349393
IF: 10.6
2024-01-01
IEEE Internet of Things Journal
Abstract:In Hierarchical Federated Learning (HFL), edge computing is introduced for partial model aggregation to reduce latency. High energy cost is an important issue to be solved in the process of parameters uploading. Our study focuses on the issue of minimizing energy cost with delay constraint through node cooperation in HFL, and the decision problem for this is NP-hard. We introduce a cost-efficient hierarchical federated learning framework CE-HFL, where nodes not only participate in model training but also transmit and aggregate model parameters for neighbours. A parameter aggregation tree is firstly generated, and parameter updates can be delivered to edge servers along paths in the tree while being aggregated simultaneously. Through theoretical analysis, it is proved that CE-HFL can achieve energy optimization with delay constraint. We also evaluate its performance through thorough experiments. In comparison with HierFAVG, CFL and HFEL, it is found that CE-HFL can save energy cost up to 24.58%, 22.02%, and 6.60% respectively.
computer science, information systems,telecommunications,engineering, electrical & electronic
What problem does this paper attempt to address?