Skip to content

Awesome-AI4Earth: a curated list of machine learning in Earth System, especially for weather and climate.

Notifications You must be signed in to change notification settings

taohan10200/Awesome_AI4Earth

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 

Repository files navigation

Awesome-ai4earth Awesome

🔥 Deep learning has been widely explored almost in all research filed. Here is a curated list of papers about deep learning methods in Earth System, especially relating to weather prediction. It also contains frameworks for data-driven Numerical Weather Prediciton (NWP) training, tools to deploy weather prediction, courses and tutorials about Ai4earth and all publicly available weather prediction checkpoints and APIs.

Table of Content

Updates

  • [2023-11-07] Creat this project, add some big model for atmospherical modeling!

ToDos

  • Add more paper, datasets, research directions for ai4earth ✨Contributions Wanted

Also check out the project that I am currently working on: EarthVision - A deep learniong framwork for Numerical Weather Prediction, Earth System Data cmpression, Precipitation Prediction)

Milestone Papers for NWP

Date keywords Institute Paper Publication
2022-02 FourCastNet Nvidia FourCastNet: Accelerating Global High-Resolution Weather Forecasting Using Adaptive Fourier Neural Operators [paper1][paper2] PASC
Dynamic JSON Badge
Dynamic JSON Badge
2022-11 PanguWeather HuaWei Accurate medium-range global weather forecasting with 3D neural networks [paper1][paper2] Nature
Dynamic JSON Badge
2022-12 GraphCast DeepMind GraphCast: Learning skillful medium-range global weather forecasting Science
Dynamic JSON Badge
2023-04 FengWu Shanghai AILab FengWu: Pushing the Skillful Global Medium-range Weather Forecast beyond 10 Days Lead arxiv
Dynamic JSON Badge
2023-06 FuXi Fudan University FuXi: A cascade machine learning forecasting system for 15-day global weather forecast npj
Dynamic JSON Badge

Preprocessing

data-assimilation

  • FengWu-Adas Towards an End-to-End Artificial Intelligence Driven Global Weather Forecasting System. (arxiv, 12/2023)
  • FengWu-4DVar FengWu-4DVar: Coupling the Data-driven Weather Forecasting Model with 4D Variational Assimilation (arxiv, 12/2023)

Forcasting

numerical-weather-prediction

  • NeuralGCM Neural General Circulation Models. (arxiv, 11/2023)
  • GenCast Diffusion-based ensemble forecasting for medium-range weather. (arxiv. 12/2023)
  • MetNet-3 Deep Learning for Day Forecasts from Sparse Observations. (arxiv, 06/2023)
  • ClimaX A foundation model for weather and climate, (ICML, 2023) [project]

precipitation-prediction

Postprocessing

downscaling

  • CorrDiff Generative residual diffusion modeling for km-scale atmospheric downscaling, (arxiv, 09/2023)

bias-correction

  • Two deep learning-based bias-correction pathways improve summer precipitation prediction over China, (Environmental Research Letters, 12/2022) paper

Applications

extreme-weather-and-prediction

  • FuXi-Extreme: Improving extreme rainfall and wind forecasts with diffusion mode. (arxiv, 10/2023)

weather-cliomate-related-application

global-wildfire-prediction

ai4earth-benchmark

WeatherBench 2: A benchmark for the next generation of data-driven global weather models [project] [LeaderBoard] (08/2023)

Other Papers

If you're interested in the field of LLM, you may find the above list of milestone papers helpful to explore its history and state-of-the-art. However, each direction of LLM offers a unique set of insights and contributions, which are essential to understanding the field as a whole. For a detailed list of papers in various subfields, please refer to the following link (it is possible that there are overlaps between different subfields):

(:exclamation: We would greatly appreciate and welcome your contribution to the following list. ❗)

Weather Forecast Leaderboard

The following list makes sure that all LLMs are compared apples to apples.

You may also find these leaderboards helpful:

  • WeatherBench2 Leaderboard - evaluating and comparing various weather forecasting models. displays up-to-date scores of many of the state-of-the-art ML and physics-based models.

Base Weather Forecasting Models

Model Size Architecture Access Date Origin Model License1
PanguWeather ~740M 3D Transformer ckpt 2022-11 Paper BY-NC-SA 4.0
GraphCast unkonwn GNN ckpt 2022-12 Paper Apache 2.0 & BY-NC-SA 4.0
FuXi unkonwn U-Transformer ckpt 2023-06 Paper -

BigModel Training Frameworks

  • DeepSpeed - DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
  • Megatron-DeepSpeed - DeepSpeed version of NVIDIA's Megatron-LM that adds additional support for several features such as MoE model training, Curriculum Learning, 3D Parallelism, and others.
  • FairScale - FairScale is a PyTorch extension library for high performance and large scale training.
  • Megatron-LM - Ongoing research training transformer models at scale.
  • Colossal-AI - Making large AI models cheaper, faster, and more accessible.
  • BMTrain - Efficient Training for Big Models.
  • Mesh Tensorflow - Mesh TensorFlow: Model Parallelism Made Easier.
  • maxtext - A simple, performant and scalable Jax LLM!
  • Alpa - Alpa is a system for training and serving large-scale neural networks.
  • GPT-NeoX - An implementation of model parallel autoregressive transformers on GPUs, based on the DeepSpeed library.

Tools for deploying LLM

  • xx - A distributed multi-model LLM serving system with web UI and OpenAI-compatible RESTful APIs.

Other Awesome Lists

  • xx - A curated (still actively updated) list of practical guide resources of earth-related research.

Contributing

This is an active repository and your contributions are always welcome!

I will keep some pull requests open if I'm not sure if they are awesome for ai4earth, you could vote for them by adding 👍 to them.


If you have any question about this opinionated list, do not hesitate to contact me hantao10200@gmail.com.

Footnotes

  1. This is not legal advice. Please contact the original authors of the models for more information.

About

Awesome-AI4Earth: a curated list of machine learning in Earth System, especially for weather and climate.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published