vijaye12 commited on
Commit
edb1e80
·
verified ·
1 Parent(s): 0715444

update readme

Browse files
Files changed (1) hide show
  1. README.md +170 -0
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
+ ![](./ttm_image.webp)
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
+ }