Asynchronous Federated and Reinforcement Learning for Mobility-Aware Edge Caching in IoV
Kai Jiang,Yue Cao,Yujie Song,Huan Zhou,Shaohua Wan,Xu Zhang
DOI: https://doi.org/10.1109/jiot.2023.3349255
IF: 10.6
2024-01-01
IEEE Internet of Things Journal
Abstract:Edge caching is a promising technology to reduce backhaul strain and content access delay in Internet of Vehicles (IoV). It precaches frequently used contents close to vehicles through intermediate roadside units. Previous edge caching works often assume that content popularity is known in advance or obeys simplified models. However, such assumptions are unrealistic, as content popularity varies with uncertain spatial-temporal traffic demands in IoVs. Federated learning (FL) enables vehicles to predict popular content with distributed training. It preserves the training data remain local, thereby addressing privacy concerns and communication resource shortages. This article investigates a mobility-aware edge caching strategy by exploiting asynchronous FL and deep reinforcement learning (DRL). We first implement a novel asynchronous FL framework for local updates and global aggregation of stacked autoencoder (SAE) models. Then, utilizing the latent features extracted by the trained SAE model, we adopt a hybrid filtering model for predicting and recommending popular content. Furthermore, we explore intelligent caching decisions after content prediction. Based on the formulated Markov decision process (MDP) problem, we propose a DRL-based solution, and adopt neural network-based parameter approximations for the curse of dimensionality in RL. Extensive simulations are conducted based on real-world data trajectory. Especially, our proposed method outperforms federated averaging, least recently used, and NoDRL, and the edge hit rate is improved by roughly 6%, 21%, and 15%, respectively, when the cache capacity reaches 350 MB.