HSTMixer: A Hierarchical MLP-Mixer for Large-Scale Traffic Forecasting

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Wang, Yongyao, Wang, Jingyuan, Yu, Xie, Ji, Jiahao, Li, Chao
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915661801848832
author Wang, Yongyao
Wang, Jingyuan
Yu, Xie
Ji, Jiahao
Li, Chao
author_facet Wang, Yongyao
Wang, Jingyuan
Yu, Xie
Ji, Jiahao
Li, Chao
contents Traffic forecasting task is significant to modern urban management. Recently, there is growing attention on large-scale forecasting, as it better reflects the complexity of real-world traffic networks. However, existing models often exhibit quadratic computational complexity, making them impractical for large-scale real-world scenarios. In this paper, we propose a novel framework, Hierarchical Spatio-Temporal Mixer (HSTMixer), which leverages an all-MLP architecture for efficient and effective large-scale traffic forecasting. HSTMixer employs a hierarchical spatiotemporal mixing block to extract multi-resolution features through bottom-up aggregation and top-down propagation. Furthermore, an adaptive region mixer generates transformation matrices based on regional semantics, enabling our model to dynamically capture evolving spatiotemporal patterns for different regions. Extensive experiments conducted on four large-scale real-world datasets demonstrate that the proposed method not only achieves state-of-the-art performance but also exhibits competitive computational efficiency.
format Preprint
id arxiv_https___arxiv_org_abs_2512_07854
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle HSTMixer: A Hierarchical MLP-Mixer for Large-Scale Traffic Forecasting
Wang, Yongyao
Wang, Jingyuan
Yu, Xie
Ji, Jiahao
Li, Chao
Machine Learning
Traffic forecasting task is significant to modern urban management. Recently, there is growing attention on large-scale forecasting, as it better reflects the complexity of real-world traffic networks. However, existing models often exhibit quadratic computational complexity, making them impractical for large-scale real-world scenarios. In this paper, we propose a novel framework, Hierarchical Spatio-Temporal Mixer (HSTMixer), which leverages an all-MLP architecture for efficient and effective large-scale traffic forecasting. HSTMixer employs a hierarchical spatiotemporal mixing block to extract multi-resolution features through bottom-up aggregation and top-down propagation. Furthermore, an adaptive region mixer generates transformation matrices based on regional semantics, enabling our model to dynamically capture evolving spatiotemporal patterns for different regions. Extensive experiments conducted on four large-scale real-world datasets demonstrate that the proposed method not only achieves state-of-the-art performance but also exhibits competitive computational efficiency.
title HSTMixer: A Hierarchical MLP-Mixer for Large-Scale Traffic Forecasting
topic Machine Learning
url https://arxiv.org/abs/2512.07854