Content popularity prediction based on quantized federated Bayesian learning in fog radio access networks

• Main Authors: Tao, Yunwei, Jiang, Yanxiang, Zheng, Fu-Chun, Wang, Zhiheng, Zhu, Pengcheng, Tao, Meixia, Niyato, Dusit, You, Xiaohu
• Other Authors: School of Computer Science and Engineering
• Format: Article
• Language: English
• Published: 2023
• Subjects: Engineering::Computer science and engineering; Bayesian Learning; Federated Learning
• Online Access: https://hdl.handle.net/10356/172071
• Institution: Nanyang Technological University

In this paper, we investigate the content popularity prediction problem in cache-enabled fog radio access networks (F-RANs). In order to predict the content popularity with high accuracy and low complexity, we propose a Gaussian process based regressor to model the content request pattern. Firstly, the relationship between content features and popularity is captured by our proposed model. Then, we utilize Bayesian learning to train the model parameters, which is robust to overfitting. However, Bayesian methods are usually unable to find a closed-form expression of the posterior distribution. To tackle this issue, we apply a stochastic variance reduced gradient Hamiltonian Monte Carlo (SVRG-HMC) method to approximate the posterior distribution. To utilize the computing resource of fog access points (F-APs) and also reduce the communication overhead, we propose a quantized federated learning (FL) framework combining with Bayesian learning. The proposed quantized federated Bayesian learning framework allows each F-AP to send gradients to the cloud server after quantizing and encoding. It can achieve a tradeoff between prediction accuracy and communication overhead effectively. Simulation results show that the performance of our proposed policy outperforms the considered baseline policies.