Resilient Collaborative Caching for Multi-Edge Systems With Robust Federated Deep Learning

Chen, Zheyi and Liang, Jie and Yu, Zhengxin and Cheng, Hongju and Min, Geyong and Li, Jie (2024) Resilient Collaborative Caching for Multi-Edge Systems With Robust Federated Deep Learning. IEEE/ACM Transactions on Networking. pp. 1-16. ISSN 1063-6692

Text (Manuscript_TNET-2024-00196)
Manuscript_TNET-2024-00196.pdf - Accepted Version
Available under License Creative Commons Attribution.
Download (9MB)

Abstract

As a key technique for future networks, the performance of emerging multi-edge caching is often limited by inefficient collaboration among edge nodes and improper resource configuration. Meanwhile, achieving optimal cache hit rates poses substantive challenges without effectively capturing the potential relations between discrete user features and diverse content libraries. These challenges become further sophisticated when caching schemes are exposed to adversarial attacks that seriously impair cache performance. To address these challenges, we introduce RoCoCache , a resilient collaborative caching framework that uniquely integrates robust federated deep learning with proactive caching strategies, enhancing performance under adversarial conditions. First, we design a novel partitioning mechanism for multi-dimensional cache space, enabling precise content recommendations in user classification intervals. Next, we develop a new Discrete-Categorical Variational Auto-Encoder (DC-VAE) to accurately predict content popularity by overcoming posterior collapse. Finally, we create an original training mode and proactive cache replacement strategy based on robust federated deep learning. Notably, the residual-based detection for adversarial model updates and similarity-based federated aggregation are integrated to avoid the model destruction caused by adversarial updates, which enables the proactive cache replacement adapting to optimized cache resources and thus enhances cache performance. Using the real-world testbed and datasets, extensive experiments verify that the RoCoCache achieves higher cache hit rates and efficiency than state-of-the-art methods while ensuring better robustness. Moreover, we validate the effectiveness of the components designed in RoCoCache for improving cache performance via ablation studies.