CacheMamba: Popularity Prediction for Mobile Edge Caching Networks via Selective State Spaces

Mobile Edge Caching (MEC) plays a pivotal role in mitigating latency in data-intensive services by dynamically caching frequently requested content on edge servers. This capability is critical for applications such as Augmented Reality (AR), Virtual Reality (VR), and Autonomous Vehicles (AV), where efficient content caching and accurate popularity prediction are essential for optimizing performance. In this paper, we explore the problem of popularity prediction in MEC by utilizing historical time-series request data of intended files, formulating this problem as a ranking task. To this aim, we propose CacheMamba model by employing Mamba, a state-space model (SSM)-based architecture, to identify the top-K files with the highest likelihood of being requested. We then benchmark the proposed model against a Transformer-based approach, demonstrating its superior performance in terms of cache-hit rate, Mean Average Precision (MAP), Normalized Discounted Cumulative Gain (NDCG), and Floating-Point Operations Per Second (FLOPS), particularly when dealing with longer sequences.

@article{kianfar2025_2502.15746,
  title={ CacheMamba: Popularity Prediction for Mobile Edge Caching Networks via Selective State Spaces },
  author={ Ghazaleh Kianfar and Zohreh Hajiakhondi-Meybodi and Arash Mohammadi },
  journal={arXiv preprint arXiv:2502.15746},
  year={ 2025 }
}