Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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.

Vijay Ekambaram, Arindam Jati, Pankaj Dayama, Sumanta Mukherjee, Nam H. Nguyen, Wesley M. Gifford, Chandra Reddy, Jayant Kalagnanam• 2024

Related benchmarks

TaskDatasetResultRank
Time Series ForecastingGIFT-Eval (test)
MASE75.6
34
Deterministic forecastingETT Avg TSFM-Bench
MSE0.441
21
Deterministic forecastingSolar TSFM-Bench
MSE0.815
21
Deterministic forecastingWeather TSFM-Bench
MSE0.265
20
Deterministic forecastingWind TSFM-Bench
MSE1.337
19
Deterministic forecastingElectricity TSFM-Bench
MSE0.222
16
Deterministic forecastingPEMS08 TSFM-Bench
MSE1.73
15
Time Series ForecastingGIFT-Eval
MASE1.02
14
Deterministic forecastingTraffic TSFM-Bench
MSE0.564
13
Time Series ForecastingGIFT-Eval Part 1
LOOP-SEATTLE-5T-48 Error0.068
13
Showing 10 of 15 rows

Other info

Follow for update