Shoaib78971 commited on
Commit
aea1474
·
verified ·
1 Parent(s): 34a6147

Align Flash checkpoint layout with PhysicsNeMo and document MRMS evaluation

Browse files

Update the six Flash .mdlus archives to the validated native layer-key layout and earth2studio.models.nn.stormscope_flash serialization path. Preserve trained tensors, expert numbering, selected weights and sampling recipes. Update deployment hashes and sizes only in the registry. Move the MRMS skill table into a standalone evaluation note linked from the model card. Baseline entries, checkpoints and shared assets are unchanged.

README.md CHANGED
@@ -22,26 +22,8 @@ GOES and MRMS/GLM inference. It uses five model evaluations for GOES and seven f
22
  MRMS/GLM per forecast step, and supports full-CONUS and regional inference through
23
  Earth2Studio.
24
 
25
- ### MRMS forecasting skill
26
-
27
- Comparison of the published StormScope teacher baseline and the distilled
28
- StormScope Flash model. Higher scores indicate better forecasting skill; bold
29
- marks the higher score for each metric and forecast lead.
30
-
31
- | Metric | Model | +30 min | +60 min | +120 min | +180 min |
32
- | --- | --- | ---: | ---: | ---: | ---: |
33
- | Object CSI | StormScope (published, 3 km / 10 min) | 0.685 | **0.533** | **0.335** | **0.220** |
34
- | Object CSI | StormScope Flash | **0.688** | 0.532 | 0.315 | 0.204 |
35
- | CSI 40 dBZ | StormScope (published, 3 km / 10 min) | **0.597** | **0.479** | **0.348** | **0.271** |
36
- | CSI 40 dBZ | StormScope Flash | 0.593 | 0.464 | 0.326 | 0.246 |
37
- | CSI 55 dBZ | StormScope (published, 3 km / 10 min) | 0.274 | **0.170** | **0.068** | **0.028** |
38
- | CSI 55 dBZ | StormScope Flash | **0.286** | 0.167 | 0.060 | 0.024 |
39
- | pFSS 40 dBZ | StormScope (published, 3 km / 10 min) | 0.907 | 0.788 | 0.588 | **0.449** |
40
- | pFSS 40 dBZ | StormScope Flash | **0.913** | **0.800** | **0.593** | 0.433 |
41
- | pFSS 55 dBZ | StormScope (published, 3 km / 10 min) | 0.719 | **0.502** | **0.201** | **0.064** |
42
- | pFSS 55 dBZ | StormScope Flash | **0.735** | 0.490 | 0.137 | 0.036 |
43
-
44
- CSI: critical success index. pFSS: probabilistic fractions skill score.
45
 
46
  ### License/Terms of Use:
47
 
 
22
  MRMS/GLM per forecast step, and supports full-CONUS and regional inference through
23
  Earth2Studio.
24
 
25
+ See [MRMS forecasting skill](mrms_forecast_skill.md) for the comparison with
26
+ StormScope, including the evaluation setup, metric definitions, and results.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
27
 
28
  ### License/Terms of Use:
29
 
