期刊
HIGH PERFORMANCE COMPUTING - ISC HIGH PERFORMANCE DIGITAL 2021 INTERNATIONAL WORKSHOPS
卷 12761, 期 -, 页码 40-55出版社
SPRINGER INTERNATIONAL PUBLISHING AG
DOI: 10.1007/978-3-030-90539-2_3
关键词
Lattice Boltzmann method; Pytorch; Machine learning; Neural networks; Automatic differentiation; Computational fluid dynamics; Flow control
The lattice Boltzmann method (LBM) is an efficient simulation technique for computational fluid mechanics that is easily compatible with modern machine learning architectures. Lettuce, a PyTorch-based LBM code, enables GPU accelerated calculations, rapid prototyping of models, and integration with deep learning. Researchers explore the possible connections between machine learning and LBM through the development of neural collision models and demonstration of PyTorch's automatic differentiation benefits in flow control and optimization.
The lattice Boltzmann method (LBM) is an efficient simulation technique for computational fluid mechanics and beyond. It is based on a simple stream-and-collide algorithm on Cartesian grids, which is easily compatible with modern machine learning architectures. While it is becoming increasingly clear that deep learning can provide a decisive stimulus for classical simulation techniques, recent studies have not addressed possible connections between machine learning and LBM. Here, we introduce Lettuce, a Py Torch-based LBM code with a threefold aim. Lettuce enables GPU accelerated calculations with minimal source code, facilitates rapid prototyping of LBM models, and enables integrating LBM simulations with PyTorch's deep learning and automatic differentiation facility. As a proof of concept for combining machine learning with the LBM, a neural collision model is developed, trained on a doubly periodic shear layer and then transferred to a different flow, a decaying turbulence. We also exemplify the added benefit of PyTorch's automatic differentiation framework in flow control and optimization. To this end, the spectrum of a forced isotropic turbulence is maintained without further constraining the velocity field. The source code is freely available from https://github.com/lettucecfd/lettuce.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据