Volltext-Downloads (blau) und Frontdoor-Views (grau)

Lettuce: PyTorch-based Lattice Boltzmann Framework

  • 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 PyTorch-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.

Export metadata

Additional Services

Search Google Scholar Check availability


Show usage statistics
Document Type:Preprint
Author:Mario Christopher Bedrunka, Dominik Wilde, Martin Kliemank, Dirk Reith, Holger Foysi, Andreas Krämer
ArXiv Id:http://arxiv.org/abs/2106.12929
Date of first publication:2021/06/24
Publication status:Published in International Conference on High Performance Computing (pp. 40-55). Springer, Cham (2021)
Keyword:Automatic Differentiation; Lattice Boltzmann Method Code; Machine learning; Neural networks; Pytorch
Departments, institutes and facilities:Fachbereich Ingenieurwissenschaften und Kommunikation
Institut für Technik, Ressourcenschonung und Energieeffizienz (TREE)
Dewey Decimal Classification (DDC):5 Naturwissenschaften und Mathematik / 51 Mathematik / 510 Mathematik
Entry in this database:2021/07/02
Licence (German):License LogoCreative Commons - CC BY - Namensnennung 4.0 International