SYNTOLOGY HomeExplorerAtlasCodeMethodologyAboutDevelopersFeedPricing
Paper · 2212.02998 · NeurIPS · 2022

Super-resolution Probabilistic Rain Prediction from Satellite Data Using 3D U-Nets and EarthFormers

Yang Li, Haiyu Dong, Pete Luferenko, Jonathan Weyn, Zuliang Fang

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 20 functions out of this paper's own repositories and ran 18 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

RepositoryRoleRan
bugsuse/weather4cast-2022-stage2 canonical 18 of 20
FunctionStatusWhere it lives
combine_metrics Ran bugsuse/weather4cast-2022-stage2/utils/evaluate.py
code served (permissive licence) · get_code("cef90636faea5cef")
compute_cuboid_self_attention_mask Ran bugsuse/weather4cast-2022-stage2/models/cuboid_transformer.py
code served (permissive licence) · get_code("767ad1f13b2a971b")
cuboid_reorder Ran bugsuse/weather4cast-2022-stage2/models/cuboid_transformer.py
code served (permissive licence) · get_code("bff27f9560afbd89")
cuboid_reorder_reverse Ran bugsuse/weather4cast-2022-stage2/models/cuboid_transformer.py
code served (permissive licence) · get_code("84f8ab35b7d0471d")
full_attention Ran bugsuse/weather4cast-2022-stage2/models/cuboid_transformer_patterns.py
code served (permissive licence) · get_code("a3015a205293148b")
get_activation Ran bugsuse/weather4cast-2022-stage2/models/utils.py
code served (permissive licence) · get_code("8c21cdd91cd386e3")
get_confusion_matrix Ran bugsuse/weather4cast-2022-stage2/utils/evaluate.py
code served (permissive licence) · get_code("90422cc65b7fc13e")
get_conv Ran bugsuse/weather4cast-2022-stage2/models/baseline_UNET3D.py
code served (permissive licence) · get_code("d473e9bce385040b")
get_convtranspose Ran bugsuse/weather4cast-2022-stage2/models/baseline_UNET3D.py
code served (permissive licence) · get_code("a3169a7b1435e0dc")
get_lossfx Ran bugsuse/weather4cast-2022-stage2/utils/losses.py
code served (permissive licence) · get_code("d29a09d87d16a364")
get_maxpool Ran bugsuse/weather4cast-2022-stage2/models/baseline_UNET3D.py
code served (permissive licence) · get_code("a12e2f04ad60d321")
get_norm_layer Ran bugsuse/weather4cast-2022-stage2/models/utils.py
code served (permissive licence) · get_code("50a5a499347fcf81")
mask_evaluation Ran bugsuse/weather4cast-2022-stage2/utils/metrics.py
code served (permissive licence) · get_code("87b33d55144cd964")
read_samples_ids Ran bugsuse/weather4cast-2022-stage2/utils/data_utils.py
code served (permissive licence) · get_code("88a2f37600b6e67d")
recall_precision_f1_acc Ran bugsuse/weather4cast-2022-stage2/utils/evaluate.py
code served (permissive licence) · get_code("7c6985027e3143dc")
round_to Ran bugsuse/weather4cast-2022-stage2/models/utils.py
code served (permissive licence) · get_code("50449b18203e5788")
self_axial Ran bugsuse/weather4cast-2022-stage2/models/cuboid_transformer_patterns.py
code served (permissive licence) · get_code("6bc7869a75f3e2db")
self_video_swin Ran bugsuse/weather4cast-2022-stage2/models/cuboid_transformer_patterns.py
code served (permissive licence) · get_code("22d035b51d33b141")
generate_and_cache_sequences Not yet run bugsuse/weather4cast-2022-stage2/utils/data_utils.py
code served (permissive licence) · get_code("40d07fba28e6e44c")
load_sample_ids Not yet run bugsuse/weather4cast-2022-stage2/utils/data_utils.py
code served (permissive licence) · get_code("2e0446710911b825")

Repositories linked to this paper

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

Accurate and timely rain prediction is crucial for decision making and is also a challenging task. This paper presents a solution which won the 2 nd prize in the Weather4cast 2022 NeurIPS competition using 3D U-Nets and EarthFormers for 8-hour probabilistic rain prediction based on multi-band satellite images. The spatial context effect of the input satellite image has been deeply explored and optimal context range has been found. Based on the imbalanced rain distribution, we trained multiple models with different loss functions. To further improve the model performance, multi-model ensemble and threshold optimization were used to produce the final probabilistic rain prediction. Experiment results and leaderboard scores demonstrate that optimal spatial context, combined loss function, multi-model ensemble, and threshold optimization all provide modest model gain. A permutation test was used to analyze the effect of each satellite band on rain prediction, and results show that satellite bands signifying cloudtop phase (8.7 µm) and cloud-top height (10.8 and 13.4 µm) are the best predictors for rain prediction. The source code is available at https://github.com/bugsuse/weather4cast-2022-stage2.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2212.02998")
get_code_for_paper("2212.02998")
have("2212.02998")

Connect an agent — have() is free.