Climate modelling gets AI boost

Computer scientists in the US have combined generative AI with physics-based data to develop better ways to model the Earth’s climate.

Rose Yu, a faculty member in the UC San Diego Department of Computer Science and Engineering, and Ph.D. student Salva Ruhling Cachay examine data
Rose Yu, a faculty member in the UC San Diego Department of Computer Science and Engineering, and Ph.D. student Salva Ruhling Cachay examine data - David Baillot/University of California San Diego

Detailed in this paper, the Spherical DYffusion model can project 100 years of climate patterns in 25 hours, a simulation that would take weeks for other models. In addition, existing advanced models run on supercomputers, but the new model can run on GPU clusters in a research lab. 

“Data-driven deep learning models are on the verge of transforming global weather and climate modelling,” the researchers from the University of California San Diego and the Allen Institute for AI, write. 

Climate simulations are currently very expensive to generate because of their complexity. Consequently, scientists and policymakers can only run simulations for a limited amount of time and consider only limited scenarios. 

One of the researchers’ key insights was that generative AI models, such as diffusion models, could be used for ensemble climate projections. They combined this with a Spherical Neural Operator, a neural network model designed to work with data on a sphere. 

The resulting model starts off with knowledge of climate patterns and then applies a series of transformations based on learned data to predict future patterns. 

“One of the main advantages over a conventional diffusion model [DM] is that our model is much more efficient. It may be possible to generate just as realistic and accurate predictions with conventional DMs but not with such speed,” the researchers write. 

MORE FROM ARTIFICIAL INTELLIGENCE

In addition to running much faster than state of the art, the model is also nearly as accurate without being as computationally expensive. 

There are some limitations to the model that researchers aim to overcome in its next iterations, such as including more elements in their simulations. Next steps include simulating how the atmosphere responds to CO2. 

“We emulated the atmosphere, which is one of the most important elements in a climate model,” said Rose Yu, a faculty member in the UC San Diego Department of Computer Science and Engineering and one of the paper’s senior authors.

The work stems from an internship that one of Yu’s Ph.D. students, Salva Ruhling Cachay, did at the Allen Institute for AI (Ai2). 

The research team is presenting their work at the NeurIPS conference 2024, December 9 to 15 in Vancouver, Canada.