Machine learning-based moment closure model for the semiconductor Boltzmann equation with uncertainties

Abstract: In this paper, we propose a ML-based moment closure model for the linearized Boltzmann equation of semiconductor devices, addressing both the deterministic and stochastic settings. Our approach leverages neural networks to learn the spatial gradient of the unclosed highest-order moment, enabling effective training through natural output normalization. For the deterministic problem, to ensure global hyperbolicity and stability, we derive and apply the constraints that enforce symmetrizable hyperbolicity of the system. For the stochastic problem, we adopt the generalized polynomial chaos (gPC)-based stochastic Galerkin method to discretize the random variables, resulting in a system for which the approach in the deterministic case can be used similarly. Several numerical experiments will be shown to demonstrate the effectiveness and accuracy of our ML-based moment closure model for the linear semiconductor Boltzmann equation with (or without) uncertainties.