journal article·2024-01-08Open access

Joint Client and Resource Optimization for Federated Learning in Wireless IoT Networks

Jie Zhao, Yiyang Ni, Yulun Cheng

Applied Sciences· Volume 14· Issue 2

Abstract

Federated learning (FL) is a promising technique to provide intelligent services for the internet of things (IoT). By transmitting the model parameters instead of user data between the client and central server, FL greatly improves the user privacy and reduces transmission latency. However, due to the fading effects of the wireless channel, the outage of wireless transmission degenerates the learning efficiency when FL is applied in wireless IoT networks. In order to address this issue, we investigate the joint optimization of client selection and wireless resource allocation in FL-aided cellular IoT networks. By taking both the amount of training data and wireless resource consumption into consideration, we formulate the problem as a mixed integer non-linear programming to maximize the utility of the network. To solve the problem effectively, an alternative direction-based algorithm is proposed by decomposing the original problem into two sub problems. The simulation results indicate that the proposed algorithm substantially improves the FL learning performance and reduces the consumption of wireless resources compared with existing methods.
Source: crossref · Quality: baseline

Research context

OpenAlex topics: Privacy-Preserving Technologies in Data · Advanced MIMO Systems Optimization · Advanced Wireless Communication Technologies

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

Affiliations

College of Physics and Information Engineering, Jiangsu Second Normal University, Nanjing 210013, China

Jiangsu Province Engineering Research Center of Basic Education Big Data Application, Jiangsu Second Normal University, Nanjing 210013, China

Jiangsu Key Laboratory of Wireless Communications, Nanjing University of Posts and Telecommunications, Nanjing 210003, China

Funding

Jiangsu University Philosophy and Social Science Research Fund · Intelligent Reflective Surface · Natural Science Foundation on Frontier Leading Technology Basic Research Project of Jiangsu

References (24)

Show source-provided references
1. A Survey on Federated Learning for Resource-Constrained IoT Devices10.1109/JIOT.2021.3095077 ↗
2. McMahan, H.B., Moore, E., Ramage, D., Hampson, S., and Arcas, B.A.Y. (2017, January 20–22). Communication-efficient Learning of Deep Networks from Decentralized Data. Proceedings of the 20th International Conference on Artificial Intelligence and Statistics, Ft. Lauderdale, FL, USA.
3. Toward Scalable Wireless Federated Learning: Challenges and Solutions10.1109/IOTM.001.2300099 ↗
4. Data Sampling in Federated Learning: Principles, Features and Taxonomy10.1109/MCOMSTD.0004.2200076 ↗
5. Client Selection and Bandwidth Allocation in Wireless Federated Learning Networks: A Long-Term Perspective10.1109/TWC.2020.3031503 ↗
6. Nishio, T., and Yonetani, R. (2019, January 20–24). Client Selection for Federated Learning with Heterogeneous Resources in Mobile Edge. Proceedings of the 2019 IEEE International Conference on Communications, Shanghai, China.10.1109/ICC.2019.8761315 ↗
7. Amiri, M.M., Gündüz, D., Kulkarni, S.R., and Poor, H.V. (2020, January 21–26). Update Aware Device Scheduling for Federated Learning at the Wireless Edge. Proceedings of the 2020 IEEE International Symposium on Information Theory, Los Angeles, CA, USA.10.1109/ISIT44484.2020.9173960 ↗
8. A Joint Learning and Communications Framework for Federated Learning over Wireless Networks10.1109/TWC.2020.3024629 ↗
9. Wireless Distributed Learning: A New Hybrid Split and Federated Learning Approach10.1109/TWC.2022.3213411 ↗
10. Distributed and Secure Federated Learning for Wireless Computing Power Networks10.1109/TVT.2023.3247859 ↗
11. FedTAR: Task and Resource-Aware Federated Learning for Wireless Computing Power Networks10.1109/JIOT.2022.3215805 ↗
12. GoMORE: Global Model Reuse for Resource-Constrained Wireless Federated Learning10.1109/LWC.2023.3281881 ↗
13. Online Client Selection for Asynchronous Federated Learning with Fairness Consideration10.1109/TWC.2022.3211998 ↗
14. Scheduling and Aggregation Design for Asynchronous Federated Learning over Wireless Networks10.1109/JSAC.2023.3242719 ↗
15. Hierarchical Personalized Federated Learning over Massive Mobile Edge Computing Networks10.1109/TWC.2023.3260141 ↗
16. Device Selection and Resource Allocation for Layerwise Federated Learning in Wireless Networks10.1109/JSYST.2022.3169461 ↗
17. Joint Scheduling and Resource Allocation for Hierarchical Federated Edge Learning10.1109/TWC.2022.3144140 ↗
18. Federated Learning over Wireless IoT Networks with Optimized Communication and Resources10.1109/JIOT.2022.3151193 ↗
19. Communication-Efficient Federated Deep Learning With Layerwise Asynchronous Model Update and Temporally Weighted Aggregation10.1109/TNNLS.2019.2953131 ↗
20. Weighted Bandwidth–Power Product Optimization in Downlink Femtocell Networks10.1109/LCOMM.2015.2456171 ↗
21. Caching in Energy Harvesting Aided Internet of Things: A Game-Theoretic Approach10.1109/JIOT.2018.2880483 ↗
22. Dual-Battery Enabled Profit Driven User Association in Green Heterogeneous Cellular Networks10.1109/TGCN.2018.2869039 ↗
23. Multiarea Inertia Estimation Using Convolutional Neural Networks and Federated Learning10.1109/JSYST.2021.3134599 ↗
24. The MNIST Database of Handwritten Digit Images for Machine Learning Research [Best of the Web]10.1109/MSP.2012.2211477 ↗