Deep learning accelerated solutions of incompressible Navier-Stokes equations on non-uniform Cartesian grids

Abstract

In incompressible flow simulations, non-uniform grids efficiently capture localized flow features; however, their spatially varying resolutions severely exacerbate computational complexity. The pressure Poisson equation (PPE) formulated on these grids yields highly complex linear systems, forming the primary computational bottleneck in fractional step method. To address this, we develop an extended hybrid framework tailored for non-uniform Cartesian grids, integrating deep learning with classical iterative solvers to accelerate PPE solutions. Specifically, the framework employs a deep operator network with a U-Net-based branch network. To effectively capture spatially varying resolutions, we propose a multi-level distance vector map construction strategy that computes discrete grid-spacing information corresponding to each hierarchical level of the U-Net. This grid-spacing information is explicitly fused into feature maps prior to convolution operations. Empowered by this grid-spacing-aware architecture, the framework seamlessly extends to simulate flows interacting with solid structures using a decoupled immersed boundary projection method. By training exclusively on fabricated linear systems rather than conventional flow-dependent datasets, the model generalizes effortlessly across diverse immersed obstacle geometries with fixed network weights. Benchmark results demonstrate that the framework significantly outperforms standalone preconditioned conjugate gradient methods and its standard convolution counterpart, underscoring its exceptional potential for real-world computational fluid dynamics applications.