Spotlight: BRLi and Toulouse INP Develop AI-Based Flood Models Using NVIDIA PhysicsNeMo

Flooding poses a significant threat to 1.5 billion people, making it the most common cause of major natural disasters. Floods cause up to $25 billion in global…

Ram Cherukuri
6 min readintermediate
--
View Original

Overview

The article discusses the collaboration between BRLi and Toulouse INP to develop AI-based flood models using NVIDIA PhysicsNeMo, addressing the limitations of traditional physics-based numerical simulations. The new approach significantly reduces computation time, enabling real-time flood forecasting and enhancing disaster preparedness.

What You'll Learn

1

How to leverage NVIDIA PhysicsNeMo for AI-based flood modeling

2

Why AI can improve real-time flood forecasting compared to traditional methods

3

When to use graph neural networks for hydrological modeling

Prerequisites & Requirements

  • Understanding of hydraulic modeling and AI techniques
  • Familiarity with NVIDIA PhysicsNeMo and Earth-2 platform(optional)
  • Experience with parallel computing and GPU utilization(optional)

Key Questions Answered

How does the AI-based flood forecasting system work?
The AI-based flood forecasting system uses a graph neural network architecture trained on detailed physics models of the Têt River basin. It emulates the physics-based solver, enabling predictions of flooded areas and flow rates in 30-minute increments, achieving results in seconds on a single GPU.
What are the performance improvements of the AI model over traditional methods?
The AI model can perform a 6-hour flood prediction in 19 milliseconds on a single NVIDIA A100 GPU, while traditional physics-based simulations require 12 hours of CPU time. This drastic reduction in computation time allows for real-time forecasting.
What metrics are used to evaluate the AI model's performance?
The evaluation metrics include mean squared error (MSE) for training optimization and critical success index (CSI) for testing, which measures the model's ability to detect and classify flooded areas based on a 5 cm threshold.

Key Statistics & Figures

Global economic damage from floods
$25 billion
This figure represents the annual economic losses caused by flooding worldwide.
Time for AI model to predict flooding
19 ms
The AI model can perform a 6-hour prediction in just 19 milliseconds on a single NVIDIA A100 GPU.
Traditional simulation time
12 hours
A similar simulation using the physics-based numerical model requires 12 hours of CPU time.
Scaling efficiency of training
91%
The model achieved a scaling efficiency of up to 91% when trained on the CALMIP Turpan supercomputer.

Technologies & Tools

AI/ML Framework
Nvidia Physicsnemo
Used for training AI models to emulate physics-based flood solvers.
Cloud Platform
Earth-2
Provides a digital twin environment for AI-augmented forecasting and visualization.
Numerical Simulation Software
Telemac-mascaret
Used for detailed numerical simulations of the Têt River basin.
GPU
Nvidia A100 Tensor Core
Utilized for training the AI model and achieving rapid predictions.

Key Actionable Insights

1
Implementing AI-based flood forecasting can significantly enhance disaster response capabilities.
By utilizing AI models like those developed with NVIDIA PhysicsNeMo, organizations can provide timely flood warnings, which are crucial for saving lives and minimizing economic losses.
2
Utilizing graph neural networks can improve the accuracy of hydrological predictions.
Graph neural networks capture complex spatial and temporal dynamics, making them suitable for modeling intricate hydrological processes that lead to flooding.
3
Parallel training on GPUs can drastically reduce model training time.
The use of up to 14 NVIDIA A100 GPUs allowed the team to achieve near-linear speedup in training, which is essential for developing real-time forecasting systems.

Common Pitfalls

1
Relying solely on simplified models for flood prediction can lead to inaccurate forecasts.
Simplified models may not capture the complexity of real-world scenarios, which is crucial for accurate flood predictions. It's important to use detailed meshes and data to improve model fidelity.

Related Concepts

AI In Environmental Modeling
Hydraulic Modeling Techniques
Real-time Forecasting Systems