TEMPO: Prompt-based Generative Pre-trained Transformer for Time Series Forecasting
About
The past decade has witnessed significant advances in time series modeling with deep learning. While achieving state-of-the-art results, the best-performing architectures vary highly across applications and domains. Meanwhile, for natural language processing, the Generative Pre-trained Transformer (GPT) has demonstrated impressive performance via training one general-purpose model across various textual datasets. It is intriguing to explore whether GPT-type architectures can be effective for time series, capturing the intrinsic dynamic attributes and leading to significant accuracy improvements. In this paper, we propose a novel framework, TEMPO, that can effectively learn time series representations. We focus on utilizing two essential inductive biases of the time series task for pre-trained models: (i) decomposition of the complex interaction between trend, seasonal and residual components; and (ii) introducing the design of prompts to facilitate distribution adaptation in different types of time series. TEMPO expands the capability for dynamically modeling real-world temporal phenomena from data within diverse domains. Our experiments demonstrate the superior performance of TEMPO over state-of-the-art methods on zero shot setting for a number of time series benchmark datasets. This performance gain is observed not only in scenarios involving previously unseen datasets but also in scenarios with multi-modal inputs. This compelling finding highlights TEMPO's potential to constitute a foundational model-building framework.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Long-term time-series forecasting | Weather | MSE0.211 | 448 | |
| Long-term time-series forecasting | ETTh1 | MAE0.406 | 446 | |
| Long-term time-series forecasting | Traffic | MSE0.476 | 362 | |
| Long-term time-series forecasting | ETTh2 | MSE0.301 | 353 | |
| Long-term time-series forecasting | ETTm1 | MSE0.438 | 334 | |
| Long-term time-series forecasting | ETTm2 | MSE0.185 | 330 | |
| Long-term time-series forecasting | ECL | MSE0.178 | 154 | |
| RR Interval Estimation | ECG (test) | MAE (ms)141.5 | 12 | |
| Long-term time-series forecasting | Average all benchmarks | MSE0.37 | 12 | |
| Age Estimation | ECG (test) | MAE13.52 | 12 |