Strategies for multi-case physics-informed neural networks for tube flows: a study using 2D flow scenarios

Hong Shen Wong,Wei Xuan Chan,Bing Huan Li,Choon Hwai Yap

DOI: https://doi.org/10.1038/s41598-024-62117-9

IF: 4.6

2024-05-23

Scientific Reports

Abstract:Fluid dynamics computations for tube-like geometries are crucial in biomedical evaluations of vascular and airways fluid dynamics. Physics-Informed Neural Networks (PINNs) have emerged as a promising alternative to traditional computational fluid dynamics (CFD) methods. However, vanilla PINNs often demand longer training times than conventional CFD methods for each specific flow scenario, limiting their widespread use. To address this, multi-case PINN approach has been proposed, where varied geometry cases are parameterized and pre-trained on the PINN. This allows for quick generation of flow results in unseen geometries. In this study, we compare three network architectures to optimize the multi-case PINN through experiments on a series of idealized 2D stenotic tube flows. The evaluated architectures include the 'Mixed Network', treating case parameters as additional dimensions in the vanilla PINN architecture; the "Hypernetwork", incorporating case parameters into a side network that computes weights in the main PINN network; and the "Modes" network, where case parameters input into a side network contribute to the final output via an inner product, similar to DeepONet. Results confirm the viability of the multi-case parametric PINN approach, with the Modes network exhibiting superior performance in terms of accuracy, convergence efficiency, and computational speed. To further enhance the multi-case PINN, we explored two strategies. First, incorporating coordinate parameters relevant to tube geometry, such as distance to wall and centerline distance, as inputs to PINN, significantly enhanced accuracy and reduced computational burden. Second, the addition of extra loss terms, enforcing zero derivatives of existing physics constraints in the PINN (similar to gPINN), improved the performance of the Mixed Network and Hypernetwork, but not that of the Modes network. In conclusion, our work identified strategies crucial for future scaling up to 3D, wider geometry ranges, and additional flow conditions, ultimately aiming towards clinical utility.

multidisciplinary sciences

What problem does this paper attempt to address?

The main problem this paper attempts to address is improving the efficiency and accuracy of fluid dynamics simulations in tubular structures through multi-case Physics-Informed Neural Networks (PINNs). Specifically, the paper focuses on how to optimize different network architectures so that PINNs can quickly generate flow field results for unseen geometries, thereby reducing the long training times required by traditional Computational Fluid Dynamics (CFD) methods. The paper compares three different network architectures: Mixed Network, Hypernetwork, and Modes Network, and evaluates their performance through a series of idealized 2D narrow tube flow experiments.

The core objectives of the paper are:
1. **Improve simulation efficiency**: By using a multi-case parameterized PINNs approach, the network can quickly generate flow field results for new, unseen geometries after pre-training, without the need for individual training for each new case.
2. **Optimize network architecture**: Compare three different network architectures to identify the one that performs best in terms of accuracy, convergence efficiency, and computational speed.
3. **Enhance network performance**: Further improve the performance of multi-case PINNs by introducing tubular structure-specific coordinate parameters (such as wall distance and centerline distance) and adding additional loss terms (such as gradient loss terms in gPINN).

Ultimately, the paper hopes that this research can provide a foundation for future 3D simulations, a broader range of geometries, and more fluid conditions, thereby advancing clinical applications.