STM3: Mixture of Multiscale Mamba for Long-Term Spatio-Temporal Time-Series Prediction
Pith reviewed 2026-05-25 07:33 UTC · model grok-4.3
The pith
STM3 integrates multiscale Mamba into a disentangled mixture-of-experts framework to capture long-term spatio-temporal dependencies efficiently.
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
STM3 integrates a Multiscale Mamba architecture within a novel Disentangled Mixture-of-Experts (DMoE) framework to capture diverse multiscale information efficiently, while utilizing an adaptive graph causal network to model complex spatial dependencies. To ensure robust representation learning, it introduces a stable routing strategy and a causal contrastive learning strategy, which work in tandem with hierarchical information aggregation to guarantee scale distinguishability. The model theoretically proves that it achieves superior routing smoothness and guarantees pattern disentanglement for each expert.
What carries the argument
The Disentangled Mixture-of-Experts (DMoE) framework with Multiscale Mamba, stable routing, and causal contrastive learning to extract and disentangle multiscale temporal patterns across nodes.
If this is right
- Achieves state-of-the-art results across 10 real-world benchmarks in long-term spatio-temporal time-series prediction.
- On the PEMSD8 dataset, surpasses the second-best model by 7.1% in MAE, 8.5% in RMSE, and 15.9% in MAPE.
- Provides theoretical guarantees of superior routing smoothness and pattern disentanglement for each expert.
- Enables efficient extraction of multiscale information while modeling spatial dependencies via an adaptive graph causal network.
Where Pith is reading between the lines
- If the DMoE successfully disentangles scales, the architecture could transfer to other long-sequence domains like video or audio forecasting with minimal changes.
- The combination of Mamba and mixture routing may reduce quadratic attention costs for very long horizons compared to transformer baselines.
- Causal contrastive learning as a regularizer might generalize to other mixture-of-experts time-series models to improve expert specialization.
Load-bearing premise
That the multiscale temporal information from different nodes is highly correlated in a manner that DMoE routing plus causal contrastive learning can resolve without introducing new fitting artifacts.
What would settle it
A new long-term spatio-temporal dataset where STM3 fails to match or exceed the second-best model's MAE, RMSE, and MAPE, or where empirical routing fails to exhibit the claimed smoothness and expert disentanglement.
Figures
read the original abstract
Recently, spatio-temporal time-series prediction has developed rapidly, yet existing deep learning methods struggle with learning complex long-term spatio-temporal dependencies efficiently. The long-term spatio-temporal dependency learning brings two new challenges: 1) The long-term temporal sequence naturally includes multiscale information, which is hard to extract efficiently; 2) The multiscale temporal information from different nodes is highly correlated and hard to model. To address these challenges, we propose Spatio-Temporal Mixture of Multiscale Mamba (STM3). STM3 integrates a Multiscale Mamba architecture within a novel Disentangled Mixture-of-Experts (DMoE) framework to capture diverse multiscale information efficiently, while utilizing an adaptive graph causal network to model complex spatial dependencies. To ensure robust representation learning, we introduce a stable routing strategy and a causal contrastive learning strategy, which work in tandem with hierarchical information aggregation to guarantee scale distinguishability. We theoretically prove that STM3 achieves superior routing smoothness and guarantees pattern disentanglement for each expert. Extensive experiments on 10 real-world benchmarks across domains demonstrate STM3's superior performance, achieving state-of-the-art results in long-term spatio-temporal time-series prediction. Notably, on the PEMSD8 dataset, it achieves significant improvements, surpassing the second-best model by 7.1% in MAE, 8.5% in RMSE, and 15.9% in MAPE. Code is available at https://github.com/IfReasonable/STM3_KDD26.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The manuscript proposes STM3 (Spatio-Temporal Mixture of Multiscale Mamba), which combines a Multiscale Mamba architecture inside a Disentangled Mixture-of-Experts (DMoE) framework, an adaptive graph causal network, stable routing, and causal contrastive learning with hierarchical aggregation. It targets two challenges in long-term spatio-temporal time-series prediction: efficient extraction of multiscale temporal information and modeling of highly correlated multiscale information across nodes. The paper asserts theoretical proofs that STM3 achieves superior routing smoothness and guarantees pattern disentanglement for each expert. It reports state-of-the-art empirical results on 10 real-world benchmarks across domains, with specific gains on PEMSD8 (surpassing the second-best model by 7.1% MAE, 8.5% RMSE, and 15.9% MAPE). Public code is provided at the cited GitHub repository.
Significance. If the empirical results and theoretical claims hold under standard verification, the work would be significant for the spatio-temporal forecasting community by offering an efficient Mamba-based approach to long-horizon dependencies that also handles spatial correlations via adaptive graphs. The public code release is a clear strength that enables direct reproducibility and extension.
major comments (3)
- [Abstract] Abstract: the manuscript asserts 'theoretical proofs' of superior routing smoothness and pattern disentanglement, yet supplies no derivation details, lemmas, or proof sketches; this is load-bearing for the novelty claim because the central methodological contribution rests on these guarantees.
- [Abstract] Abstract (performance claims): concrete percentage improvements (e.g., PEMSD8 MAE/RMSE/MAPE) are reported without accompanying error bars, baseline tables, data-split descriptions, or statistical testing; the SOTA assertion on 10 benchmarks therefore rests on uninspectable evidence.
- [Abstract (challenges and proposed solution paragraph)] The weakest assumption (multiscale temporal sequences contain extractable, highly correlated information that DMoE routing plus causal contrastive learning resolves without new fitting artifacts) is stated but not subjected to an ablation that isolates whether the observed gains arise from the proposed mechanisms versus standard scaling or regularization effects.
minor comments (1)
- [Abstract] The abstract mentions 'hierarchical information aggregation' without defining the aggregation operator or its relation to the DMoE experts.
Simulated Author's Rebuttal
We thank the referee for the constructive comments, which help clarify the presentation of our contributions. We respond to each major comment below and indicate planned revisions to the manuscript.
read point-by-point responses
-
Referee: [Abstract] Abstract: the manuscript asserts 'theoretical proofs' of superior routing smoothness and pattern disentanglement, yet supplies no derivation details, lemmas, or proof sketches; this is load-bearing for the novelty claim because the central methodological contribution rests on these guarantees.
Authors: The abstract summarizes the contribution; the full proofs appear in Section 3.4, including Lemma 3.1 (routing smoothness via bounded Lipschitz constants on the gating function) and Theorem 3.2 (pattern disentanglement via mutual information bounds under causal contrastive loss), with complete derivations and proof sketches. We will revise the abstract to explicitly reference Section 3.4 and include a one-sentence proof outline if space allows. revision: yes
-
Referee: [Abstract] Abstract (performance claims): concrete percentage improvements (e.g., PEMSD8 MAE/RMSE/MAPE) are reported without accompanying error bars, baseline tables, data-split descriptions, or statistical testing; the SOTA assertion on 10 benchmarks therefore rests on uninspectable evidence.
Authors: The abstract condenses results; full experimental details, including error bars (std over 5 random seeds), complete baseline tables, data-split protocols, and paired t-test p-values, are reported in Section 4.2 and Tables 1–3. The PEMSD8 gains are computed from those tables. We will add a parenthetical reference in the abstract directing readers to the experimental section. revision: partial
-
Referee: [Abstract (challenges and proposed solution paragraph)] The weakest assumption (multiscale temporal sequences contain extractable, highly correlated information that DMoE routing plus causal contrastive learning resolves without new fitting artifacts) is stated but not subjected to an ablation that isolates whether the observed gains arise from the proposed mechanisms versus standard scaling or regularization effects.
Authors: Section 4.4 already contains component-wise ablations (removing DMoE, contrastive loss, and stable routing individually) that demonstrate gains exceed those from simple scaling or L2 regularization. To further isolate against fitting artifacts, we will add a controlled comparison against a capacity-matched vanilla Mamba baseline with equivalent parameter count and training regime. revision: yes
Circularity Check
No significant circularity identified
full rationale
The paper's central claims consist of empirical SOTA results on 10 held-out real-world benchmarks (with explicit gains reported on PEMSD8) plus a theoretical argument for routing smoothness and pattern disentanglement derived from the DMoE architecture, stable routing, and causal contrastive learning. These results are obtained from standard train/test splits on external datasets rather than quantities defined by fitted parameters inside the model equations. No self-definitional steps, fitted-input predictions, or load-bearing self-citations appear in the provided text; the derivation chain remains self-contained against external benchmarks and publicly released code.
Axiom & Free-Parameter Ledger
axioms (1)
- domain assumption Standard deep-learning assumptions that neural networks with the stated components can represent the target spatio-temporal functions and that benchmark datasets are representative of the intended use cases.
Forward citations
Cited by 1 Pith paper
-
PIMSM: Physics-Informed Multi-Scale Mamba for Stable Neural Representations under Distribution Shift
PIMSM is a Mamba-based architecture that maps knee frequencies from spectra to multi-scale discretization parameters to reduce representation drift under distribution shifts in fMRI and weather forecasting.
Reference graph
Works this paper leans on
-
[1]
Khaled Alkilane, Yihang He, and Der-Horng Lee. 2024. MixMamba: Time series modeling with adaptive expertise.Information Fusion112 (2024), 102589
work page 2024
-
[2]
Lei Bai, Lina Yao, Can Li, Xianzhi Wang, and Can Wang. 2020. Adaptive graph convolutional recurrent network for traffic forecasting.Advances in neural information processing systems33 (2020), 17804–17815
work page 2020
-
[3]
Gianni Barlacchi, Marco De Nadai, Roberto Larcher, Antonio Casella, Cristiana Chitic, Giovanni Torrisi, Fabrizio Antonelli, Alessandro Vespignani, Alex Pent- land, and Bruno Lepri. 2015. A multi-source dataset of urban life in the city of Milan and the Province of Trentino.Scientific data2, 1 (2015), 1–15
work page 2015
- [4]
-
[5]
Chao Chen, Karl Petty, Alexander Skabardonis, Pravin Varaiya, and Zhanfeng Jia. 2001. Freeway performance measurement system: mining loop detector data. Transportation research record1748, 1 (2001), 96–102
work page 2001
-
[6]
Min Chen, Guansong Pang, Wenjun Wang, and Cheng Yan. 2025. Information Bottleneck-guided MLPs for Robust Spatial-temporal Forecasting. InForty-second International Conference on Machine Learning
work page 2025
-
[7]
Peng Chen, Yingying ZHANG, Yunyao Cheng, Yang Shu, Yihang Wang, Qingsong Wen, Bin Yang, and Chenjuan Guo. 2023. Pathformer: Multi-scale Transformers with Adaptive Pathways for Time Series Forecasting. InThe Twelfth International Conference on Learning Representations
work page 2023
- [8]
-
[9]
Jeongwhan Choi, Hwangyong Choi, Jeehyun Hwang, and Noseong Park. 2022. Graph neural controlled differential equations for traffic forecasting. InProceed- ings of the AAAI conference on artificial intelligence, Vol. 36. 6367–6374
work page 2022
- [10]
-
[11]
Damai Dai, Chengqi Deng, Chenggang Zhao, Rx Xu, Huazuo Gao, Deli Chen, Jiashi Li, Wangding Zeng, Xingkai Yu, Y Wu, et al . 2024. DeepSeekMoE: To- wards Ultimate Expert Specialization in Mixture-of-Experts Language Models. InProceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 1280–1297
work page 2024
-
[12]
Tri Dao, Daniel Y Fu, Khaled K Saab, Armin W Thomas, Atri Rudra, and Christo- pher Ré. 2023. Hungry Hungry Hippos: Towards Language Modeling with State Space Models. InProceedings of the 11th International Conference on Learning Representations (ICLR)
work page 2023
-
[13]
Nan Du, Yanping Huang, Andrew M Dai, Simon Tong, Dmitry Lepikhin, Yuanzhong Xu, Maxim Krikun, Yanqi Zhou, Adams Wei Yu, Orhan Firat, et al
-
[14]
In International conference on machine learning
Glam: Efficient scaling of language models with mixture-of-experts. In International conference on machine learning. PMLR, 5547–5569
-
[15]
Yuchen Fang, Yanjun Qin, Haiyong Luo, Fang Zhao, Bingbing Xu, Liang Zeng, and Chenxing Wang. 2023. When spatio-temporal meet wavelets: Disentangled traffic forecasting via efficient spectral graph attention networks. In2023 IEEE 39th international conference on data engineering (ICDE). IEEE, 517–529
work page 2023
-
[16]
William Fedus, Barret Zoph, and Noam Shazeer. 2022. Switch transformers: Scaling to trillion parameter models with simple and efficient sparsity.Journal of Machine Learning Research23, 120 (2022), 1–39
work page 2022
-
[17]
Albert Gu and Tri Dao. 2024. Mamba: Linear-Time Sequence Modeling with Selective State Spaces. InFirst Conference on Language Modeling
work page 2024
-
[18]
Albert Gu, Karan Goel, and Christopher Ré. 2021. Efficiently modeling long sequences with structured state spaces.arXiv preprint arXiv:2111.00396(2021)
work page internal anchor Pith review Pith/arXiv arXiv 2021
-
[19]
Shengnan Guo, Youfang Lin, Ning Feng, Chao Song, and Huaiyu Wan. 2019. Attention based spatial-temporal graph convolutional networks for traffic flow forecasting. InProceedings of the AAAI conference on artificial intelligence, Vol. 33. 922–929
work page 2019
-
[20]
Robert A Jacobs, Michael I Jordan, Steven J Nowlan, and Geoffrey E Hinton. 1991. Adaptive mixtures of local experts.Neural computation3, 1 (1991), 79–87
work page 1991
-
[21]
Albert Q Jiang, Alexandre Sablayrolles, Antoine Roux, Arthur Mensch, Blanche Savary, Chris Bamford, Devendra Singh Chaplot, Diego de las Casas, Emma Bou Hanna, Florian Bressand, et al . 2024. Mixtral of experts.arXiv preprint arXiv:2401.04088(2024)
work page internal anchor Pith review Pith/arXiv arXiv 2024
-
[22]
Dmitry Lepikhin, HyoukJoong Lee, Yuanzhong Xu, Dehao Chen, Orhan Firat, Yanping Huang, Maxim Krikun, Noam Shazeer, and Zhifeng Chen. 2020. Gshard: Scaling giant models with conditional computation and automatic sharding.arXiv preprint arXiv:2006.16668(2020)
work page internal anchor Pith review Pith/arXiv arXiv 2020
- [23]
- [24]
- [25]
-
[26]
Yaguang Li, Rose Yu, Cyrus Shahabi, and Yan Liu. 2018. Diffusion Convolutional Recurrent Neural Network: Data-Driven Traffic Forecasting. InInternational Conference on Learning Representations
work page 2018
-
[27]
Zhonghang Li, Lianghao Xia, Yong Xu, and Chao Huang. 2023. GPT-ST: genera- tive pre-training of spatio-temporal graph neural networks.Advances in neural information processing systems36 (2023), 70229–70246
work page 2023
-
[28]
Dachuan Liu, Jin Wang, Shuo Shang, and Peng Han. 2022. Msdr: Multi-step dependency relation networks for spatial temporal forecasting. InProceedings of the 28th ACM SIGKDD conference on knowledge discovery and data mining. 1042–1050
work page 2022
-
[29]
Hangchen Liu, Zheng Dong, Renhe Jiang, Jiewen Deng, Jinliang Deng, Quan- jun Chen, and Xuan Song. 2023. Spatio-temporal adaptive embedding makes vanilla transformer sota for traffic forecasting. InProceedings of the 32nd ACM international conference on information and knowledge management. 4125–4129
work page 2023
-
[30]
Shang Liu, Miao He, Zhiqiang Wu, Peng Lu, and Weixi Gu. 2024. Spatial–temporal graph neural network traffic prediction based load balancing with reinforcement learning in cellular networks.Information Fusion103 (2024), 102079
work page 2024
-
[31]
Ali Mehrabian, Shahab Bahrami, and Vincent WS Wong. 2023. A dynamic Bernstein graph recurrent network for wireless cellular traffic prediction. InICC 2023-IEEE International Conference on Communications. IEEE, 3842–3847
work page 2023
-
[32]
Ali Mehrabian and Vincent WS Wong. 2025. A-Gamba: An Adaptive Graph- Mamba Model for Traffic Prediction in Wireless Cellular Networks.IEEE Wireless Communications Letters(2025)
work page 2025
-
[33]
Harsh Mehta, Ankit Gupta, Ashok Cutkosky, and Behnam Neyshabur. 2023. Long Range Language Modeling via Gated State Spaces. InInternational Conference on Learning Representations
work page 2023
- [34]
- [35]
-
[36]
Mohammad Amin Shabani, Amir H Abdi, Lili Meng, and Tristan Sylvain. [n. d.]. Scaleformer: Iterative Multi-scale Refining Transformers for Time Series Fore- casting. InThe Eleventh International Conference on Learning Representations
- [37]
- [38]
-
[39]
Jimmy TH Smith, Andrew Warrington, and Scott W Linderman. 2023. Simplified State Space Layers for Sequence Modeling. InICLR
work page 2023
-
[40]
Chao Song, Youfang Lin, Shengnan Guo, and Huaiyu Wan. 2020. Spatial- temporal synchronous graph convolutional networks: A new framework for spatial-temporal network data forecasting. InProceedings of the AAAI conference on artificial intelligence, Vol. 34. 914–921
work page 2020
-
[41]
Laurens Van der Maaten and Geoffrey Hinton. 2008. Visualizing data using t-SNE. Journal of machine learning research9, 11 (2008)
work page 2008
-
[42]
Shuo Wang, Yanran Li, Jiang Zhang, Qingye Meng, Lingwei Meng, and Fei Gao
-
[43]
5-gnn: A domain knowledge enhanced graph neural network for pm2
Pm2. 5-gnn: A domain knowledge enhanced graph neural network for pm2. 5 forecasting. InProceedings of the 28th international conference on advances in geographic information systems. 163–166
-
[44]
Tan Wang, Zhongqi Yue, Jianqiang Huang, Qianru Sun, and Hanwang Zhang
-
[45]
Advances in Neural Information Processing Systems34 (2021), 18225–18240
Self-supervised learning disentangled group representation as feature. Advances in Neural Information Processing Systems34 (2021), 18225–18240
work page 2021
-
[46]
Yuankai Wu, Dingyi Zhuang, Aurelie Labbe, and Lijun Sun. 2021. Inductive graph neural networks for spatiotemporal kriging. InProceedings of the AAAI conference on artificial intelligence, Vol. 35. 4478–4485
work page 2021
-
[47]
Zonghan Wu, Shirui Pan, Guodong Long, Jing Jiang, Xiaojun Chang, and Chengqi Zhang. 2020. Connecting the dots: Multivariate time series forecasting with graph neural networks. InProceedings of the 26th ACM SIGKDD international conference on knowledge discovery & data mining. 753–763
work page 2020
-
[48]
Zonghan Wu, Shirui Pan, Guodong Long, Jing Jiang, and Chengqi Zhang. 2019. Graph wavenet for deep spatial-temporal graph modeling.arXiv preprint arXiv:1906.00121(2019)
work page internal anchor Pith review Pith/arXiv arXiv 2019
- [49]
-
[50]
Yang Yao, Bo Gu, Zhou Su, and Mohsen Guizani. 2021. MVSTGN: A multi-view spatial-temporal graph network for cellular traffic prediction.IEEE Transactions on Mobile Computing22, 5 (2021), 2837–2849
work page 2021
-
[51]
Bing Yu, Haoteng Yin, and Zhanxing Zhu. 2017. Spatio-temporal graph con- volutional networks: A deep learning framework for traffic forecasting.arXiv preprint arXiv:1709.04875(2017)
work page internal anchor Pith review Pith/arXiv arXiv 2017
-
[52]
Haonan Yuan, Qingyun Sun, Zhaonan Wang, Xingcheng Fu, Cheng Ji, Yongjian Wang, Bo Jin, and Jianxin Li. 2025. DG-Mamba: Robust and Efficient Dynamic Graph Structure Learning with Selective State Space Models. InProceedings of the AAAI Conference on Artificial Intelligence, Vol. 39. 22272–22280
work page 2025
-
[53]
Zijian Zhang, Ze Huang, Zhiwei Hu, Xiangyu Zhao, Wanyu Wang, Zitao Liu, Junbo Zhang, S Joe Qin, and Hongwei Zhao. 2023. Mlpst: Mlp is all you need for spatio-temporal prediction. InProceedings of the 32nd ACM International Conference on Information and Knowledge Management. 3381–3390
work page 2023
-
[54]
Zijian Zhang, Xiangyu Zhao, Qidong Liu, Chunxu Zhang, Qian Ma, Wanyu Wang, Hongwei Zhao, Yiqi Wang, and Zitao Liu. 2023. Promptst: Prompt-enhanced spatio-temporal multi-attribute prediction. InProceedings of the 32nd ACM Inter- national Conference on Information and Knowledge Management. 3195–3205
work page 2023
-
[55]
Ling Zhao, Yujiao Song, Chao Zhang, Yu Liu, Pu Wang, Tao Lin, Min Deng, and Haifeng Li. 2019. T-GCN: A temporal graph convolutional network for traffic prediction.IEEE transactions on intelligent transportation systems21, 9 (2019), 3848–3858
work page 2019
-
[56]
torch.cuda.max_memory_allocated()
Barret Zoph. 2022. Designing effective sparse expert models. In2022 IEEE In- ternational Parallel and Distributed Processing Symposium Workshops (IPDPSW). IEEE, 1044–1044. A More Related Work State Space Models.SSMs have demonstrated exceptional capa- bility in modeling sequential dependencies via state space. The structured state-space sequence model (S4...
work page 2022
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.