TrackFormer v8: partial-lead data expansion, best on every metric
Browse files
README.md
CHANGED
|
@@ -21,7 +21,7 @@ maritime, or emergency decisions.**
|
|
| 21 |
|
| 22 |
| model | params | inputs | training data |
|
| 23 |
|---|---|---|---|
|
| 24 |
-
| **TrackFormer
|
| 25 |
| StormFusion-MT v2 | 3.3M (fp16, 6.7MB) | ERA5 patches + track history | WP, 2000+, 1,337 storm-centered windows |
|
| 26 |
| TrackFormer (v1) | 21M (fp16, 43MB) | track history only (single-stream) | all basins, 1980+, 84,150 windows |
|
| 27 |
|
|
@@ -34,7 +34,8 @@ Weights and full reproducible code (dataset builders, training, eval) are in the
|
|
| 34 |
|---|---|---|---|---|---|
|
| 35 |
| StormFusion-MT v2 (3.3M, ERA5) | 729 | 24.2 | 21.6 | 16.2 | 31.8 |
|
| 36 |
| TrackFormer v1 (21M, single-stream) | 720 | 22.1 | 21.2 | 11.8 | 31.5 |
|
| 37 |
-
|
|
|
|
|
| 38 |
|
| 39 |
**Key findings.** (1) A track-only model that never sees ERA5 **matches or beats** the full ERA5
|
| 40 |
model, so **data diversity > engineered features > parameters** (a 17.7M ERA5 model overfit and did
|
|
|
|
| 21 |
|
| 22 |
| model | params | inputs | training data |
|
| 23 |
|---|---|---|---|
|
| 24 |
+
| **TrackFormer v8** | 15M (fp16, 30MB) | **track history only, protected dual-stream** | all basins, 1980+, 193k partial-lead windows |
|
| 25 |
| StormFusion-MT v2 | 3.3M (fp16, 6.7MB) | ERA5 patches + track history | WP, 2000+, 1,337 storm-centered windows |
|
| 26 |
| TrackFormer (v1) | 21M (fp16, 43MB) | track history only (single-stream) | all basins, 1980+, 84,150 windows |
|
| 27 |
|
|
|
|
| 34 |
|---|---|---|---|---|---|
|
| 35 |
| StormFusion-MT v2 (3.3M, ERA5) | 729 | 24.2 | 21.6 | 16.2 | 31.8 |
|
| 36 |
| TrackFormer v1 (21M, single-stream) | 720 | 22.1 | 21.2 | 11.8 | 31.5 |
|
| 37 |
+
| TrackFormer v3 (15M, dual-stream) | 659 | 21.6 | 18.1 | 11.8 | 28.8 |
|
| 38 |
+
| **TrackFormer v8 (15M, +partial-lead data)** | **649** | **20.7** | **15.9** | **11.3** | **27.8** |
|
| 39 |
|
| 40 |
**Key findings.** (1) A track-only model that never sees ERA5 **matches or beats** the full ERA5
|
| 41 |
model, so **data diversity > engineered features > parameters** (a 17.7M ERA5 model overfit and did
|