checkpoints/goes/3km_10min_flash/expert_0.mdlus CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:75aeb03cddd5596570a9e1feaa6444173abc24ae736b84c63daf74218cc07007
3
- size 723168328
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de2f5a2127f62d6dd868a8c4043c033604b77ca5492b9cd0de9aa5a6cd277e83
3
+ size 723168495
checkpoints/goes/3km_10min_flash/expert_1.mdlus CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4f53361f589ddb3ee4097e4d362d59c027b8551022ee42bb81460df1c3ec11b4
3
- size 704267546
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c860c5ef5088ea7a2718231d50c5026438d5d99e774e28db64845c3ee29a31e
3
+ size 704268145
checkpoints/goes/3km_10min_flash/expert_2.mdlus CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6b8cb151b07c9b754f6422f3f7a763e5154a2165d2482e2b9db37c46f2e0c362
3
- size 697967288
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e761285cc87e4d8088cb7fdbebd94c7e7e135b81306d4d0152c27649e40eafd
3
+ size 697968031
checkpoints/mrms/3km_10min_flash/expert_0.mdlus CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7c3112543db0e4cf299fea12caec180508251230822cc975891079ba3d19e29e
3
- size 703869930
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:322228b92fbdbcca34c3ef50c2e15d37e8b0ebf22b4e0c15992e9aefd384fb40
3
+ size 703869841
checkpoints/mrms/3km_10min_flash/expert_1.mdlus CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:078cfc51e040f156cba7a0db326e1a04fc30a6b899e13c0310662bb70009dbba
3
- size 696780988
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e572f356680b1a8a7811916aae38052a9c92c12ca2cd1370828224f334ac5dea
3
+ size 696781331
checkpoints/mrms/3km_10min_flash/expert_2.mdlus CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1aea9704cdf1372ef186574619e4671c6b1ed4e4157e571b6ef35f2501e61a9e
3
- size 694418010
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d690fa0e363abaff772978fd662ebe15dfd47f3647acfc196906a9a70e2aa2b8
3
+ size 694418497
mrms_forecast_skill.csv ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Metric,Model,30 min,60 min,120 min,180 min
2
+ Object CSI,StormScope,0.685,0.533,0.335,0.220
3
+ Object CSI,StormScope Flash,0.688,0.532,0.315,0.204
4
+ CSI 40 dBZ,StormScope,0.597,0.479,0.348,0.271
5
+ CSI 40 dBZ,StormScope Flash,0.593,0.464,0.326,0.246
6
+ CSI 55 dBZ,StormScope,0.274,0.170,0.068,0.028
7
+ CSI 55 dBZ,StormScope Flash,0.286,0.167,0.060,0.024
8
+ pFSS 40 dBZ,StormScope,0.907,0.788,0.588,0.449
9
+ pFSS 40 dBZ,StormScope Flash,0.913,0.800,0.593,0.433
10
+ pFSS 55 dBZ,StormScope,0.719,0.502,0.201,0.064
11
+ pFSS 55 dBZ,StormScope Flash,0.735,0.490,0.137,0.036
mrms_forecast_skill.md ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # StormScope Flash: MRMS forecasting skill
2
+
3
+ This note compares the 3 km, 10-minute StormScope baseline with its distilled
4
+ StormScope Flash variant for composite radar reflectivity. Higher scores are better;
5
+ 1 is perfect. StormScope values are the published baseline scores; Flash values
6
+ come from the archived full-domain evaluation.
7
+
8
+ ## Evaluation
9
+
10
+ The evaluation uses 929 WoFS-catalog initializations from March 4 through August 9,
11
+ 2025. Verification is observed MRMS composite reflectivity on the 3 km model grid
12
+ at matching valid times. Flash runs coupled GOES and MRMS/GLM forecasts over the
13
+ full 1024 x 1792 domain, then selects each catalog region for scoring. Forecasts
14
+ contain 18 ten-minute leads and 16 ensemble members; GOES and MRMS use five and
15
+ seven network calls per lead, respectively.
16
+
17
+ Object CSI and neighborhood CSI use member 0, not the ensemble mean. pFSS uses all
18
+ 16 members. Contingency counts and pFSS numerator/denominator terms are summed
19
+ over cases at each lead before computing the score; these are not averages of
20
+ individual case scores.
21
+
22
+ ## Metrics
23
+
24
+ - **Object CSI** measures storm detection using connected regions at or above
25
+ 40 dBZ, each containing a 45 dBZ core and at least 12 grid cells (108 km²).
26
+ One-to-one matching uses centroid and boundary distances with 40 km cutoffs,
27
+ area-ratio weighting, five matching passes, and total interest greater than 0.2.
28
+ CSI = hits / (hits + false alarms + misses). Redundant unmatched objects
29
+ classified as extras are excluded. The table pools all storm ages.
30
+ - **CSI at 40/55 dBZ** is neighborhood-grid CSI, not object CSI or exact-pixel CSI.
31
+ Forecast and observation exceedances are each dilated with an 11 x 11 cell
32
+ (33 x 33 km) window before forming the same contingency ratio.
33
+ - **pFSS at 40/55 dBZ** compares neighborhood exceedance fractions. Forecast
34
+ fractions are averaged across the 16 members and compared with the observed
35
+ fraction: pFSS = 1 - sum((forecast fraction - observed fraction)²) /
36
+ sum(forecast fraction² + observed fraction²). The window is also 11 x 11 cells.
37
+
38
+ ## Results
39
+
40
+ | Metric | Model | +30 min | +60 min | +120 min | +180 min |
41
+ |---|---|---:|---:|---:|---:|
42
+ | Object CSI | StormScope | 0.685 | 0.533 | 0.335 | 0.220 |
43
+ | Object CSI | StormScope Flash | 0.688 | 0.532 | 0.315 | 0.204 |
44
+ | CSI 40 dBZ | StormScope | 0.597 | 0.479 | 0.348 | 0.271 |
45
+ | CSI 40 dBZ | StormScope Flash | 0.593 | 0.464 | 0.326 | 0.246 |
46
+ | CSI 55 dBZ | StormScope | 0.274 | 0.170 | 0.068 | 0.028 |
47
+ | CSI 55 dBZ | StormScope Flash | 0.286 | 0.167 | 0.060 | 0.024 |
48
+ | pFSS 40 dBZ | StormScope | 0.907 | 0.788 | 0.588 | 0.449 |
49
+ | pFSS 40 dBZ | StormScope Flash | 0.913 | 0.800 | 0.593 | 0.433 |
50
+ | pFSS 55 dBZ | StormScope | 0.719 | 0.502 | 0.201 | 0.064 |
51
+ | pFSS 55 dBZ | StormScope Flash | 0.735 | 0.490 | 0.137 | 0.036 |
52
+
53
+ These results describe the archived evaluation, not a new evaluation of the
54
+ repackaged Earth2Studio checkpoints. Baseline and Flash scoring conventions were
55
+ not fully harmonized, and statistical equivalence was not assessed.
56
+
57
+ [Download table values](mrms_forecast_skill.csv).
registry.json CHANGED
@@ -157,8 +157,8 @@
157
  "source_checkpoint": "checkpoint_92800.pt",
