euler314 commited on
Commit
9688662
·
verified ·
1 Parent(s): 1e3abbf

Add TrackFormer v3 (protected dual-stream), new best model

Browse files
Files changed (1) hide show
  1. README.md +14 -7
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
- | **TrackFormer** | 21M (fp16, 43MB) | **track history only (no ERA5)** | all basins, 1980+, 84,150 windows |
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
- | **TrackFormer (21M, no ERA5)** | **720** | **22.1** | 21.2 | **12.9** | 31.5 |
36
-
37
- **Key finding:** a track-only model that never sees ERA5 **matches or beats** the full ERA5
38
- model on every metric. The expensive ERA5 atmospheric patches add little over past-track history
39
- plus more storms. Across experiments the pattern was consistent: **data diversity > engineered
40
- features > parameters** a 17.7M ERA5 model overfit and did *worse* than the 3.3M one.
 
 
 
 
 
 
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