RelaxNet: A structure-preserving neural network to approximate the Boltzmann collision operator

被引:2
作者
Xiao, Tianbai [1 ]
Frank, Martin [2 ]
机构
[1] Chinese Acad Sci, Inst Mech, State Key Lab High Temp Gas Dynam, Beijing, Peoples R China
[2] Karlsruhe Inst Technol, Steinbuch Ctr Comp, Karlsruhe, Germany
关键词
Kinetic theory; Computational fluid dynamics; Scientific machine learning; Artificial neural network; MAXIMUM-ENTROPY; MOMENT; MODELS; CLOSURE;
D O I
10.1016/j.jcp.2023.112317
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
The extremely high dimensionality and nonlinearity in the Boltzmann equation bring tremendous difficulties to the study of rarefied gas dynamics. This paper addresses a neural network-based surrogate model that provides a structure-preserving approximation for the fivefold collision integral. The notion originates from the similarity in structure between the BGK-type relaxation model and residual neural network (ResNet) when a particle distribution function is treated as the input to the neural network function. Therefore, we extend the ResNet architecture and construct what we call the relaxation neural network (RelaxNet). Specifically, two feed-forward neural networks with physics informed connections and activations are introduced as building blocks in RelaxNet, which provide bounded and physically realizable approximations of the equilibrium distribution and velocity-dependent relaxation time respectively. The evaluation of the collision term is thus significantly accelerated due to the fact that the convolution in the fivefold integral is replaced by tensor multiplication in the neural network. We fuse the mechanical advection operator and the RelaxNet-based collision operator into a unified model named the universal Boltzmann equation (UBE). We prove that UBE preserves the key structural properties in a many-particle system, i.e., positivity, conservation, invariance, H-theorem, and correct fluid dynamic limit. These properties promise that RelaxNet is superior to strategies that naively approximate the right-hand side of the Boltzmann equation using a machine learning model. A novel sampling strategy based on closure hierarchies of the moment system of the Boltzmann equation is developed to generate reliable and unbiased sampling for the supervised learning. The construction of the RelaxNet-based UBE and its solution algorithm are demonstrated in detail. Several numerical experiments, where the ground-truth datasets are produced by the Shakhov model, velocity-dependent & nu;-BGK model, and the full Boltzmann equation, are investigated. The capability of the current approach for simulating non-equilibrium flow physics is validated through satisfactory in and out-of-distribution performance. The open-source codes to reproduce the numerical results are available under the MIT license [1].& COPY; 2023 Elsevier Inc. All rights reserved.
引用
收藏
页数:31
相关论文
共 52 条
  • [1] Abelson H., 1996, Structure and Interpretation of Computer Programs
  • [2] Fast evaluation of the Boltzmann collision operator using data driven reduced order models
    Alekseenko, Alexander
    Martin, Robert
    Wood, Aihua
    [J]. JOURNAL OF COMPUTATIONAL PHYSICS, 2022, 470
  • [3] A REGULARIZED ENTROPY-BASED MOMENT METHOD FOR KINETIC EQUATIONS
    Alldredge, Graham W.
    Frank, Martin
    Hauck, Cory D.
    [J]. SIAM JOURNAL ON APPLIED MATHEMATICS, 2019, 79 (05) : 1627 - 1653
  • [4] Nabian MA, 2019, Arxiv, DOI arXiv:1810.05547
  • [5] IMPLICIT EXPLICIT METHODS FOR TIME-DEPENDENT PARTIAL-DIFFERENTIAL EQUATIONS
    ASCHER, UM
    RUUTH, SJ
    WETTON, BTR
    [J]. SIAM JOURNAL ON NUMERICAL ANALYSIS, 1995, 32 (03) : 797 - 823
  • [6] Babovsky Hans, 1998, Die Boltzmann-Gleichung: Modellbildung-Numerik-Anwendungen
  • [7] Bekkers E, 2021, Arxiv, DOI arXiv:1909.12057
  • [8] A MODEL FOR COLLISION PROCESSES IN GASES .1. SMALL AMPLITUDE PROCESSES IN CHARGED AND NEUTRAL ONE-COMPONENT SYSTEMS
    BHATNAGAR, PL
    GROSS, EP
    KROOK, M
    [J]. PHYSICAL REVIEW, 1954, 94 (03): : 511 - 525
  • [9] Bishop C. M., 1995, Neural networks for pattern recognition
  • [10] Bouchut Francois, 2000, KINETIC EQUATIONS AS