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 条
[21]   Energy Optimization in Statistical AoI-Aware MEC Systems [J].
Meng, Qi ;
Lu, Hancheng ;
Qin, Langtian .
IEEE COMMUNICATIONS LETTERS, 2024, 28 (10) :2263-2267
[22]  
Mnih V, 2016, PR MACH LEARN RES, V48
[23]  
Moltafet M, 2020, IEEE INT SYMP INFO, P1765, DOI [10.1109/ISIT44484.2020.9174099, 10.1109/isit44484.2020.9174099]
[24]   Empowering Edge Intelligence by Air-Ground Integrated Federated Learning [J].
Qu, Yuben ;
Dong, Chao ;
Zheng, Jianchao ;
Dai, Haipeng ;
Wu, Fan ;
Guo, Song ;
Anpalagan, Alagan .
IEEE NETWORK, 2021, 35 (05) :34-41
[25]  
Reisizadeh A, 2020, Arxiv, DOI arXiv:2012.14453
[26]   FedTAR: Task and Resource-Aware Federated Learning for Wireless Computing Power Networks [J].
Sun, Wen ;
Li, Zongjun ;
Wang, Qubeijian ;
Zhang, Yan .
IEEE INTERNET OF THINGS JOURNAL, 2023, 10 (05) :4257-4270
[27]   Adaptive Federated Learning and Digital Twin for Industrial Internet of Things [J].
Sun, Wen ;
Lei, Shiyu ;
Wang, Lu ;
Liu, Zhiqiang ;
Zhang, Yan .
IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2021, 17 (08) :5605-5614
[28]   On the Average Age-of-Information for Hybrid Multiple Access Protocols [J].
Vineeth, B.S. ;
Thomas, Reuben Chacko .
IEEE Networking Letters, 2022, 4 (02) :87-91
[29]   Distributed and Secure Federated Learning for Wireless Computing Power Networks [J].
Wang, Peng ;
Sun, Wen ;
Zhang, Haibin ;
Ma, Wenqiang ;
Zhang, Yan .
IEEE TRANSACTIONS ON VEHICULAR TECHNOLOGY, 2023, 72 (07) :9381-9393
[30]   Minimizing the Age-of-Critical-Information: An Imitation Learning-Based Scheduling Approach Under Partial Observations [J].
Wang, Xiaojie ;
Ning, Zhaolong ;
Guo, Song ;
Wen, Miaowen ;
Poor, H. Vincent .
IEEE TRANSACTIONS ON MOBILE COMPUTING, 2022, 21 (09) :3225-3238