update readme
Browse files
README.md
CHANGED
|
@@ -1,3 +1,173 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
+
pipeline_tag: time-series-forecasting
|
| 4 |
+
tags:
|
| 5 |
+
- time series
|
| 6 |
+
- forecasting
|
| 7 |
+
- pretrained models
|
| 8 |
+
- foundation models
|
| 9 |
+
- time series foundation models
|
| 10 |
+
- time-series
|
| 11 |
+
- tinytimemixer
|
| 12 |
+
- granite-ttm-r3
|
| 13 |
---
|
| 14 |
+
|
| 15 |
+
# Granite-TTM-R3 Model Card
|
| 16 |
+
|
| 17 |
+

|
| 18 |
+
|
| 19 |
+
Building on top of **TTM-R1** and **TTM-R2**, we introduce the next generation of TinyTimeMixer under the Granite time-series foundation model family — **Granite-TTM-R3**. This release incorporates several novel *tiny-neural architectural innovations* designed to push the limits of **accuracy in high-speed forecasting**, a critical requirement for real-world production deployments.
|
| 20 |
+
|
| 21 |
+
---
|
| 22 |
+
|
| 23 |
+
## Supported Tasks & Capabilities
|
| 24 |
+
|
| 25 |
+
**Granite-TTM-R3** is a family of pretrained models supporting multiple real-world forecasting scenarios:
|
| 26 |
+
|
| 27 |
+
- **Zero-shot forecasting** across unseen datasets
|
| 28 |
+
- **Few-shot adaptation** effective with as few as ~1K samples
|
| 29 |
+
- **Full fine-tuning** for domain-specific optimization
|
| 30 |
+
- **Multivariate time-series forecasting**
|
| 31 |
+
- **Exogenous / control variable integration**
|
| 32 |
+
- **High-throughput batch inference** for production systems
|
| 33 |
+
|
| 34 |
+
---
|
| 35 |
+
|
| 36 |
+
## Accuracy & Speed
|
| 37 |
+
|
| 38 |
+
**Granite-TTM-R3** is engineered to achieve a strong balance between **state-of-the-art accuracy** and **extreme inference efficiency**, making it well-suited for real-world, high-throughput deployments.
|
| 39 |
+
|
| 40 |
+
### Accuracy on GIFT-Eval
|
| 41 |
+
|
| 42 |
+
- Maintains **top-tier performance** on the GIFT-Eval benchmarks.
|
| 43 |
+
|
| 44 |
+
**Fine-tuned FM**
|
| 45 |
+
|
| 46 |
+
- MASE: **0.718**
|
| 47 |
+
- CRPS: **514**
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
**Pre-trained FM**
|
| 51 |
+
|
| 52 |
+
- MASE: **0.727**
|
| 53 |
+
- CRPS: **523**
|
| 54 |
+
|
| 55 |
+
---
|
| 56 |
+
|
| 57 |
+
## ⚡ Inference Throughput
|
| 58 |
+
|
| 59 |
+
**Granite-TTM-R3** delivers orders-of-magnitude faster inference compared to existing popular SOTA models.
|
| 60 |
+
|
| 61 |
+
### GPU Throughput
|
| 62 |
+
|
| 63 |
+
- Typical SOTA models: **~20–500 samples/sec**
|
| 64 |
+
- **Granite-TTM-R3**: **~7,500 samples/sec**
|
| 65 |
+
|
| 66 |
+
### CPU Throughput
|
| 67 |
+
|
| 68 |
+
- Typical SOTA models: **~1–20 samples/sec**
|
| 69 |
+
- **Granite-TTM-R3**: **~180 samples/sec**
|
| 70 |
+
|
| 71 |
+
**Granite-TTM-R3 achieves ~15× speedup** over many existing approaches, without compromising accuracy, setting a strong benchmark for fast and reliable time-series forecasting.
|
| 72 |
+
|
| 73 |
+
---
|
| 74 |
+
|
| 75 |
+
## Architecture Overview
|
| 76 |
+
|
| 77 |
+
Granite-TTM adopts a mixture-of-experts paradigm composed of models with varying complexities — approximately **1M–35M parameters** — coupled with a lightweight routing mechanism that automatically selects or blends the most suitable expert based on input data characteristics.
|
| 78 |
+
|
| 79 |
+
This enables adaptive model selection, improving both accuracy and efficiency across diverse time-series scenarios.
|
| 80 |
+
|
| 81 |
+
The architecture is built on efficient mixer-based designs that avoid expensive self-attention. Instead, Granite-TTM-R3 leverages linear gating-based attention mechanisms to capture temporal dependencies with significantly lower computational overhead.
|
| 82 |
+
|
| 83 |
+
This combination allows **Granite-TTM-R3** to deliver scalable, adaptive, and ultra-fast forecasting performance suitable for real-time and large-scale deployments.
|
| 84 |
+
|
| 85 |
+
---
|
| 86 |
+
|
| 87 |
+
## What’s New in Granite-TTM-R3
|
| 88 |
+
|
| 89 |
+
- **Trend–Residual Decomposition**
|
| 90 |
+
Separately models long-term trends and high-frequency residuals for improved structural learning.
|
| 91 |
+
|
| 92 |
+
- **Three-Stage Pre-Training**
|
| 93 |
+
Sequential training of trend, residual, and joint components for better stability and convergence.
|
| 94 |
+
|
| 95 |
+
- **Student-Teacher Pretraining**
|
| 96 |
+
Student-teacher based pretraining for stable learning in noisy datasets.
|
| 97 |
+
|
| 98 |
+
- **Enhanced Data Augmentation**
|
| 99 |
+
Structured perturbations improve robustness across domains.
|
| 100 |
+
|
| 101 |
+
- **Improved Normalization Strategy**
|
| 102 |
+
Stabilizes training across scale shifts and heterogeneous datasets.
|
| 103 |
+
|
| 104 |
+
- **GLU Gating**
|
| 105 |
+
Dynamic information flow control within mixer blocks.
|
| 106 |
+
|
| 107 |
+
- **Multi-Resolution Temporal Layer**
|
| 108 |
+
Captures dependencies across short-, medium-, and long-term horizons.
|
| 109 |
+
|
| 110 |
+
- **FFT-Based Embeddings**
|
| 111 |
+
Incorporates frequency-domain signals to model periodicity and seasonality.
|
| 112 |
+
|
| 113 |
+
- **Register Tokens**
|
| 114 |
+
Learnable global tokens to encode sequence-level semantics.
|
| 115 |
+
|
| 116 |
+
- **Multi-Quantile Forecasting Head**
|
| 117 |
+
Enables probabilistic forecasting with multiple quantiles.
|
| 118 |
+
|
| 119 |
+
- **Refined Loss Weighting**
|
| 120 |
+
Balances trend, residual, and quantile objectives for improved calibration.
|
| 121 |
+
|
| 122 |
+
---
|
| 123 |
+
|
| 124 |
+
## ⚡ Why Granite-TTM-R3
|
| 125 |
+
|
| 126 |
+
**Granite-TTM-R3** is purpose-built for production-grade time-series systems where:
|
| 127 |
+
|
| 128 |
+
- **Low latency** is critical for real-time inference
|
| 129 |
+
- **High scale** is required for millions of forecasts
|
| 130 |
+
- **Data is heterogeneous and evolving**
|
| 131 |
+
- **Compute constraints exist**, including CPU-friendly deployment
|
| 132 |
+
|
| 133 |
+
It provides a practical combination of compactness, speed, and accuracy, making it suitable for industrial deployment.
|
| 134 |
+
|
| 135 |
+
---
|
| 136 |
+
|
| 137 |
+
## Example Recipes and Notebooks
|
| 138 |
+
|
| 139 |
+
To be released.
|
| 140 |
+
|
| 141 |
+
---
|
| 142 |
+
|
| 143 |
+
## Publication
|
| 144 |
+
|
| 145 |
+
- **Granite-TTM-R3 Paper**: To be released.
|
| 146 |
+
- **Base TTM Paper**: [Tiny Time Mixers (TTMs): Fast Pre-trained Models for Enhanced Zero/Few-Shot Forecasting of Multivariate Time Series](https://proceedings.neurips.cc/paper_files/paper/2024/hash/874a4d89f2d04b4bcf9a2c19545cf040-Abstract-Conference.html)
|
| 147 |
+
|
| 148 |
+
---
|
| 149 |
+
|
| 150 |
+
## Training Data
|
| 151 |
+
|
| 152 |
+
- Select Datasets from [GiftEvalPretrain](https://huggingface.co/datasets/Salesforce/GiftEvalPretrain) and [GIFT-Eval Train](https://huggingface.co/datasets/Salesforce/GiftEval), using non-leaking historical context.
|
| 153 |
+
- Custom synthesized data based on [KernelSynth](https://github.com/amazon-science/chronos-forecasting/blob/main/scripts/kernel-synth.py).
|
| 154 |
+
|
| 155 |
+
---
|
| 156 |
+
|
| 157 |
+
## Model Card Authors
|
| 158 |
+
|
| 159 |
+
Vijay Ekambaram, Arindam Jati, Haoxiang Qiu, Takayuki Katsuki, Tomoya Sakai, Priyanshul Govil, Pankaj Dayama
|
| 160 |
+
|
| 161 |
+
---
|
| 162 |
+
|
| 163 |
+
## Citation
|
| 164 |
+
|
| 165 |
+
Please cite the following paper if you intend to use this model or its associated architectures and approaches in your work.
|
| 166 |
+
|
| 167 |
+
```bibtex
|
| 168 |
+
@inproceedings{ekambaram2024tinytimemixersttms,
|
| 169 |
+
title={Tiny Time Mixers (TTMs): Fast Pre-trained Models for Enhanced Zero/Few-Shot Forecasting of Multivariate Time Series},
|
| 170 |
+
author={Vijay Ekambaram and Arindam Jati and Pankaj Dayama and Sumanta Mukherjee and Nam H. Nguyen and Wesley M. Gifford and Chandra Reddy and Jayant Kalagnanam},
|
| 171 |
+
booktitle={Advances in Neural Information Processing Systems (NeurIPS 2024)},
|
| 172 |
+
year={2024},
|
| 173 |
+
}
|