Dynamic TMoE: A Drift-Aware Dynamic Mixture of Experts Framework for Non-Stationary Time Series Forecasting

Jiawen Zhu, Shuhan Liu, Di Weng, Yingcai Wu
5/20/2026
cs.LGcs.AI

Official code: andone-07/Dynamic-TMoE (14 stars, last commit 2026-05-13).

Abstract

Non-stationary time series forecasting is challenged by evolving distribution shifts that static models struggle to capture. While Mixture-of-Experts (MoE) architectures offer a promising paradigm for decoupling complex drift patterns, existing approaches are limited by fixed expert pools and memoryless routing, hampering their ability to adapt to abrupt regime shifts. To address this, we propose Dynamic TMoE, a framework that unifies architectural evolution with temporal continuity during learning phase. By detecting distribution shifts via Maximum Mean Discrepancy (MMD), we dynamically instantiate heterogeneous experts and prune redundant ones to optimize capacity. Additionally, a temporal memory router leverages recurrent states and an anomaly repository to ensure stable, context-aware expert selection without requiring test-time updates. Experiments on nine benchmarks demonstrate state-of-the-art performance, reducing MSE by 10.4% and MAE by 7.8%. Code is available at https://github.com/andone-07/Dynamic-TMoE.

View on arXivView PDF

Code Implementations(1)

This repository contains the official implementation of Dynamic TMoE, accepted as a poster at ICML 2026.

141Shell, PythonMay 13, 20265 months ago

Similar Papers

Cite this paper

@article{zhu2026dynamic,
  title  = {Dynamic TMoE: A Drift-Aware Dynamic Mixture of Experts Framework for Non-Stationary Time Series Forecasting},
  author = {Jiawen Zhu and Shuhan Liu and Di Weng and Yingcai Wu},
  year   = {2026},
  eprint = {2605.20678},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2605.20678v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2605.20678.svg)](https://codeofpaper.com/paper/2605.20678)

Discussion