Skip to main content

Using a vision MAE for time series forecasting.

Project description

VisionTS

Visual Masked Autoencoders Are Free-Lunch Zero-Shot Time Series Forecasters

Paper PyPI - Version

🔍 About | 🚀 Quick Start | 📊 Evaluation | 🔗 Citation

🔍 About

  • We propose VisionTS, a time series forecasting (TSF) foundation model building from rich, high-quality natural images 🖼️.

    • This is conceptually different from the existing TSF foundation models (text-based 📝 or time series-based 📈), but it shows a comparable or even better performance without any adaptation on time series data.
  • We reformulate the TSF task as an image reconstruction task, which is further processed by a visual masked autoencoder (MAE).

🚀 Quick Start

We have uploaded our package to PyPI. Please first install pytorch, then running the following command for installing VisionTS:

pip install visionts

Then, you can refer to demo.ipynb about forecasting time series using VisionTS, with a clear visualization of the image reconstruction.

📊 Evaluation

Our repository is built on Time-Series-Library, MAE, and GluonTS. Please install the dependencies through requirements.txt before running the evaluation.

Long-Term TSF Benchmarks (Zero-Shot)

We evaluate our methods on 6 long-term TSF benchmarks for zero-shot forecasting. The scripts are under long_term_tsf/scripts/vision_ts_zeroshot. Before running, you should first follow the instructions of Time-Series-Library to download datasets into long_term_tsf/dataset. Using the following command for reproduction:

cd long_term_tsf/
bash scripts/vision_ts_zeroshot/$SOME_DATASET.sh

Monash (Zero-Shot)

We evaluate our methods on 29 Monash TSF benchmarks. You can use the following command for reproduction, where the benchmarks will be automatically downloaded.

cd monash/
python run.py

Long-Term TSF Benchmarks (Full-Shot)

We evaluate our methods on 8 long-term TSF benchmarks for full-shot forecasting. The scripts are under long_term_tsf/scripts/vision_ts_fullshot. Using the following command for reproduction:

cd long_term_tsf/
bash scripts/vision_ts_fullshot/$SOME_DATASET.sh

🔗 Citation

@misc{chen2024visionts,
      title={VisionTS: Visual Masked Autoencoders Are Free-Lunch Zero-Shot Time Series Forecasters}, 
      author={Mouxiang Chen and Lefei Shen and Zhuo Li and Xiaoyun Joy Wang and Jianling Sun and Chenghao Liu},
      year={2024},
      eprint={2408.17253},
      archivePrefix={arXiv},
      url={https://arxiv.org/abs/2408.17253}, 
}

⭐ Star History

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

visionts-0.1.2.tar.gz (10.5 kB view details)

Uploaded Source

Built Distribution

visionts-0.1.2-py3-none-any.whl (10.1 kB view details)

Uploaded Python 3

File details

Details for the file visionts-0.1.2.tar.gz.

File metadata

  • Download URL: visionts-0.1.2.tar.gz
  • Upload date:
  • Size: 10.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.8.15

File hashes

Hashes for visionts-0.1.2.tar.gz
Algorithm Hash digest
SHA256 c00238af1d9175561a2cbbba3411d658bf6d6dc792a45a6b8b768b8a540ece89
MD5 6de3dab99af58368f398db5f93e91cc3
BLAKE2b-256 64fa14ec0a4cedf4402ec3db052be076466c5abe7b7745b53ca7ef85a686fbc6

See more details on using hashes here.

File details

Details for the file visionts-0.1.2-py3-none-any.whl.

File metadata

  • Download URL: visionts-0.1.2-py3-none-any.whl
  • Upload date:
  • Size: 10.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.8.15

File hashes

Hashes for visionts-0.1.2-py3-none-any.whl
Algorithm Hash digest
SHA256 a9c43f5b12b646990276eee7484a27936414b1c3c886fbb692116cd47820b46b
MD5 e99622906f4b7282b338e12456eafe09
BLAKE2b-256 ec24683d44fb4723a4ca15f88cf496b23eea836c3354e770dac9f09bf6d3f712

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page