Add TrackFormer v3 (protected dual-stream), new best model
Browse files
README.md
CHANGED
|
@@ -21,8 +21,9 @@ maritime, or emergency decisions.**
|
|
| 21 |
|
| 22 |
| model | params | inputs | training data |
|
| 23 |
|---|---|---|---|
|
|
|
|
| 24 |
| StormFusion-MT v2 | 3.3M (fp16, 6.7MB) | ERA5 patches + track history | WP, 2000+, 1,337 storm-centered windows |
|
| 25 |
-
|
|
| 26 |
|
| 27 |
Weights and full reproducible code (dataset builders, training, eval) are in the GitHub repo:
|
| 28 |
**https://github.com/yu314-coder/typhoon-predict** (`models/`).
|
|
@@ -32,12 +33,18 @@ Weights and full reproducible code (dataset builders, training, eval) are in the
|
|
| 32 |
| model | track km | vmax kt | pres hPa | rmw km | radius km |
|
| 33 |
|---|---|---|---|---|---|
|
| 34 |
| StormFusion-MT v2 (3.3M, ERA5) | 729 | 24.2 | 21.6 | 16.2 | 31.8 |
|
| 35 |
-
|
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
|
| 42 |
## Architectures
|
| 43 |
|
|
|
|
| 21 |
|
| 22 |
| model | params | inputs | training data |
|
| 23 |
|---|---|---|---|
|
| 24 |
+
| **TrackFormer v3** | 15M (fp16, 30MB) | **track history only, protected dual-stream** | all basins, 1980+, 84,150 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 |
|
| 28 |
Weights and full reproducible code (dataset builders, training, eval) are in the GitHub repo:
|
| 29 |
**https://github.com/yu314-coder/typhoon-predict** (`models/`).
|
|
|
|
| 33 |
| model | track km | vmax kt | pres hPa | rmw km | radius km |
|
| 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 |
+
|
| 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
|
| 41 |
+
*worse* than the 3.3M one). (2) Naively adding motion-dynamics features to a single-stream model
|
| 42 |
+
improves intensity but hurts track through **negative transfer**; **TrackFormer v3** fixes this with a
|
| 43 |
+
protected dual-stream architecture (separate kinematic/thermodynamic encoders, gradient routing, a
|
| 44 |
+
zero-init gated thermo→track adapter, and a persistence-residual track head), cutting WP-2020+ track
|
| 45 |
+
error to 659 km (−61, storm-bootstrap 95% CI [−103, −16] km, p≈0.995) while keeping the intensity
|
| 46 |
+
gains. Full architecture and derivation (incl. a random-matrix block-covariance uncertainty head) in
|
| 47 |
+
`paper/trackformer.pdf` in the GitHub repo.
|
| 48 |
|
| 49 |
## Architectures
|
| 50 |
|