Update Trackformer 1.2 model card with completed DeepMind CUDA benchmark
Browse files
README.md
CHANGED
|
@@ -44,27 +44,44 @@ This selected example illustrates the model, not typical skill. Observations are
|
|
| 44 |
|
| 45 |
Trackformer 1.1 predicts track and scalar intensity/structure outputs. **1.2 adds evolving sea-level-pressure fields and a moving pressure core**, giving the route forecast a spatial weather representation that can be inspected on a map.
|
| 46 |
|
| 47 |
-
 is underway on the same 1,473 starts and twenty leads. Its scores are **pending**, not borrowed from older benchmarks. This reference uses one member and its own causal ERA5 input pipeline; 1.2 uses the saved mean of 50. Common masks are recomputed for the three-way comparison. Historical cases overlap Mini's training years, so recent and historical groups are reported separately. [Benchmark protocol](https://huggingface.co/euler314/typhoon-predict/blob/main/docs/deepmind_daily_benchmark.md)
|
| 60 |
-
|
| 61 |
The cohort contains 40 recent storms and 230 storms from 1980–1999. It lies outside the selected Trackformer's fitting/validation years but is a **development comparison, not a certified untouched holdout**. Different input pipelines mean this is not a controlled architecture ablation. Position, direction and route-shape similarity are separate measures; good shape alone does not prove geographical alignment.
|
| 62 |
|
| 63 |
[Full daily results](https://huggingface.co/euler314/typhoon-predict/blob/main/evaluation/daily_storm_final.json) · [Shared metrics and uncertainty](https://huggingface.co/euler314/typhoon-predict/blob/main/evaluation/released_daily/released_daily_benchmark.json) · [Forecast/hash audit](https://huggingface.co/euler314/typhoon-predict/blob/main/evaluation/released_daily/released_daily_verification.json) · [Daily protocol](https://huggingface.co/euler314/typhoon-predict/blob/main/docs/daily_storm_benchmark.md) · [Pressure, wind and radius evaluation](https://huggingface.co/euler314/typhoon-predict/blob/main/docs/intensity_benchmark.md)
|
| 64 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 65 |
## How Trackformer 1.2 works
|
| 66 |
|
| 67 |
-

|
| 48 |
|
| 49 |
+
| Development metric | 1.1 | 1.2 · mean of 50 | DeepMind Mini · one member | Shared coverage |
|
| 50 |
+
| --- | ---: | ---: | ---: | --- |
|
| 51 |
+
| Mean track error, +6 to +120 h · lower is better | 798.4 km | **471.2 km** | 498.4 km | 1,473 days / 270 storms |
|
| 52 |
+
| Six-hour track-direction error · lower is better | 51.58° | **34.96°** | 39.76° | 1,473 days / 270 storms |
|
| 53 |
+
| Central-pressure MAE · JMA · lower is better | 13.53 hPa | 12.84 hPa | **10.20 hPa** | 134 days / 40 storms |
|
| 54 |
+
| Centred route-shape similarity · higher is better | 0.7544 | 0.8837 | **0.9010** | 1,473 days / 270 storms |
|
| 55 |
+
| Pressure-curve similarity · JMA · higher is better | 0.7074 | 0.7118 | **0.8088** | 134 days / 40 storms |
|
| 56 |
|
| 57 |
Mean track error is **41.0% lower** for 1.2 on this cohort. Central-pressure error has a **small mean reduction**; the paired whole-storm 95% interval includes no improvement (−2.68 to +1.20 hPa). These results do not establish a statistically reliable pressure advantage or improvements in every output or lead.
|
| 58 |
|
| 59 |
**One storm-day is one case.** Scores average valid +6 to +120 h leads within each day, then days within each typhoon, then typhoons with equal weight. Both versions use the same frozen starts and exact valid times. Missing labels and unsupported outputs are excluded, never treated as zero errors. Track and pressure have different valid coverage, shown above.
|
| 60 |
|
|
|
|
|
|
|
| 61 |
The cohort contains 40 recent storms and 230 storms from 1980–1999. It lies outside the selected Trackformer's fitting/validation years but is a **development comparison, not a certified untouched holdout**. Different input pipelines mean this is not a controlled architecture ablation. Position, direction and route-shape similarity are separate measures; good shape alone does not prove geographical alignment.
|
| 62 |
|
| 63 |
[Full daily results](https://huggingface.co/euler314/typhoon-predict/blob/main/evaluation/daily_storm_final.json) · [Shared metrics and uncertainty](https://huggingface.co/euler314/typhoon-predict/blob/main/evaluation/released_daily/released_daily_benchmark.json) · [Forecast/hash audit](https://huggingface.co/euler314/typhoon-predict/blob/main/evaluation/released_daily/released_daily_verification.json) · [Daily protocol](https://huggingface.co/euler314/typhoon-predict/blob/main/docs/daily_storm_benchmark.md) · [Pressure, wind and radius evaluation](https://huggingface.co/euler314/typhoon-predict/blob/main/docs/intensity_benchmark.md)
|
| 64 |
|
| 65 |
+
### Completed DeepMind comparison
|
| 66 |
+
|
| 67 |
+
The reference is Google's official [WeatherNext Cyclones Mini `<2024`](https://github.com/google-deepmind/weathernext#weathernext-cyclones-mini), a **1° model trained through 2023**, run with **WeatherNext software v0.3.0**. The software version is not the checkpoint name. This is **Mini, not the full-sized WeatherNext 2/Cyclones model**.
|
| 68 |
+
|
| 69 |
+
All **1,473 daily starts / 270 storms** completed on an **NVIDIA RTX 3070 using CUDA**. Each saved forecast contains twenty exact six-hour route and native pressure-map outputs through +120 h. The returned archive's hashes, original Trackformer forecasts, common masks and equal-storm scores were independently rechecked. Mini uses one seeded member and its own causal ERA5 analyses; 1.2 uses a 50-input-member mean. No missing result is scored as zero.
|
| 70 |
+
|
| 71 |
+
The combined cohort favours 1.2 for position and heading, while Mini has lower central-pressure error and higher pressure-curve similarity. **The recent-only track results favour Mini**, so the combined numbers must not be read as a general advantage over DeepMind:
|
| 72 |
+
|
| 73 |
+
| Recent storms beginning in 2024+ · 137 days / 40 storms | 1.1 | 1.2 · mean of 50 | DeepMind Mini · one member |
|
| 74 |
+
| --- | ---: | ---: | ---: |
|
| 75 |
+
| Mean track error · lower is better | 901.4 km | 477.2 km | **288.5 km** |
|
| 76 |
+
| Six-hour direction error · lower is better | 52.90° | 34.68° | **31.16°** |
|
| 77 |
+
|
| 78 |
+
The remaining 1,336 days / 230 historical storms overlap Mini's fitting years. JMA pressure scores use 2,637 shared valid leads from the smaller 134-day subset, not all 1,473 days. Curve similarity measures timing/shape after removing pressure level and amplitude; it does not replace hPa error. These are point estimates, not a claim of statistically proven superiority.
|
| 79 |
+
|
| 80 |
+
[Completed results and period breakdowns](https://huggingface.co/euler314/typhoon-predict/blob/main/evaluation/deepmind_daily/benchmark.json) · [CUDA completion receipt](https://huggingface.co/euler314/typhoon-predict/blob/main/evaluation/deepmind_daily/verification.json) · [Independent publication audit](https://huggingface.co/euler314/typhoon-predict/blob/main/evaluation/deepmind_daily/publication_audit.json) · [Model identity and reproducible protocol](https://huggingface.co/euler314/typhoon-predict/blob/main/docs/deepmind_daily_benchmark.md)
|
| 81 |
+
|
| 82 |
## How Trackformer 1.2 works
|
| 83 |
|
| 84 |
+

|
| 85 |
|
| 86 |
1. **Initialize from the past.** Nine six-hour analyses provide MSLP, 500 hPa height and winds at 850/500/200 hPa, ending at issue time. Static geography, the current observed centre, recent motion, current intensity and validity masks accompany the history. Native regional pressure detail is used when available; missing detail stays flagged. Current observations initialize the core, not future labels.
|
| 87 |
2. **Evolve the environment.** A convolutional recurrent network combines steering-based advection with learned pressure tendencies. Multiscale spatial tokens and attention condition the environmental memory.
|