Improving CFD simulations by local machine-learned correction
Peetak Mitra, Majid Haghshenas, Niccolo Dal Santo, Conor Daly, David P. Schmidt
TL;DR
The paper tackles the high computational cost of CFD by introducing a neural-network–based local discretization-error correction that augments coarse-mesh simulations. The correction is learned from fine-mesh ground truth and applied as a source term in the Navier–Stokes equations, enabling low-cost coarse simulations to approach high-fidelity results. Key contributions include a 3D engineering-relevant demonstration with numerical stability, a hybrid $L_2$/$L_1$ loss and Bayesian hyperparameter optimization, and reported $3$–$5\times$ speedups with minimal accuracy loss. This work strengthens the CFD cost/accuracy trade-off and points to scalable improvements for design-space explorations, especially when applied to larger problems and near-wall dynamics in turbulent flows.
Abstract
High-fidelity computational fluid dynamics (CFD) simulations for design space explorations can be exceedingly expensive due to the cost associated with resolving the finer scales. This computational cost/accuracy trade-off is a major challenge for modern CFD simulations. In the present study, we propose a method that uses a trained machine learning model that has learned to predict the discretization error as a function of largescale flow features to inversely estimate the degree of lost information due to mesh coarsening. This information is then added back to the low-resolution solution during runtime, thereby enhancing the quality of the under-resolved coarse mesh simulation. The use of a coarser mesh produces a non-linear benefit in speed while the cost of inferring and correcting for the lost information has a linear cost. We demonstrate the numerical stability of a problem of engineering interest, a 3D turbulent channel flow. In addition to this demonstration, we further show the potential for speedup without sacrificing solution accuracy using this method, thereby making the cost/accuracy trade-off of CFD more favorable.
