DeRegiME: Deep Regime Mixtures for Probabilistic Forecasting under Distribution Shift
On the map: Mixture-of-experts → MoE for time series and graphs
Abstract (arXiv)
We introduce DeRegiME -- Deep Regime Mixture of Experts -- a direct multi-horizon probabilistic forecaster that separates latent uncertainty regimes from the underlying signal and softly assigns each forecast location to learned recurring regimes using a sparse variational Gaussian process (GP) whose nonstationary regime-mixing kernel and Student-t likelihood combine per-regime sub-kernels and noise processes via a shared gate. This yields a single sparse-GP posterior, not a mixture of GP experts. DeRegiME addresses a key limitation of neural forecasters: point forecasts discard residual uncertainty, and probabilistic heads -- whether single marginals, uninterpreted mixtures, quantile sets, or diffusion samples -- rarely expose the regime structure of the residual. Yet distribution shift in noisy heteroskedastic time series may be abrupt, gradual, or horizon-dependent and often appears in residual uncertainty rather than the conditional mean. DeRegiME yields an interpretable mean-residual-noise decomposition with a direct-sum feature-space representation that anchors regimes as clusters of residual similarity whose transitions surface as implicit changepoints. The effective number of regimes is pruned by the stick-breaking gate. We prove kernel validity and predictive-density propriety, and across ten benchmarks and three encoder grids DeRegiME improves negative log predictive density (NLPD) by 20.3% over the strongest encoder-matched baseline, a DeepAR/GluonTS-style dynamic Student-t head, with parallel gains on CRPS (3.0%) and MSE (4.7%). Improvements are consistent across all datasets, which span abrupt, gradual, and seasonal shifts.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in MoE for time series and graphs, one of 12 directions in Mixture-of-experts. Builds on 24 of its 45 references that are in the map. First posted on arXiv 19 May 2026.
Builds on
24 of 45 references.
References · 24
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2020Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecastingcited by 2,141
- 2022A Time Series is Worth 64 Words: Long-term Forecasting with Transformerscited by 1,981
- 2021Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecastingcited by 1,654
- 2022Are Transformers Effective for Time Series Forecasting?cited by 1,400
- 2017DeepAR: Probabilistic Forecasting with Autoregressive Recurrent Networkscited by 1,004
- 2023iTransformer: Inverted Transformers Are Effective for Time Series Forecastingcited by 906
- 2019Temporal Fusion Transformers for Interpretable Multi-horizon Time Series Forecastingcited by 748
- 2024Chronos: Learning the Language of Time Seriescited by 712
- 2023A decoder-only foundation model for time-series forecastingcited by 650
- 2024Unified Training of Universal Time Series Forecasting Transformerscited by 588
12 more
- 2024MOMENT: A Family of Open Time-series Foundation Modelscited by 453
- 2015Deep Kernel Learningcited by 440
- 2018Conditional Neural Processescited by 421
- 2013Gaussian Processes for Big Datacited by 377
- 2024TimeMixer: Decomposable Multiscale Mixing for Time Series Forecastingcited by 333
- 2021Autoregressive Denoising Diffusion Models for Multivariate Probabilistic Time Series Forecastingcited by 332
- 2024TFB: Towards Comprehensive and Fair Benchmarking of Time Series Forecasting Methodscited by 242
- 2020Multivariate Probabilistic Time Series Forecasting via Conditioned Normalizing Flowscited by 142
- 2023Predict, Refine, Synthesize: Self-Guiding Diffusion Models for Probabilistic Time Series Forecastingcited by 86
- 2021ScoreGrad: Multivariate Probabilistic Time Series Forecasting with Continuous Energy-based Generative Modelscited by 42
- 2021Slow Momentum with Fast Reversion: A Trading Strategy Using Deep Learning and Changepoint Detectioncited by 13
- 2018Diagnostic-Driven Nonstationary Emulators Using Kernel Mixturescited by 4
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026MoHETS: Long-term Time Series Forecasting with Mixture-of-Heterogeneous-Experts
- 2025Probabilistic Forecasting via Autoregressive Flow Matching
- 2021Learning Interpretable Deep State Space Model for Probabilistic Time Series Forecasting
- 2026Proper Scoring Rule-based Diffusion for Probabilistic Weather Forecasting
- 2025Adaptive Regime-Switching Forecasts with Distribution-Free Uncertainty: Deep Switching State-Space Models Meet Conformal Prediction
- 2026AME-TS: Anchored Mixture-of-Experts for Time Series Forecasting
- 2025MoGU: Mixture-of-Gaussians with Uncertainty-based Gating for Time Series Forecasting
- 2026RAVEN: A Regime-Aware Variable-context Expert Network for Financial Time Series Forecasting