ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
70
Citations
1
Influential Citations
Nature Communications
Venue
2024
Year
We introduce generative models for accelerating simulations of high-dimensional systems through learning and evolving their effective dynamics. In the proposed Generative Learning of Effective Dynamics (G-LED), instances of high dimensional data are down sampled to a lower dimensional manifold that is evolved through an auto-regressive attention mechanism. In turn, Bayesian diffusion models, that map this low-dimensional manifold onto its corresponding high-dimensional space, operate on batches of physics correlated, time sequences of data and capture the statistics of the system dynamics. We demonstrate the capabilities and drawbacks of G-LED in simulations of several benchmark systems, including the Kuramoto-Sivashinsky (KS) equation, two-dimensional high Reynolds number flow over a backward-facing step, and simulations of three-dimensional turbulent channel flow. The results demonstrate that generative learning offers new frontiers for the accurate forecasting of the statistical properties of high-dimensional systems at a reduced computational cost. The forecasting of critical phenomena in complex systems governed by partial differential equations remains challenging and computationally expensive. The authors propose a generative learning approach for the forecasting of the statistical properties of high-dimensional systems at a reduced computational cost.
This paper addresses a critical bottleneck in computational science: the high cost of simulating complex, high-dimensional systems governed by partial differential equations (PDEs). Traditional numerical methods, while accurate, become prohibitively expensive for systems like turbulent flows or chaotic dynamics. The authors propose G-LED, a generative learning framework that learns the effective dynamics of such systems on a lower-dimensional manifold, then uses diffusion models to reconstruct the full state. This approach is significant because it leverages recent advances in generative AI—specifically diffusion models and attention mechanisms—to achieve both speed and statistical accuracy. For AI practitioners, this work demonstrates how generative models can be applied beyond image and text generation to solve fundamental problems in physics and engineering.
The paper demonstrates that G-LED accurately forecasts statistical properties (e.g., mean, variance, spectra) of all three benchmark systems. For the Kuramoto-Sivashinsky equation, the model captures chaotic spatiotemporal patterns. For 2D backward-facing step flow at high Reynolds number, it reproduces recirculation zones and turbulence statistics. For 3D turbulent channel flow, it predicts mean velocity profiles and Reynolds stresses. The key metric is computational cost reduction: G-LED achieves these forecasts at a fraction of the cost of direct numerical simulation (DNS), though exact speedup factors are not provided in the abstract. The results highlight that generative learning can match the statistical accuracy of traditional solvers while being significantly faster.
This work bridges generative AI and computational physics, offering a new paradigm for surrogate modeling. For AI researchers, it shows how diffusion models can be adapted for physics-based tasks, potentially inspiring similar approaches in other scientific domains (e.g., climate modeling, materials science). For practitioners, G-LED provides a practical tool for accelerating simulations where full DNS is infeasible. The main limitation is that the method requires training data from high-fidelity simulations, which may be expensive to generate. However, the ability to then run fast, accurate forecasts makes it valuable for many-query scenarios like uncertainty quantification or design optimization.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba