mlsanityData to 14 Oct 2026

Cellular Traffic Prediction via Deep State Space Models with Attention Mechanism

arXiv 2506.15688 ↗First posted 26 May 2025cs.LG cs.AIcited by 3

Not on the map. Why?

Abstract (arXiv)

Cellular traffic prediction is of great importance for operators to manage network resources and make decisions. Traffic is highly dynamic and influenced by many exogenous factors, which would lead to the degradation of traffic prediction accuracy. This paper proposes an end-to-end framework with two variants to explicitly characterize the spatiotemporal patterns of cellular traffic among neighboring cells. It uses convolutional neural networks with an attention mechanism to capture the spatial dynamics and Kalman filter for temporal modelling. Besides, we can fully exploit the auxiliary information such as social activities to improve prediction performance. We conduct extensive experiments on three real-world datasets. The results show that our proposed models outperform the state-of-the-art machine learning techniques in terms of prediction accuracy.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 16 of its 74 references that are in the map. First posted on arXiv 26 May 2025.

Builds on

16 of 74 references.

References · 16

  1. 2017Attention Is All You Needcited by 60,361
  2. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  3. 2017Modeling Long- and Short-Term Temporal Patterns with Deep Neural Networkscited by 879
  4. 2017A Dual-Stage Attention-Based Recurrent Neural Network for Time Series Predictioncited by 245
  5. 2019Deep Learning for Spatio-Temporal Data Mining: A Surveycited by 172
  6. 2021KalmanNet: Neural Network Aided Kalman Filtering for Partially Known Dynamicscited by 132
  7. 2019Recurrent Kalman Networks: Factorized Inference in High-Dimensional Deep Feature Spacescited by 67
  8. 2017Long Short-Term Memory Kalman Filters:Recurrent Neural Estimators for Pose Regularizationcited by 36
  9. 2021Graph-based Deep Learning for Communication Networks: A Surveycited by 21
  10. 2018Active Learning for Wireless IoT Intrusion Detectioncited by 19
  11. 2019Towards Better Forecasting by Fusing Near and Distant Future Visionscited by 14
  12. 2019DynaNet: Neural Kalman Dynamical Model for Motion Estimation and Predictioncited by 13
4 more
  1. 2020From Twitter to Traffic Predictor: Next-Day Morning Traffic Prediction Using Social Media Datacited by 10
  2. 2022Entry-Flipped Transformer for Inference and Prediction of Participant Behaviorcited by 7
  3. 2021Interpretable Deep Feature Propagation for Early Action Recognitioncited by 5
  4. 2019DeepAuto: A Hierarchical Deep Learning Framework for Real-Time Prediction in Cellular Networkscited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 1; 2026: 2.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026FAST: A Synergistic Framework of Attention and State-space Models for Spatiotemporal Traffic Prediction
  2. 2025HiSTM: Hierarchical Spatiotemporal Mamba for Cellular Traffic Forecasting
  3. 2026Spatial PDE-aware Selective State-space with Nested Memory for Mobile Traffic Grid Forecasting
  4. 2026Spatiotemporal Graph Transformer for Traffic Intelligence in Edge Computing
  5. 2025DKGCM: A Spatio-Temporal Prediction Model for Traffic Flow by Fusing Spatial Node Clustering Method and Fourier Bidirectional Mamba Mechanism
  6. 2024Knowledge Distillation on Spatial-Temporal Graph Convolutional Network for Traffic Prediction
  7. 2026GAMMA-Net: Adaptive Long-Horizon Traffic Spatio-Temporal Forecasting Model based on Interleaved Graph Attention and Multi-Axis Mamba
  8. 2025Efficient Traffic Prediction Through Spatio-Temporal Distillation