Content Popularity Prediction via Federated Learning in Cache-Enabled Wireless Networks

doi:10.12142/ZTECOM.202302004

ZTE Communications ›› 2023, Vol. 21 ›› Issue (2): 18-24.DOI: 10.12142/ZTECOM.202302004

收稿日期:2023-03-02 出版日期:2023-06-13 发布日期:2023-06-13

Content Popularity Prediction via Federated Learning in Cache-Enabled Wireless Networks

YAN Yuna¹, LIU Ying², NI Tao², LIN Wensheng¹, LI Lixin¹()

^1.Northwestern Polytechnical University, Xi?an 710072, China
^2.Shanghai Satellite Engineering Research Institute, Shanghai 200240, China

Received:2023-03-02 Online:2023-06-13 Published:2023-06-13
About author:YAN Yuna is currently working toward her master’s degree under the supervision of Prof. LI Lixin with the School of Electronics and Information, Northwestern Polytechnical University, China. Her research interests include federated learning, deep learning and semantic communications.|LIU Ying is an engineer of Shanghai Satellite Engineering Research Institute, China, mainly engaged in satellite system design and satellite communications.|NI Tao is a senior engineer of Shanghai Satellite Engineering Research Institute, China, mainly engaged in satellite system design and satellite communications.|LIN Wensheng received his BE degree in communication engineering and ME degree in electronic and communication engineering from Northwestern Polytechnical University, China in 2013 and 2016. He received his PhD degree in information science from the Japan Advanced Institute of Science and Technology in 2019. He is currently an associate professor with the School of Electronics and Information, Northwestern Polytechnical University. His research interests include network information theory, distributed source coding, and age of information.|LI Lixin (lilixin@nwpu.edu.cn) received his BS degree (Hons.), MS degree (Hons.), and PhD degree from Northwestern Polytechnical University (NPU), China in 2001, 2004, and 2008, respectively. From 2009 to 2011, he was a postdoctoral research fellow with NPU. In 2011, he joined the School of Electronics and Information, NPU, where he is currently a full professor and chair of Department of Communication Engineering. In 2017, he held the visiting scholar position with the University of Houston, USA. He holds 26 granted patents and has authored or coauthored five books and more than 200 peer-reviewed papers in many prestigious journals and conferences. His research interests include 5G/6G wireless networks, federated learning, game theory, and machine learning for wireless communications. He was the recipient of the 2016 NPU Outstanding Young Teacher Award, which is the highest research and education honors for young faculties in NPU. He was an exemplary reviewer for IEEE Transactions on Communications in 2020.
Supported by:
the National Natural Science Foundation of China (NSFC)(62001387);the Young Elite Scientists Sponsorship Program by the China Association for Science and Technology (CAST)(2022QNRC001);Shanghai Academy of Spaceflight Technology (SAST)(SAST2022052)

摘要/Abstract

Abstract:

With the rapid development of networks, users are increasingly seeking richer and high-quality content experience, and there is an urgent need to develop efficient content caching strategies to improve the content distribution efficiency of caching. Therefore, it will be an effective solution to combine content popularity prediction based on machine learning (ML) and content caching to enable the network to predict and analyze popular content. However, the data sets which contain users’ private data cause the risk of privacy leakage. In this paper, to address this challenge, we propose a privacy-preserving algorithm based on federated learning (FL) and long short-term memory (LSTM), which is referred to as FL-LSTM, to predict content popularity. Simulation results demonstrate that the performance of the proposed algorithm is close to the centralized LSTM and better than other benchmark algorithms in terms of privacy protection. Meanwhile, the caching policy in this paper raises about 14.3% of the content hit rate.

Key words: content popularity prediction, privacy protection, federated learning, long short-term memory

. [J]. ZTE Communications, 2023, 21(2): 18-24.

YAN Yuna, LIU Ying, NI Tao, LIN Wensheng, LI Lixin. Content Popularity Prediction via Federated Learning in Cache-Enabled Wireless Networks[J]. ZTE Communications, 2023, 21(2): 18-24.

图/表 6

Figure 1 Scenario of a cache-enabled wireless network

Figure 2 Illustration of long short-term memory (LSTM) algorithm

Figure 3 Illustration of FL-LSTM algorithm

Figure 4 Content popularity prediction by different algorithms

Figure 5 MSE loss of LSTM and FL-LSTM algorithms

Figure 6 Total hit rates versus memory capacity

参考文献 20

