Age-Aware Scheduling for Federated Learning with Caching in Wireless Computing Power Networks

被引:1
作者
Zhuang, Xiaochong [1 ]
Luo, Chuanbai [2 ]
Xie, Zhenghao [2 ]
Li, Yu [3 ]
Jiang, Li [4 ]
机构
[1] Guangdong Univ Technol, Guangdong Hong Kong Macao Joint Lab Smart Discrete, Key Lab Intelligent Detect & Internet Mfg Things, Minist Educ, Guangzhou 510006, Peoples R China
[2] Guangdong Univ Technol, Guangdong Prov Key Lab Intelligent Syst & Optimiza, Guangzhou 510006, Peoples R China
[3] Chongqing Technol & Business Univ, Chongqing Key Lab Intelligent Percept & BlockChain, Chongqing 400067, Peoples R China
[4] Guangdong Univ Technol, Ctr Intelligent Batch Mfg Based IoT Technol 111, Key Lab Intelligent Informat Proc & Syst Integrat, Minist Educ, Guangzhou 510006, Peoples R China
关键词
federated learning (FL); wireless computing power network (WCPN); cache mechanism; parametric age; resource scheduling; AVERAGE AGE; INFORMATION;
D O I
10.3390/electronics14040663
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
With the rapid development of Wireless Computing Power Networks (WCPNs), the urgent need for data privacy protection and communication efficiency has led to the emergence of the federated learning (FL) framework. However, the time delay leads to dragging problems and reduces the convergence performance of FL in the training process. In this article, we propose an FL resource scheduling strategy based on information age perception in WCPNs, which can effectively reduce the time delay and enhance the convergence performance of FL. Moreover, a data cache buffer and a model cache buffer are set up at the user end and the central server, respectively. Next, we formulate the parametric age-aware problem to simultaneously minimize the global parameter age, energy consumption, and FL service delays. Considering the dynamic WCPN environment, the optimization target is modeled as a Markov decision process (MDP), and the Proximal Policy Optimization (PPO) algorithm is used to achieve the optimal solution. Numerical simulation results demonstrate that the proposed method significantly outperforms baseline schemes across critical metrics. Specifically, the proposed approach reduces FL service delays by 25.2%. It also decreases the global parameter age by 45.5% through the joint optimization of the data collection frequency, computation frequency, and bandwidth allocation. The method attains a reward value of 65 at convergence, 18.2% higher than the WithoutAnyCache scheme and 8.3% higher than the OnlyLocalCache scheme. FL accuracy improves to 98.2% with a 0.08 final loss. Finally, numerical simulation results further confirm the superiority and outstanding performance of the proposed method.
引用
收藏
页数:22
相关论文
共 41 条
[1]   FedCo: A Federated Learning Controller for Content Management in Multi-party Edge Systems [J].
Balasubramanian, Venkatraman ;
Aloqaily, Moayad ;
Reisslein, Martin .
30TH INTERNATIONAL CONFERENCE ON COMPUTER COMMUNICATIONS AND NETWORKS (ICCCN 2021), 2021,
[2]  
Bie Tong, 2023, Journal of Beijing University of Aeronautics and Astronautics, P2108, DOI 10.13700/j.bh.1001-5965.2021.0580
[3]   An analytic study of caching in computer systems [J].
Buck, D ;
Singhal, M .
JOURNAL OF PARALLEL AND DISTRIBUTED COMPUTING, 1996, 32 (02) :205-214
[4]   Convergence Time Minimization of Federated Learning over Wireless Networks [J].
Chen, Mingzhe ;
Poor, H. Vincent ;
Saad, Walid ;
Cui, Shuguang .
ICC 2020 - 2020 IEEE INTERNATIONAL CONFERENCE ON COMMUNICATIONS (ICC), 2020,
[5]  
Dai W., 2019, P INT C LEARN REPR I, P1
[6]  
Dai W, 2018, Arxiv, DOI arXiv:1810.03264
[7]   Age of Information Based Client Selection for Wireless Federated Learning With Diversified Learning Capabilities [J].
Dong, Liran ;
Zhou, Yiqing ;
Liu, Ling ;
Qi, Yanli ;
Zhang, Yu .
IEEE TRANSACTIONS ON MOBILE COMPUTING, 2024, 23 (12) :14934-14945
[8]   Mean Peak Age of Information Analysis of Energy-Aware Computation Offloading in IIoT Networks [J].
Ernest, Tan Zheng Hui ;
Madhukumar, A. S. .
2024 IEEE 99TH VEHICULAR TECHNOLOGY CONFERENCE, VTC2024-SPRING, 2024,
[9]  
Fu F., 2022, arXiv, DOI DOI 10.14778/3547305.3547316
[10]  
Ha S, 2019, Arxiv, DOI arXiv:1901.05239