Skip to main content

VisionTS

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

VisionTS VisionTS++ PyPI - Version AI horizon forecast 机器之心

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

🎉 What's New

  • 🔥 Aug 2025: We released VisionTS++, a SOTA time series foundation model by continual pretraining visual MAE on large-scale time series data, supporting multi-channel forecasting and probablistic forecasting!

  • May 2025: Our paper is accepted by ICML 2025!

  • Nov 2024: VisionTS achieved the #1 rank 🏆 for zero-shot point forecasting (MASE) on GIFT-EVAL (as of Nov 2024, surpassing Moirai, TimesFM, chronos, etc) — without any time series training!

🔍 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 eval_gluonts/
bash run_monash.sh

[!IMPORTANT] The results in the paper are evaluated based on python==3.8.18, torch==1.7.1, torchvision==0.8.2, and timm==0.3.2. Different versions may lead to slightly different performance.

PF (Zero-Shot)

We evaluate our methods on 6 long-term TSF benchmarks for zero-shot forecasting. Before running, you should first follow the instructions of Time-Series-Library to download datasets into long_term_tsf/dataset, in addition to the following three datasets:

You can use the following command for reproduction.

cd eval_gluonts/
bash run_pf.sh

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

Release files for visionts 1.0.1

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for visionts 1.0.1
File Size Uploaded
visionts-1.0.1.tar.gz 19.6 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for visionts 1.0.1
File Interpreter ABI Platform
visionts-1.0.1-py3-none-any.whl Python 3 none any Details

Total release size: 36.8 kB

Release files / visionts-1.0.1.tar.gz

Download URL visionts-1.0.1.tar.gz
Size 19.6 kB
Tags Source
SHA-256 checksum
How to use checksums
69d6e38114b2f45e65142c5424094631c41299e986da6353b199b30f47777bf8
BLAKE2b-256 checksum
How to use checksums
770988989a859363128a338e1dfe2c3910c600d8b8550de80ef21b03efb793b8
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/5.1.1 CPython/3.8.15

Release files / visionts-1.0.1-py3-none-any.whl

Download URL visionts-1.0.1-py3-none-any.whl
Size 17.2 kB
Tags Python 3
SHA-256 checksum
How to use checksums
0af0540575e18194f6fbeefff5f864d404e8a3c06b7fee15efd837d5daa47b83
BLAKE2b-256 checksum
How to use checksums
aec8cd8475085d47eddabd1928b5d59eae4b5a579d47ff26d968e09ffb364bd2
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/5.1.1 CPython/3.8.15

Release history Release notifications | RSS feed

This release

1.0.1 This release

2 release files

1.0.0

2 release files

0.2.0

2 release files

0.1.4

2 release files

0.1.3

2 release files

0.1.2

2 release files

0.1.1

2 release files

0.1.0

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page