arXiv subjects

Heming Bai

Publications and source records attributed to Heming Bai.

3 recordsLinked to original sources

arXiv2026

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

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.

physics.flu-dyn↗

arXiv2025

Hybrid deep learning and iterative methods for accelerated solutions of viscous incompressible flow

The pressure Poisson equation, central to the fractional step method in incompressible flow simulations, incurs high computational costs due to the iterative solution of large-scale linear systems. To address this challenge, we introduce HyDEA, a novel framework that synergizes deep learning with classical iterative solvers. It leverages the complementary strengths of a DeepONet - capable of capturing large-scale features of the solution - and the CG or a PCG method, which efficiently resolves fine-scale errors. Specifically, within the framework of line-search methods, the DeepONet predicts search directions to accelerate convergence in solving sparse, symmetric-positive-definite linear systems, while the CG/ PCG method ensures robustness through iterative refinement. The framework seamlessly extends to flows over solid structures via the decoupled immersed boundary projection method. Crucially, the DeepONet is trained on fabricated linear systems rather than flow specific data, endowing it with inherent generalization across geometric complexities and Reynolds numbers without retraining. Benchmarks demonstrate superior efficiency and accuracy of HyDEA over the CG/PCG methods for flows with no obstacles, single or multiple stationary obstacles, and one moving obstacle - using fixed network weights. Remarkably, HyDEA also exhibits super-resolution capability: although the DeepONet is trained on a 128*128 grid for Re=1000, the hybrid solver delivers accurate solutions on a 512*512 grid for Re=10000 via interpolation, despite discretizations mismatch. In contrast, a purely data-driven DeepONet fails for complex flows, underscoring the necessity of hybridizing deep learning with iterative methods. Robustness, efficiency, and generalization across geometries, resolutions, and Reynolds numbers of HyDEA highlight its potential as a transformative solver for real world fluid dynamics problems.

physics.flu-dyn↗

arXiv2024

Data-driven modeling of unsteady flow based on deep operator network

Time-dependent flow fields are typically generated by a computational fluid dynamics (CFD) method, which is an extremely time-consuming process. However, the latent relationship between the flow fields is governed by the Navier-Stokes equations and can be described by an operator. We therefore train a deep operator network, or simply DeepONet, to learn the temporal evolution between flow snapshots. Once properly trained, given a few consecutive snapshots as input, the network has a great potential to generate the next snapshot accurately and quickly. Using the output as a new input, the network iterates the process, generating a series of successive snapshots with little wall time. Specifically, we consider 2D flow around a circular cylinder at Reynolds number 1000, and prepare a set of high-fidelity data using a high-order spectral/hp element method as ground truth. Although the flow fields are periodic, there are many small-scale features in the wake flow that are difficult to generate accurately. Furthermore, any discrepancy between the prediction and the ground truth for the first snapshots can easily accumulate during the iterative process, which eventually amplifies the overall deviations. Therefore, we propose two alternative techniques to improve the training of DeepONet. The first one enhances the feature extraction of the network by harnessing the "multi-head non-local block". The second one refines the network parameters by leveraging the local smooth optimization technique. Both techniques prove to be highly effective in reducing the cumulative errors and our results outperform those of the dynamic mode decomposition method.

physics.flu-dyn↗