Application of machine learning to model the pressure poisson equation for fluid flow on generic geometries

Abstract

This study addresses the importance of enhancing traditional fluid-flow solvers by introducing a Machine Learning procedure to model pressure fields computed by standard fluid-flow solvers. The conventional approach involves enforcing pressure–velocity coupling through a Poisson equation, combining the Navier–Stokes and continuity equations. The solution to this Poisson equation constitutes a substantial percentage of the overall computational cost in fluid flow simulations, therefore improving its efficiency can yield significant gains in computational speed. The study aims to create a versatile method applicable to any geometry, ultimately providing a more efficient alternative to the conventional pressure solver. Machine Learning models were trained with flow fields generated by a Computational Fluid Dynamics solver applied to the confined flow over multiple geometries, namely wall-bounded cylinders with circular, rectangular, triangular, and plate cross-sections. To achieve applicability to any geometry, a method was developed to estimate pressure fields in fixed-shape blocks sampled from the flow domain and subsequently assemble them to reconstruct the entire physical domain. The model relies on multilayer perceptron neural networks combined with Principal Component Analysis transformations. The developed Machine Learning models achieved acceptable accuracy with errors of around 3%. Furthermore, the model demonstrated enhanced computational efficiency, outperforming the classical PISO algorithm by up to 30 times.