Hurricane Katrina killed hundreds of people as it made landfall on New Orleans in 2005 - many of these deaths could have been avoided if alerts had been given one day earlier. Accurate weather forecasts are really life-saving.
๐ฅย Now, NASA and IBM just dropped a game-changing new model: the first ever foundation model for weather! This means, it's the first time we have a generalist model not restricted to one task, but able to predict 160 weather variables!
Prithvi WxC (Prithvi, โเคชเฅเคฅเฅเคตเฅโ, is the Sanskrit name for Earth) - is a 2.3 billion parameter model, with an architecture close to previous vision transformers like Hiera.
๐กย But it comes with some important tweaks: under the hood, Prithvi WxC uses a clever transformer-based architecture with 25 encoder and 5 decoder blocks. It alternates between "local" and "global" attention to capture both regional and global weather patterns.
๐๐ฒ๐ ๐ถ๐ป๐๐ถ๐ด๐ต๐๐: ๐ฎ Nails short-term forecasts - Prithvi WxC crushed it on 6-12 hour predictions, even outperforming some traditional numerical weather models ๐ Tracks hurricanes like a champ - For Hurricane Ida, it predicted the landfall location within 5 km (vs 20+ km errors from other AI models), which is a huge progress! ๐ 6x downscaling power - Can zoom in on weather data to 6x higher resolution with 4x lower error than basic methods ๐ Models elusive gravity waves - Accurately simulates these crucial but hard-to-capture atmospheric oscillations
As climate change intensifies, tools like Prithvi WxC will become more and more crucial to avoid disasters!
Details I am still rigorously testing different hyperparameters and comparing impact of each one to find the best workflow So far done 16 different full trainings and completing 8 more at the moment I am using my poor overfit 15 images dataset for experimentation (4th image) I have already proven that when I use a better dataset it becomes many times betters and generate expressions perfectly Here example case : https://www.reddit.com/r/FluxAI/comments/1ffz9uc/tried_expressions_with_flux_lora_training_with_my/ Conclusions When the results are analyzed, Fine Tuning is way lesser overfit and more generalized and better quality In first 2 images, it is able to change hair color and add beard much better, means lesser overfit In the third image, you will notice that the armor is much better, thus lesser overfit I noticed that the environment and clothings are much lesser overfit and better quality Disadvantages Kohya still doesnโt have FP8 training, thus 24 GB GPUs gets a huge speed drop Moreover, 48 GB GPUs has to use Fused Back Pass optimization, thus have some speed drop 16 GB GPUs gets way more aggressive speed drop due to lack of FP8 Clip-L and T5 trainings still not supported Speeds Rank 1 Fast Config โ uses 27.5 GB VRAM, 6.28 second / it (LoRA is 4.85 second / it) Rank 1 Slower Config โ uses 23.1 GB VRAM, 14.12 second / it (LoRA is 4.85 second / it) Rank 1 Slowest Config โ uses 15.5 GB VRAM, 39 second / it (LoRA is 6.05 second / it) Final Info Saved checkpoints are FP16 and thus 23.8 GB (no Clip-L or T5 trained) According to the Kohya, applied optimizations doesnโt change quality so all configs are ranked as Rank 1 at the moment I am still testing whether these optimizations make any impact on quality or not