EAPS: Edge-Assisted Privacy-Preserving Federated Prediction Systems

To reduce the delay and network congestion for content delivery in wireless networks, proactive caching scheme has attracted lots of attentions from both academia and industry. However, traditional caching prediction methods require to collect user data in a centralized server, which is becoming unreliable and impractical due to regulatory restrictions. To circumvent this issue, deploying caching prediction system in a federated learning (FL) fashion becomes a promising solution. However, there still exist privacy risks, and even worse, the FL is vulnerable to low-cost attacks. To solve this problem, a novel federated prediction system (FPS) is studied to provide high robustness and privacy. Firstly, to keep a balance between further enhancing privacy protection and alleviating the performance degradation caused by additional protection schemes, we propose an edge-assisted, robust and privacy-preserving FPS framework based on the local differential privacy (LDP) scheme. Secondly, to mitigate the impact of heterogeneous data, we add a regularization term to the local loss function. Furthermore, an attention-based aggregation scheme is proposed to defend against Byzantine attacks during the training process. Finally, the experiment results are provided to show the superiority of our proposed algorithm in terms of prediction accuracy and robustness.