TY - JOUR
T1 - Energy-Efficient Federated Learning Through UAV Edge under Location Uncertainties
AU - Wang, Chen
AU - Tang, Xiao
AU - Zhai, Daosen
AU - Zhang, Ruonan
AU - Ussipov, Nurzhan
AU - Zhang, Yan
N1 - Publisher Copyright:
© 2013 IEEE.
PY - 2024
Y1 - 2024
N2 - Federated Learning (FL) and Mobile Edge Computing (MEC) technologies alleviate the burden of deploying artificial intelligence (AI) on wireless devices with low computational capabilities. However, they also introduce energy consumption challenges in FL model training and data processing. In this paper, we employ Unmanned Aerial Vehicles (UAVs) to collect data from wireless devices and carry edge servers to assist the central server located at the base station in training FL model. We also consider the deviation of UAVs' locations to address its impact on network performance. Specifically, we formulate a robust joint optimization problem to minimize the energy consumption of UAVs, considering the computational resources, transmit power, transmission time, and FL model accuracy. Moreover, Gaussian-distributed uncertainties caused by deviation in UAV locations result in probabilistic constraints on data offloading. We initially employ the Bernstein -type inequality (BTI) to transform probabilistic constraints into deterministic forms. Subsequently, we adopt the Block Coordinate Descent (BCD) to separate the problem into three subproblems. Simulation results demonstrate a significant reduction in energy consumption and superiority in robustness.
AB - Federated Learning (FL) and Mobile Edge Computing (MEC) technologies alleviate the burden of deploying artificial intelligence (AI) on wireless devices with low computational capabilities. However, they also introduce energy consumption challenges in FL model training and data processing. In this paper, we employ Unmanned Aerial Vehicles (UAVs) to collect data from wireless devices and carry edge servers to assist the central server located at the base station in training FL model. We also consider the deviation of UAVs' locations to address its impact on network performance. Specifically, we formulate a robust joint optimization problem to minimize the energy consumption of UAVs, considering the computational resources, transmit power, transmission time, and FL model accuracy. Moreover, Gaussian-distributed uncertainties caused by deviation in UAV locations result in probabilistic constraints on data offloading. We initially employ the Bernstein -type inequality (BTI) to transform probabilistic constraints into deterministic forms. Subsequently, we adopt the Block Coordinate Descent (BCD) to separate the problem into three subproblems. Simulation results demonstrate a significant reduction in energy consumption and superiority in robustness.
KW - energy consumption
KW - federated learning
KW - Mobile edge computing
KW - robust optimization
UR - http://www.scopus.com/inward/record.url?scp=85208366926&partnerID=8YFLogxK
U2 - 10.1109/TNSE.2024.3489554
DO - 10.1109/TNSE.2024.3489554
M3 - 文章
AN - SCOPUS:85208366926
SN - 2327-4697
JO - IEEE Transactions on Network Science and Engineering
JF - IEEE Transactions on Network Science and Engineering
ER -