Add p8-s4 to model table (21 models)
Browse files
README.md
CHANGED
|
@@ -1,83 +1,84 @@
|
|
| 1 |
-
---
|
| 2 |
-
language: en
|
| 3 |
-
license: apache-2.0
|
| 4 |
-
tags:
|
| 5 |
-
- time-series
|
| 6 |
-
- forecasting
|
| 7 |
-
- chronos
|
| 8 |
-
- patch-aliasing
|
| 9 |
-
base_model: amazon/chronos-bolt-tiny
|
| 10 |
-
---
|
| 11 |
-
|
| 12 |
-
# Chronos-Bolt Patch-Stride Sweep
|
| 13 |
-
|
| 14 |
-
Retrained variants of [Chronos-Bolt Tiny](https://huggingface.co/amazon/chronos-bolt-tiny) (~8.7M params) with different **patch size (P)** and **patch stride (S)** configurations, trained from scratch on the full official Chronos pre-training data.
|
| 15 |
-
|
| 16 |
-
## Purpose
|
| 17 |
-
|
| 18 |
-
These models support the **structural patch-aliasing** study: investigating how the (P, S) patch geometry of Chronos-Bolt creates frequency-dependent artifacts in time-series forecasting. Each model is trained identically except for the patch geometry, so any downstream difference in aliasing probes is attributable to (P, S).
|
| 19 |
-
|
| 20 |
-
## Models
|
| 21 |
-
|
| 22 |
-
| Subfolder | P | S | Overlap | Status |
|
| 23 |
-
|---|---|---|---|---|
|
| 24 |
-
| `p8-
|
| 25 |
-
| `
|
| 26 |
-
| `p16-
|
| 27 |
-
| `p16-
|
| 28 |
-
| `p16-
|
| 29 |
-
| `p16-
|
| 30 |
-
| `
|
| 31 |
-
| `p24-
|
| 32 |
-
| `p24-
|
| 33 |
-
| `p24-
|
| 34 |
-
| `p24-
|
| 35 |
-
| `p24-
|
| 36 |
-
| `
|
| 37 |
-
| `p32-
|
| 38 |
-
| `p32-
|
| 39 |
-
| `p32-
|
| 40 |
-
| `p32-
|
| 41 |
-
| `p32-
|
| 42 |
-
| `p32-
|
| 43 |
-
| `p32-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
- **
|
| 51 |
-
- **
|
| 52 |
-
- **
|
| 53 |
-
- **
|
| 54 |
-
- **
|
| 55 |
-
- **
|
| 56 |
-
- **
|
| 57 |
-
- **
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
- `
|
| 77 |
-
- `
|
| 78 |
-
- `
|
| 79 |
-
- `
|
| 80 |
-
|
| 81 |
-
|
| 82 |
-
|
| 83 |
-
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: en
|
| 3 |
+
license: apache-2.0
|
| 4 |
+
tags:
|
| 5 |
+
- time-series
|
| 6 |
+
- forecasting
|
| 7 |
+
- chronos
|
| 8 |
+
- patch-aliasing
|
| 9 |
+
base_model: amazon/chronos-bolt-tiny
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# Chronos-Bolt Patch-Stride Sweep
|
| 13 |
+
|
| 14 |
+
Retrained variants of [Chronos-Bolt Tiny](https://huggingface.co/amazon/chronos-bolt-tiny) (~8.7M params) with different **patch size (P)** and **patch stride (S)** configurations, trained from scratch on the full official Chronos pre-training data.
|
| 15 |
+
|
| 16 |
+
## Purpose
|
| 17 |
+
|
| 18 |
+
These models support the **structural patch-aliasing** study: investigating how the (P, S) patch geometry of Chronos-Bolt creates frequency-dependent artifacts in time-series forecasting. Each model is trained identically except for the patch geometry, so any downstream difference in aliasing probes is attributable to (P, S).
|
| 19 |
+
|
| 20 |
+
## Models
|
| 21 |
+
|
| 22 |
+
| Subfolder | P | S | Overlap | Status |
|
| 23 |
+
|---|---|---|---|---|
|
| 24 |
+
| `p8-s4-seed42` | 8 | 4 | 0.500 | done |
|
| 25 |
+
| `p8-s8-seed42` | 8 | 8 | 0.000 | done |
|
| 26 |
+
| `p16-s4-seed42` | 16 | 4 | 0.750 | done |
|
| 27 |
+
| `p16-s8-seed42` | 16 | 8 | 0.500 | done |
|
| 28 |
+
| `p16-s12-seed42` | 16 | 12 | 0.250 | done |
|
| 29 |
+
| `p16-s15-seed42` | 16 | 15 | 0.062 | done |
|
| 30 |
+
| `p16-s16-seed42` | 16 | 16 | 0.000 | done |
|
| 31 |
+
| `p24-s8-seed42` | 24 | 8 | 0.667 | done |
|
| 32 |
+
| `p24-s12-seed42` | 24 | 12 | 0.500 | done |
|
| 33 |
+
| `p24-s15-seed42` | 24 | 15 | 0.375 | done |
|
| 34 |
+
| `p24-s16-seed42` | 24 | 16 | 0.333 | done |
|
| 35 |
+
| `p24-s20-seed42` | 24 | 20 | 0.167 | done |
|
| 36 |
+
| `p24-s24-seed42` | 24 | 24 | 0.000 | done |
|
| 37 |
+
| `p32-s8-seed42` | 32 | 8 | 0.750 | done |
|
| 38 |
+
| `p32-s12-seed42` | 32 | 12 | 0.625 | done |
|
| 39 |
+
| `p32-s15-seed42` | 32 | 15 | 0.531 | done |
|
| 40 |
+
| `p32-s16-seed42` | 32 | 16 | 0.500 | done |
|
| 41 |
+
| `p32-s20-seed42` | 32 | 20 | 0.375 | done |
|
| 42 |
+
| `p32-s24-seed42` | 32 | 24 | 0.250 | done |
|
| 43 |
+
| `p32-s28-seed42` | 32 | 28 | 0.125 | done |
|
| 44 |
+
| `p32-s32-seed42` | 32 | 32 | 0.000 | done |
|
| 45 |
+
|
| 46 |
+
## Training setup
|
| 47 |
+
|
| 48 |
+
All models share the same training regime (only P and S vary):
|
| 49 |
+
|
| 50 |
+
- **Architecture**: Chronos-Bolt Tiny (T5-based, ~8.7M params), random initialization
|
| 51 |
+
- **Data**: Official Chronos pre-training corpus β TSMixup (10M series) + KernelSynth (1M series) at 9:1 ratio
|
| 52 |
+
- **Steps**: 100,000 (fixed budget across all runs)
|
| 53 |
+
- **Optimizer**: AdamW, lr=1e-3, linear decay, no warmup
|
| 54 |
+
- **Batch size**: 32
|
| 55 |
+
- **Precision**: fp32 + TF32 matmuls (Ampere+ GPUs)
|
| 56 |
+
- **Context**: 2048 tokens, prediction horizon: 64
|
| 57 |
+
- **Quantiles**: 9 (0.1 to 0.9)
|
| 58 |
+
- **Seed**: 42
|
| 59 |
+
|
| 60 |
+
Full provenance is recorded in each subfolder's `run_config.json`.
|
| 61 |
+
|
| 62 |
+
## Usage
|
| 63 |
+
|
| 64 |
+
```python
|
| 65 |
+
from chronos import BaseChronosPipeline
|
| 66 |
+
|
| 67 |
+
pipe = BaseChronosPipeline.from_pretrained(
|
| 68 |
+
"federicosabbadini/chronos-bolt-patch-sweep",
|
| 69 |
+
subfolder="p24-s16-seed42",
|
| 70 |
+
device_map="cpu",
|
| 71 |
+
)
|
| 72 |
+
```
|
| 73 |
+
|
| 74 |
+
## Files per model
|
| 75 |
+
|
| 76 |
+
- `config.json` β model architecture config
|
| 77 |
+
- `model.safetensors` β trained weights (~34 MB)
|
| 78 |
+
- `run_config.json` β full training provenance + result metrics
|
| 79 |
+
- `loss_history.npy` β per-step training loss
|
| 80 |
+
- `loss_curve.png` β training loss plot
|
| 81 |
+
|
| 82 |
+
## License
|
| 83 |
+
|
| 84 |
+
The models are derived from the Chronos architecture (Apache-2.0) and trained on the official Chronos datasets. See the original [Chronos repository](https://github.com/amazon-science/chronos-forecasting) for details.
|