Tiny Time Mixers (TTMs): Fast Pre-trained Models for Enhanced Zero/Few-Shot Forecasting of Multivariate Time Series
About
Large pre-trained models excel in zero/few-shot learning for language and vision tasks but face challenges in multivariate time series (TS) forecasting due to diverse data characteristics. Consequently, recent research efforts have focused on developing pre-trained TS forecasting models. These models, whether built from scratch or adapted from large language models (LLMs), excel in zero/few-shot forecasting tasks. However, they are limited by slow performance, high computational demands, and neglect of cross-channel and exogenous correlations. To address this, we introduce Tiny Time Mixers (TTM), a compact model (starting from 1M parameters) with effective transfer learning capabilities, trained exclusively on public TS datasets. TTM, based on the light-weight TSMixer architecture, incorporates innovations like adaptive patching, diverse resolution sampling, and resolution prefix tuning to handle pre-training on varied dataset resolutions with minimal model capacity. Additionally, it employs multi-level modeling to capture channel correlations and infuse exogenous signals during fine-tuning. TTM outperforms existing popular benchmarks in zero/few-shot forecasting by (4-40%), while reducing computational requirements significantly. Moreover, TTMs are lightweight and can be executed even on CPU-only machines, enhancing usability and fostering wider adoption in resource-constrained environments. The model weights for reproducibility and research use are available at https://huggingface.co/ibm/ttm-research-r2/, while enterprise-use weights under the Apache license can be accessed as follows: the initial TTM-Q variant at https://huggingface.co/ibm-granite/granite-timeseries-ttm-r1, and the latest variants (TTM-B, TTM-E, TTM-A) weights are available at https://huggingface.co/ibm-granite/granite-timeseries-ttm-r2.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Time Series Forecasting | GIFT-Eval (test) | MASE75.6 | 34 | |
| Deterministic forecasting | ETT Avg TSFM-Bench | MSE0.441 | 21 | |
| Deterministic forecasting | Solar TSFM-Bench | MSE0.815 | 21 | |
| Deterministic forecasting | Weather TSFM-Bench | MSE0.265 | 20 | |
| Deterministic forecasting | Wind TSFM-Bench | MSE1.337 | 19 | |
| Deterministic forecasting | Electricity TSFM-Bench | MSE0.222 | 16 | |
| Deterministic forecasting | PEMS08 TSFM-Bench | MSE1.73 | 15 | |
| Time Series Forecasting | GIFT-Eval | MASE1.02 | 14 | |
| Deterministic forecasting | Traffic TSFM-Bench | MSE0.564 | 13 | |
| Time Series Forecasting | GIFT-Eval Part 1 | LOOP-SEATTLE-5T-48 Error0.068 | 13 |