Unified Training of Universal Time Series Forecasting Transformers

By Gerald Woo, Chenghao Liu, Akshat Kumar, Caiming Xiong, Silvio Savarese, Doyen Sahoo. International Conference on Machine Learning, 2024

656 citations recorded by the source corpus.

Deep learning for time series forecasting has traditionally operated within a one-model-per-dataset framework, limiting its potential to leverage the game-changing impact of large pre-trained models. The concept of universal forecasting, emerging from pre-training on a vast collection of time series datasets, envisions a single Large Time Series Model capable of addressing diverse downstream forecasting tasks. However, constructing such a model poses unique challenges specific to time series data: i) cross-frequency learning, ii) accommodating an arbitrary number of variates for multivariate time series, and iii) addressing the varying distributional properties inherent in large-scale data. To address these challenges, we present novel enhancements to the conventional time series Transformer architecture, resulting in our proposed Masked Encoder-based Universal Time Series Forecasting Transformer (Moirai). Trained on our newly introduced Large-scale Open Time Series Archive (LOTSA) featuring over 27B observations across nine domains, Moirai achieves competitive or superior performance as a zero-shot forecaster when compared to full-shot models. Code, data, and model weights can be found at https://github.com/SalesforceAIResearch/uni2ts.

Subfields: Artificial Intelligence, Management Science and Operations Research, Signal Processing

View Paper: https://www.semanticscholar.org/paper/4a111f7a3b56d0468f13104999844885157ef17d

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Artur Dubrawski
  2. Mononito Goswami
  3. Xiaoming Shi
  4. Ming Jin
  5. Shiyu Wang
  6. Konrad Szafer
  7. Yifu Cai
  8. Arjun Choudhry

In Management Science and Operations Research

  1. Artur Dubrawski
  2. Shiyu Wang

In Signal Processing

  1. Artur Dubrawski
  2. Xiaoming Shi
  3. Shiyu Wang

Citations in the corpus