158
  "source_sha256": "3185d14dc34549b52e02d93bf31fdc89b2ae78b325e963786736894028bfda47",
159
  "source_size_bytes": 2093144509,
160
- "deployment_sha256": "6b8cb151b07c9b754f6422f3f7a763e5154a2165d2482e2b9db37c46f2e0c362",
161
- "deployment_size_bytes": 697967288,
162
  "deployment_format": "flash-mdlus-v1",
163
  "tensor_parity": "exact"
164
  },
@@ -168,8 +168,8 @@
168
  "source_checkpoint": "checkpoint_88800.pt",
169
  "source_sha256": "01d1aa63ce440602fd9a56c5692450ed90e943f8e9745760680054bdf66f22e2",
170
  "source_size_bytes": 2112043773,
171
- "deployment_sha256": "4f53361f589ddb3ee4097e4d362d59c027b8551022ee42bb81460df1c3ec11b4",
172
- "deployment_size_bytes": 704267546,
173
  "deployment_format": "flash-mdlus-v1",
174
  "tensor_parity": "exact"
175
  },
@@ -179,8 +179,8 @@
179
  "source_checkpoint": "checkpoint_82400.pt",
180
  "source_sha256": "a7a1eef05f21ebf64b2eeca6989cbab60da4e471c3c91e8b8cd753730ff37696",
181
  "source_size_bytes": 2168741437,
182
- "deployment_sha256": "75aeb03cddd5596570a9e1feaa6444173abc24ae736b84c63daf74218cc07007",
183
- "deployment_size_bytes": 723168328,
184
  "deployment_format": "flash-mdlus-v1",
185
  "tensor_parity": "exact"
186
  }
@@ -328,8 +328,8 @@
328
  "source_checkpoint": "checkpoint_76400.pt",
329
  "source_sha256": "813d47717b7be8325c36d59292a4895d61a1b152bb1739fba5143de546e4034e",
330
  "source_size_bytes": 2082983991,
331
- "deployment_sha256": "1aea9704cdf1372ef186574619e4671c6b1ed4e4157e571b6ef35f2501e61a9e",
332
- "deployment_size_bytes": 694418010,
333
  "deployment_format": "flash-mdlus-v1",
334
  "tensor_parity": "exact"
335
  },
@@ -339,8 +339,8 @@
339
  "source_checkpoint": "checkpoint_84400.pt",
340
  "source_sha256": "f9767d0ac01ab61094559b9cc445c4c0e8a312a9ebe059bfc3af650182a9329d",
341
  "source_size_bytes": 2090071415,
342
- "deployment_sha256": "078cfc51e040f156cba7a0db326e1a04fc30a6b899e13c0310662bb70009dbba",
343
- "deployment_size_bytes": 696780988,
344
  "deployment_format": "flash-mdlus-v1",
