Mamba4Cast: Efficient Zero-Shot Time Series Forecasting with State Space Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Bhethanabhotla, Sathya Kamesh, Swelam, Omar, Siems, Julien, Salinas, David, Hutter, Frank
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916435203194880
author Bhethanabhotla, Sathya Kamesh
Swelam, Omar
Siems, Julien
Salinas, David
Hutter, Frank
author_facet Bhethanabhotla, Sathya Kamesh
Swelam, Omar
Siems, Julien
Salinas, David
Hutter, Frank
contents This paper introduces Mamba4Cast, a zero-shot foundation model for time series forecasting. Based on the Mamba architecture and inspired by Prior-data Fitted Networks (PFNs), Mamba4Cast generalizes robustly across diverse time series tasks without the need for dataset specific fine-tuning. Mamba4Cast's key innovation lies in its ability to achieve strong zero-shot performance on real-world datasets while having much lower inference times than time series foundation models based on the transformer architecture. Trained solely on synthetic data, the model generates forecasts for entire horizons in a single pass, outpacing traditional auto-regressive approaches. Our experiments show that Mamba4Cast performs competitively against other state-of-the-art foundation models in various data sets while scaling significantly better with the prediction length. The source code can be accessed at https://github.com/automl/Mamba4Cast.
format Preprint
id arxiv_https___arxiv_org_abs_2410_09385
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Mamba4Cast: Efficient Zero-Shot Time Series Forecasting with State Space Models
Bhethanabhotla, Sathya Kamesh
Swelam, Omar
Siems, Julien
Salinas, David
Hutter, Frank
Machine Learning
Artificial Intelligence
This paper introduces Mamba4Cast, a zero-shot foundation model for time series forecasting. Based on the Mamba architecture and inspired by Prior-data Fitted Networks (PFNs), Mamba4Cast generalizes robustly across diverse time series tasks without the need for dataset specific fine-tuning. Mamba4Cast's key innovation lies in its ability to achieve strong zero-shot performance on real-world datasets while having much lower inference times than time series foundation models based on the transformer architecture. Trained solely on synthetic data, the model generates forecasts for entire horizons in a single pass, outpacing traditional auto-regressive approaches. Our experiments show that Mamba4Cast performs competitively against other state-of-the-art foundation models in various data sets while scaling significantly better with the prediction length. The source code can be accessed at https://github.com/automl/Mamba4Cast.
title Mamba4Cast: Efficient Zero-Shot Time Series Forecasting with State Space Models
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2410.09385