1	WANG C M, HE Y, YU F R, et al. Integration of networking, caching, and computing in wireless systems: a survey, some research issues, and challenges [J]. IEEE communications surveys & tutorials, 2018, 20(1): 7–38. DOI: 10.1109/COMST.2017.2758763 DOI
2	NDIKUMANA A, TRAN N H, HO T M, et al. Joint communication, computation, caching, and control in big data multi-access edge computing [J]. IEEE transactions on mobile computing, 2020, 19(6): 1359–1374. DOI: 10.1109/TMC.2019.2908403 DOI
3	PASCHOS G S, IOSIFIDIS G, TAO M X, et al. The role of caching in future communication systems and networks [J]. IEEE journal on selected areas in communications, 2018, 36(6): 1111–1125. DOI: 10.1109/JSAC.2018.2844939 DOI
4	WEI Y F, YU F R, SONG M, et al. Joint optimization of caching, computing, and radio resources for fog-enabled IoT using natural actor-critic deep reinforcement learning [J]. IEEE Internet of Things journal, 2019, 6(2): 2061–2073. DOI: 10.1109/JIOT.2018.2878435 DOI
5	AHMED M, TRAVERSO S, GIACCONE P, et al. Analyzing the performance of LRU caches under non-stationary traffic patterns [EB/OL]. (2013-01-21)[2023-03-01].
6	JALEEL A, THEOBALD K B, STEELY S C, et al. High performance cache replacement using re-reference interval prediction (RRIP) [C]//37th annual international symposium on computer architecture. ACM, 2010: 60–71. DOI: 10.1145/1815961.1815971 DOI
7	LI L X, XU Y, YIN J Y, et al. Deep reinforcement learning approaches for content caching in cache-enabled D2D networks [J]. IEEE Internet of Things journal, 2020, 7(1): 544–557. DOI: 10.1109/JIOT.2019.2951509 DOI
8	WON D U, KIM H S. A prediction scheme for movie preference rating based on DeepFM model [C]//International Conference on Information Networking (ICOIN). IEEE, 2022: 385–390. DOI: 10.1109/ICOIN53446.2022.9687136 DOI
9	LI D Y, ZHANG H X, DING H, et al. User preference learning-based proactive edge caching for D2D-assisted wireless networks [J]. IEEE Internet of Things journal, 2023, early access. DOI: 10.1109/JIOT.2023.3244621 DOI
10	JIANG Y X, FENG H J, ZHENG F C, et al. Deep learning-based edge caching in fog radio access networks [J]. IEEE transactions on wireless communications, 2020, 19(12): 8442–8454. DOI: 10.1109/TWC.2020.3022907 DOI
11	MCMAHAN H B, MOORE E, RAMAGE D, et al. Communication-efficient learning of deep networks from decentralized data [EB/OL]. (2016-02-17)[2023-03-01].
12	SHI Y M, YANG K, JIANG T, et al. Communication-efficient edge AI: algorithms and systems [J]. IEEE communications surveys & tutorials, 2020, 22(4): 2167–2191. DOI: 10.1109/COMST.2020.3007787 DOI
13	KHAN L U, PANDEY S R, TRAN N H, et al. Federated learning for edge networks: resource optimization and incentive mechanism [J]. IEEE communications magazine, 2020, 58(10): 88–93. DOI: 10.1109/MCOM.001.1900649 DOI
14	LIM W Y B, LUONG N C, HOANG D T, et al. Federated learning in mobile edge networks: a comprehensive survey [J]. IEEE communications surveys & tutorials, 2020, 22(3): 2031–2063. DOI: 10.1109/COMST.2020.2986024 DOI
15	FAROOQ M S, TEHSEEN R, QURESHI J N, et al. FFM: flood forecasting model using federated learning [J]. IEEE access, 2023, 11: 24472–24483. DOI: 10.1109/ACCESS.2023.3252896 DOI
16	WANG K L, DENG N, LI X H. An efficient content popularity prediction of privacy preserving based on federated learning and Wasserstein GAN [J]. IEEE Internet of Things journal, 2023, 10(5): 3786–3798. DOI: 10.1109/JIOT.2022.3176360 DOI
17	HOCHREITER S, SCHMIDHUBER J. Long short-term memory [J]. Neural computation, 1997, 9(8): 1735–1780. DOI: 10.1162/neco.1997.9.8.1735 DOI
18	DOAN K N, VAN NGUYEN T, QUEK T Q S, et al. Content-aware proactive caching for backhaul offloading in cellular network [J]. IEEE transactions on wireless communications, 2018, 17(5): 3128–3140. DOI: 10.1109/TWC.2018.2806971 DOI
19	HARPER F M, KONSTAN J A. The MovieLens datasets [J]. ACM transactions on interactive intelligent systems, 2016, 5(4): 1–19. DOI: 10.1145/2827872 DOI
20	GUO J Q, HE H W, SUN C. ARIMA-based road gradient and vehicle velocity prediction for hybrid electric vehicle energy management [J]. IEEE transactions on vehicular technology, 2019, 68(6): 5309–5320. DOI: 10.1109/TVT.2019.2912893 DOI

Content Popularity Prediction via Federated Learning in Cache-Enabled Wireless Networks

RichHTML

PDF

可视化

摘要/Abstract

引用本文

使用本文

图/表 6

参考文献 20

相关文章 0

编辑推荐

Metrics