345
  "tensor_parity": "exact"
346
  },
@@ -350,8 +350,8 @@
350
  "source_checkpoint": "checkpoint_80000.pt",
351
  "source_sha256": "0736cc36330ac87f7f71f0a5b1b7431a094f6f947aebf5063813bf01f4e2e8e1",
352
  "source_size_bytes": 2111333559,
353
- "deployment_sha256": "7c3112543db0e4cf299fea12caec180508251230822cc975891079ba3d19e29e",
354
- "deployment_size_bytes": 703869930,
355
  "deployment_format": "flash-mdlus-v1",
356
  "tensor_parity": "exact"
357
  }
@@ -381,4 +381,4 @@
381
  "glm_preprocessing": "bilinear_raw_counts_then_log1p",
382
  "coordinate": "edm_linear_t"
383
  }
384
- }
 
157
  "source_checkpoint": "checkpoint_92800.pt",
158
  "source_sha256": "3185d14dc34549b52e02d93bf31fdc89b2ae78b325e963786736894028bfda47",
159
  "source_size_bytes": 2093144509,
160
+ "deployment_sha256": "6e761285cc87e4d8088cb7fdbebd94c7e7e135b81306d4d0152c27649e40eafd",
161
+ "deployment_size_bytes": 697968031,
162
  "deployment_format": "flash-mdlus-v1",
163
  "tensor_parity": "exact"
164
  },
 
168
  "source_checkpoint": "checkpoint_88800.pt",
169
  "source_sha256": "01d1aa63ce440602fd9a56c5692450ed90e943f8e9745760680054bdf66f22e2",
170
  "source_size_bytes": 2112043773,
171
+ "deployment_sha256": "0c860c5ef5088ea7a2718231d50c5026438d5d99e774e28db64845c3ee29a31e",
172
+ "deployment_size_bytes": 704268145,
173
  "deployment_format": "flash-mdlus-v1",
174
  "tensor_parity": "exact"
175
  },
 
179
  "source_checkpoint": "checkpoint_82400.pt",
180
  "source_sha256": "a7a1eef05f21ebf64b2eeca6989cbab60da4e471c3c91e8b8cd753730ff37696",
181
  "source_size_bytes": 2168741437,
182
+ "deployment_sha256": "de2f5a2127f62d6dd868a8c4043c033604b77ca5492b9cd0de9aa5a6cd277e83",
183
+ "deployment_size_bytes": 723168495,
184
  "deployment_format": "flash-mdlus-v1",
185
  "tensor_parity": "exact"
186
  }
 
328
  "source_checkpoint": "checkpoint_76400.pt",
329
  "source_sha256": "813d47717b7be8325c36d59292a4895d61a1b152bb1739fba5143de546e4034e",
330
  "source_size_bytes": 2082983991,
331
+ "deployment_sha256": "d690fa0e363abaff772978fd662ebe15dfd47f3647acfc196906a9a70e2aa2b8",
332
+ "deployment_size_bytes": 694418497,
333
  "deployment_format": "flash-mdlus-v1",
334
  "tensor_parity": "exact"
335
  },
 
339
  "source_checkpoint": "checkpoint_84400.pt",
340
  "source_sha256": "f9767d0ac01ab61094559b9cc445c4c0e8a312a9ebe059bfc3af650182a9329d",
341
  "source_size_bytes": 2090071415,
342
+ "deployment_sha256": "e572f356680b1a8a7811916aae38052a9c92c12ca2cd1370828224f334ac5dea",
343
+ "deployment_size_bytes": 696781331,
344
  "deployment_format": "flash-mdlus-v1",
345
  "tensor_parity": "exact"
346
  },
 
350
  "source_checkpoint": "checkpoint_80000.pt",
351
  "source_sha256": "0736cc36330ac87f7f71f0a5b1b7431a094f6f947aebf5063813bf01f4e2e8e1",
352
  "source_size_bytes": 2111333559,
353
+ "deployment_sha256": "322228b92fbdbcca34c3ef50c2e15d37e8b0ebf22b4e0c15992e9aefd384fb40",
354
+ "deployment_size_bytes": 703869841,
355
  "deployment_format": "flash-mdlus-v1",
356
  "tensor_parity": "exact"
357
  }
 
381
  "glm_preprocessing": "bilinear_raw_counts_then_log1p",
382
  "coordinate": "edm_linear_t"
383
  }
384
+ }