Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

TOTEM: TOkenized Time Series EMbeddings for General Time Series Analysis

About

This work studies the problem of time series analysis with generalist (or foundation) models, which are models trained across many data domains. Drawing inspiration from the widespread success of large language models, we consider the simple strategy of discretely tokenizing time series data drawn from a myriad of datasets via self-supervision, then using the fixed tokenization to solve a variety of tasks across many data domains. Canonically, time series models are either trained on a single dataset or built in a task-specific manner (e.g., a forecasting-only model), where many use patches of time as inputs to the model. As such, performant generalist, discrete representation time series models explored across many tasks are of value. Our method, TOkenized Time Series EMbeddings (TOTEM), produces such generalist time series models with minimal or no fine-tuning while exhibiting strong zero-shot performance. We evaluate TOTEM extensively over nearly 500 experiments on three commonly-studied time series tasks with real-world data: imputation (17 baselines, 12 datasets), anomaly detection (19 baselines, 25 datasets), and forecasting (14 baselines, 12 datasets). We conclude that TOTEM matches or outperforms existing state-of-the-art models in both the canonical specialist setting (i.e., training one model on one domain) as well as the generalist setting (i.e., training a single model on many domains), which demonstrates the efficacy of tokenization for general time series analysis. The open-source implementation is available here: https://github.com/SaberaTalukder/TOTEM; a video summary is available here: https://www.youtube.com/watch?v=OqrCpdb6MJk.

Sabera Talukder, Yisong Yue, Georgia Gkioxari• 2024

Related benchmarks

TaskDatasetResultRank
Time Series ForecastingETTh2
MSE0.299
561
Long-term forecastingETTm1
MSE0.811
375
Long-term forecastingETTm2
MSE0.38
310
Long-term forecastingExchange
MSE0.34
64
Time Series ForecastingETTm1 few-shot 10% data
MSE0.788
54
Long-term forecastingWeather 10% few-shot
MSE0.188
32
Long-term forecastingWeather 5% few-shot
MSE0.253
32
Long-term forecastingETTm2 10% few-shot
MSE0.26
32
Long-term forecastingETTm2 5% few-shot
MSE0.382
32
Long-term forecastingETTm1 5% few-shot
MSE0.892
32
Showing 10 of 14 rows

Other info

Follow for update