GNN Surrogates Forecast CO2 Migration in Complex Geological Formations

Rodrigo S. Luna, Thiago H. N. Coelho, Luiz S. L. Neto, Roberto M. Velho, Adriano M. A. Cortes, Renato N. Elias, Alexandre G. Evsukoff, Fernando A. Rochinha, Mauricio Araya-Polo, Herve Gross, Alvaro L. G. A. Coutinho· June 17, 2026 View original

Summary

This research proposes an end-to-end graph neural network surrogate model to forecast CO2 plume migration in geological storage. The method uses an anisotropic message-passing mechanism and autoregressive residual formulation to predict gas saturation and liquid-phase density.

Researchers have developed a novel graph neural network (GNN) surrogate model designed to predict the movement of CO2 plumes within intricate geological formations. This data-driven machine learning approach aims to accurately reproduce the physical behavior of multiphase flows, specifically focusing on CO2 storage scenarios. The GNN model reformulates geological benchmarks as graphs, where computational cells are nodes and interactions are edges. It incorporates an anisotropic message-passing mechanism, which accounts for directional transport influenced by grid geometry, permeability, and geological heterogeneity. The model also uses an autoregressive residual formulation for temporal evolution, trained with multi-step supervision. Evaluated on the SPE11A benchmark, an industry standard for CO2 storage assessment, the proposed GNN surrogate demonstrates competitive forecasting capabilities for key indicators like gas saturation and liquid-phase density. It maintains moderate cumulative errors over extended prediction horizons, offering a promising tool for monitoring CO2 storage.

Why it matters

This technology offers a faster and more efficient way to monitor and predict CO2 storage, crucial for the success and safety of carbon capture and storage (CCS) initiatives. Professionals in energy and environmental sectors can leverage such models for better risk assessment and operational planning.

How to implement this in your domain

  1. 1Integrate GNN surrogates into existing CO2 storage simulation workflows to accelerate prediction times.
  2. 2Validate the model's predictions against real-world geological data and traditional simulation methods for accuracy.
  3. 3Develop user interfaces for geologists and engineers to easily input parameters and interpret CO2 migration forecasts.
  4. 4Utilize the forecasting capabilities to optimize injection strategies and monitor potential leakage risks in storage sites.

Who benefits

EnergyEnvironmental ConsultingOil & GasGovernment (Regulatory)

Key takeaways

  • A new GNN model accurately forecasts CO2 plume migration in complex geological formations.
  • The model uses an anisotropic message-passing mechanism to capture geological heterogeneity.
  • It provides competitive predictions for gas saturation and liquid-phase density over long horizons.
  • This approach significantly reduces computational costs compared to traditional simulations.

Original post by Rodrigo S. Luna, Thiago H. N. Coelho, Luiz S. L. Neto, Roberto M. Velho, Adriano M. A. Cortes, Renato N. Elias, Alexandre G. Evsukoff, Fernando A. Rochinha, Mauricio Araya-Polo, Herve Gross, Alvaro L. G. A. Coutinho

"arXiv:2606.17180v1 Announce Type: new Abstract: This chapter discusses how a data-driven machine learning approach can reproduce key aspects of the physical behavior of multiphase flows in complex geological formations. We propose an end-to-end graph neural surrogate tailored to…"

View on X

Originally posted by Rodrigo S. Luna, Thiago H. N. Coelho, Luiz S. L. Neto, Roberto M. Velho, Adriano M. A. Cortes, Renato N. Elias, Alexandre G. Evsukoff, Fernando A. Rochinha, Mauricio Araya-Polo, Herve Gross, Alvaro L. G. A. Coutinho on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses