Unified Training of Universal Time Series Forecasting Transformers

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Woo, Gerald, Liu, Chenghao, Kumar, Akshat, Xiong, Caiming, Savarese, Silvio, Sahoo, Doyen
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866917673271558144
author Woo, Gerald
Liu, Chenghao
Kumar, Akshat
Xiong, Caiming
Savarese, Silvio
Sahoo, Doyen
author_facet Woo, Gerald
Liu, Chenghao
Kumar, Akshat
Xiong, Caiming
Savarese, Silvio
Sahoo, Doyen
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2402_02592
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Unified Training of Universal Time Series Forecasting Transformers
Woo, Gerald
Liu, Chenghao
Kumar, Akshat
Xiong, Caiming
Savarese, Silvio
Sahoo, Doyen
Machine Learning
Artificial Intelligence
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.
title Unified Training of Universal Time Series Forecasting Transformers
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2402.02592