Physics-informed convolutional neural networks for fluid flow through porous media

· Source: cs.LG updates on arXiv.org · Field: Science & Research — Engineering & Applied Sciences, Mathematics & Computational Sciences, Artificial Intelligence & Machine Learning · Depth: Expert, extended

Summary

A neural-network-based framework for predicting pore-scale velocity fields in porous media is presented, utilizing a convolutional encoder-decoder (U-Net) architecture with skip connections. The method incorporates a custom loss function that enforces physical consistency, including terms for incompressibility, no-flow conditions within solids, periodicity constraints, and agreement with the global tortuosity index. The ResNet-101 architecture demonstrated superior performance, achieving a velocity RMSE of 5.1e-03 and a tortuosity R^2 of 0.983. The model exhibits strong generalization capabilities across variations in obstacle geometry, boundary conditions (e.g., pipe-like flows), and porosities within the training range of 0.70 to 0.95, and even to real Li-O2 electrode microstructures. Furthermore, using the network's predictions to initialize Lattice-Boltzmann Method (LBM) simulations significantly accelerated convergence in over 90% of cases, reducing iterations by a median of 50%. Inference time is 5 ms on GPU, substantially faster than LBM's 560 ms on CPU.

Key takeaway

For fluid dynamics researchers or engineers simulating porous media flow, you should consider integrating physics-informed convolutional neural networks. This approach significantly accelerates Lattice-Boltzmann Method (LBM) simulations by providing warm starts, reducing convergence time by a median of 50% in over 90% of cases. You can achieve high accuracy in pore-scale velocity predictions and macroscopic properties like tortuosity and permeability, even for out-of-distribution geometries, using a ResNet-101 backbone and a custom multi-term loss function.

Key insights

Physics-informed CNNs accurately predict fluid flow in porous media, significantly accelerating traditional LBM simulations.

Principles

Method

A U-Net encoder-decoder with skip connections predicts pore-scale velocity from binary geometry. A custom loss function combines velocity MSE with penalties for incompressibility, no-flow in solids, periodicity, and tortuosity matching.

In practice

Topics

Code references

Best for: AI Scientist, Machine Learning Engineer, Research Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.LG updates on arXiv.org.