HiSTM: Hierarchical Spatiotemporal Mamba for Cellular Traffic Forecasting

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bettouche, Zineddine, Ali, Khalid, Fischer, Andreas, Kassler, Andreas
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912535508156416
author Bettouche, Zineddine
Ali, Khalid
Fischer, Andreas
Kassler, Andreas
author_facet Bettouche, Zineddine
Ali, Khalid
Fischer, Andreas
Kassler, Andreas
contents Cellular traffic forecasting is essential for network planning, resource allocation, or load-balancing traffic across cells. However, accurate forecasting is difficult due to intricate spatial and temporal patterns that exist due to the mobility of users. Existing AI-based traffic forecasting models often trade-off accuracy and computational efficiency. We present Hierarchical SpatioTemporal Mamba (HiSTM), which combines a dual spatial encoder with a Mamba-based temporal module and attention mechanism. HiSTM employs selective state space methods to capture spatial and temporal patterns in network traffic. In our evaluation, we use a real-world dataset to compare HiSTM against several baselines, showing a 29.4% MAE improvement over the STN baseline while using 94% fewer parameters. We show that the HiSTM generalizes well across different datasets and improves in accuracy over longer time-horizons.
format Preprint
id arxiv_https___arxiv_org_abs_2508_09184
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle HiSTM: Hierarchical Spatiotemporal Mamba for Cellular Traffic Forecasting
Bettouche, Zineddine
Ali, Khalid
Fischer, Andreas
Kassler, Andreas
Networking and Internet Architecture
Artificial Intelligence
Cellular traffic forecasting is essential for network planning, resource allocation, or load-balancing traffic across cells. However, accurate forecasting is difficult due to intricate spatial and temporal patterns that exist due to the mobility of users. Existing AI-based traffic forecasting models often trade-off accuracy and computational efficiency. We present Hierarchical SpatioTemporal Mamba (HiSTM), which combines a dual spatial encoder with a Mamba-based temporal module and attention mechanism. HiSTM employs selective state space methods to capture spatial and temporal patterns in network traffic. In our evaluation, we use a real-world dataset to compare HiSTM against several baselines, showing a 29.4% MAE improvement over the STN baseline while using 94% fewer parameters. We show that the HiSTM generalizes well across different datasets and improves in accuracy over longer time-horizons.
title HiSTM: Hierarchical Spatiotemporal Mamba for Cellular Traffic Forecasting
topic Networking and Internet Architecture
Artificial Intelligence
url https://arxiv.org/abs/2508.09184