Rapid and accurate permeability prediction in porous media is a critical yet computationally demanding task in fields like geoscience and materials engineering. To address this, a surrogate model based on Fourier Neural Operators, which are neural networks specialized in learning partial differential equations, is introduced. The core concept is to first predict the entire physical pressure field from the input geometry, from which permeability is then derived. This approach grounds the model in the underlying flow physics. FNOs leverage the frequency domain to apply their weights and are independent of input resolution. Additionally, the improved factorized Fourier neural operators (FFNO) with separate convolutions for every dimension and skip connections was compared to the original and a hybrid approach in regard to quality and performance. In a 3D dataset consisting of the flow around randomly shifted spheres an
-
permFNO: Python package for the Fourier neural operator permeability prediction
- data: Handles the datasets, I/O and transformations
- models: Contains the FNO and FFNO together with its sub-parts
- learning: Methods needed for neural network training
- main.py: The main function is used for setup and running the training process
-
lbmpy: Code used for LBM simulation with the package lbmpy
- LBM2D.py: Code for simulating the 2D geometries
- LBM3D.py: Code for simulating the 3D geometries
- permeability_comparison.py: Comparison method for given and calculated values
The datasets used in this research is present at https://data.mendeley.com/preview/29m6gyhy2r?a=ba112c32-49fa-460f-9d76-b2f3ad8818cb
The thesis and its presentation slides are available as PDFs.