In this study, we present a new approach based on Separable Physics-Informed Neural Networks (SPINNs), specifically designed to efficiently solve the BGK model. While the mesh-free nature of Physics-Informed Neural Networks (PINNs) offers significant advantages for handling high-dimensional partial differential equations (PDEs), applying quadrature rules for accurate integral evaluation in the BGK operator can compromise these benefits and increase computational costs. To address this issue, we leverage the canonical polyadic decomposition structure of SPINNs and the linear nature of moment calculation to significantly reduce the computational expense associated with quadrature rules. However, the multi-scale nature of the particle density function poses challenges for precisely approximating macroscopic moments using neural networks. To overcome this, we introduce SPINN-BGK, a specialized variant of SPINNs that fuses SPINNs with Gaussian functions and utilizes a relative loss approach. This modification enables SPINNs to decay as rapidly as Maxwellian distributions, enhancing the accuracy of macroscopic moment approximations. The relative loss design ensures that both large and small-scale features are effectively captured by the SPINNs. The effectiveness of our approach is validated through six numerical experiments, including a complex 3D Riemann problem. These experiments demonstrate the potential of our method to efficiently and accurately tackle intricate challenges in computational physics, offering significant improvements in computational efficiency and solution accuracy.
Kn=0.01
1d_0.01.mp4
Kn=1
1d_1.0.mp4
Kn=0.01
2d_0.01.mp4
Kn=1
2d_1.0.mp4
Kn=0.01
Kn=1
- Python version: 3.12
- Install JAX
- Install additional dependencies: run
pip install -r requirements.txt
Change the arguments, if necessary.
Run python smooth_3d.py --Kn=0.01 --rank=256 for the (3+3+1) smooth problem.
(3+3+1) dimensional problems: run python error_3d.py --problem="smooth" --Kn=1.0
If you find this repository useful in your research, please consider citing us!
@article{oh2025separable,
title={Separable physics-informed neural networks for solving the bgk model of the boltzmann equation},
author={Oh, Jaemin and Cho, Seung Yeon and Yun, Seok-Bae and Park, Eunbyung and Hong, Youngjoon},
journal={SIAM Journal on Scientific Computing},
volume={47},
number={2},
pages={C451--C474},
year={2025},
publisher={SIAM}
}