RiemannONets: Interpretable Neural Operators for Riemann Problems
arXiv:2401.08886 · doi:10.1016/j.cma.2024.116996
Abstract
Developing the proper representations for simulating high-speed flows with strong shock waves, rarefactions, and contact discontinuities has been a long-standing question in numerical analysis. Herein, we employ neural operators to solve Riemann problems encountered in compressible flows for extreme pressure jumps (up to pressure ratio). In particular, we first consider the DeepONet that we train in a two-stage process, following the recent work of \cite{lee2023training}, wherein the first stage, a basis is extracted from the trunk net, which is orthonormalized and subsequently is used in the second stage in training the branch net. This simple modification of DeepONet has a profound effect on its accuracy, efficiency, and robustness and leads to very accurate solutions to Riemann problems compared to the vanilla version. It also enables us to interpret the results physically as the hierarchical data-driven produced basis reflects all the flow features that would otherwise be introduced using ad hoc feature expansion layers. We also compare the results with another neural operator based on the U-Net for low, intermediate, and very high-pressure ratios that are very accurate for Riemann problems, especially for large pressure ratios, due to their multiscale nature but computationally more expensive. Overall, our study demonstrates that simple neural network architectures, if properly pre-trained, can achieve very accurate solutions of Riemann problems for real-time forecasting. The source code, along with its corresponding data, can be found at the following URL: https://github.com/apey236/RiemannONet/tree/main
References in corpus (15)
- DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operators
- Gaussian Error Linear Units (GELUs)
- A guide to convolution arithmetic for deep learning
- Adaptive activation functions accelerate convergence in deep and physics-informed neural networks
- A physics-informed variational DeepONet for predicting the crack path in brittle materials
- DeepM&Mnet: Inferring the electroconvection multiphysics fields based on operator approximation by neural networks
- Operator learning for predicting multiscale bubble growth dynamics
- SVD Perspectives for Augmenting DeepONet Flexibility and Interpretability
- Scalable Uncertainty Quantification for Deep Operator Networks using Randomized Priors
- Wavelet neural operator: a neural operator for parametric partial differential equations
- U-NO: U-shaped Neural Operators
- Towards Multi-spatiotemporal-scale Generalized PDE Modeling
- High-Order Methods for Hypersonic Flows with Strong Shocks and Real Chemistry
- Multiscale DeepONet for Nonlinear Operators in Oscillatory Function Spaces for Building Seismic Wave Responses
- Real-time Inference and Extrapolation via a Diffusion-inspired Temporal Transformer Operator (DiTTO)