Situation-Aware Cluster and Quantization Level Selection Algorithm for Fast Federated Learning

doi:10.1109/JIOT.2023.3262582

Situation-Aware Cluster and Quantization Level Selection Algorithm for Fast Federated Learning

2023-08-01

2

PRE

AI

DOI：10.1109/JIOT.2023.3262582

Original

Original request for help

Save

Abstract

En 中文

In federated learning (FL), which clients and quantization levels are selected for the deep model parameters has a significant impact on learning time as well as learning accuracy. This is not a trivial issue because it is also significantly affected by factors, such as computational power, communication capacity, and data distribution. Considering these factors, we formulate a joint optimization problem for clustering and selecting clusters with quantization levels. Due to the high complexity of the formulated problem, we propose a situation-aware cluster and quantization level selection (SITUA-CQ) algorithm. In this algorithm, the FL server first assembles clients into clusters to mitigate the impact of biased data distributions and determines the most suitable clusters and quantization levels based on their computing power and channel quality. Extensive simulation results show that SITUA-CQ can reduce the round time by up to 80.3% compared to conventional algorithms.

Keywords:

Cluster and quantization level selection

clustering

federated learning (FL)

Situation-Aware Cluster and Quantization Level Selection Algorithm for Fast Federated Learning

Abstract

Journal

Organization