Fuzzy Logic Assisted Client Selection and Energy-Efficient Joint Optimization for Hierarchical Federated Learning

Zhihao Dong,Xu Zhu,Jie Cao,Yufei Jiang,Vincent K. N. Lau,Sumei Sun
DOI: https://doi.org/10.1109/icc45041.2023.10279572
2023-01-01
Abstract:In this paper, we investigate multi-criteria client selection and energy consumption minimization for hierarchical federated learning (HFL) to deal with clients' heterogeneity and limited energy. To the best of our knowledge, this is the first work to investigate multi-criteria client selection for HFL. A fuzzy logic assisted client selection (FLACS) scheme is proposed, where multiple criteria are taken into account, including the distance, clients' battery capacity and computational resource. The FLACS scheme enables a significant performance gain in terms of the clients' average normalized suitability over the previous schemes. A joint communication and learning factors optimization (JCLFO) algorithm is proposed to minimize the system energy consumption. Thanks to the derived closed-form expressions for the optimal aggregation intervals, computation frequency and transmission power, the JCLFO algorithm can achieve the optimal performance in terms of the system energy consumption and converge within only 3 iterations, with a significant complexity reduction over exhaustive search.
What problem does this paper attempt to address?