Enforcing statistical constraints in generative adversarial networks for modeling chaotic dynamical systems

被引:83
作者
Wu, Jin-Long [1 ,4 ]
Kashinath, Karthik [2 ]
Albert, Adrian [2 ]
Chirila, Dragos [3 ]
Prabhat [2 ]
Xiao, Heng [1 ]
机构
[1] Virginia Tech, Kevin T Crofton Dept Aerosp & Ocean Engn, Blacksburg, VA 24060 USA
[2] Lawrence Berkeley Natl Lab, Berkeley, CA 94720 USA
[3] Helmholtz Ctr Polar & Marine Res AWI, Alfred Wegener Inst, Bremerhaven, Germany
[4] CALTECH, Pasadena, CA 91125 USA
关键词
Machine learning; Generative adversarial networks; Statistical constraint; Partial differential equations; Rayleigh-Benard convection; NEURAL-NETWORKS;
D O I
10.1016/j.jcp.2019.109209
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Simulating complex physical systems often involves solving partial differential equations (PDEs) with some closures due to the presence of multi-scale physics that cannot be fully resolved. Although the advancement of high performance computing has made resolving small-scale physics possible, such simulations are still very expensive. Therefore, reliable and accurate closure models for the unresolved physics remains an important requirement for many computational physics problems, e.g., turbulence simulation. Recently, several researchers have adopted generative adversarial networks (GANs), a novel paradigm of training machine learning models, to generate solutions of PDEs-governed complex systems without having to numerically solve these PDEs. However, GANs are known to be difficult in training and likely to converge to local minima, where the generated samples do not capture the true statistics of the training data. In this work, we present a statistical constrained generative adversarial network by enforcing constraints of covariance from the training data, which results in an improved machine-learning-based emulator to capture the statistics of the training data generated by solving fully resolved PDEs. We show that such a statistical regularization leads to better performance compared to standard GANs, measured by (1) the constrained model's ability to more faithfully emulate certain physical properties of the system and (2) the significantly reduced (by up to 80%) training time to reach the solution. We exemplify this approach on the Rayleigh-Benard convection, a turbulent flow system that is an idealized model of the Earth's atmosphere. With the growth of high-fidelity simulation databases of physical systems, this work suggests great potential for being an alternative to the explicit modeling of closures or parameterizations for unresolved physics, which are known to be a major source of uncertainty in simulating multi-scale physical systems, e.g., turbulence or Earth's climate. (C) 2019 Elsevier Inc. All rights reserved.
引用
收藏
页数:20
相关论文
共 54 条
  • [1] Abadi M, 2016, ACM SIGPLAN NOTICES, V51, P1, DOI [10.1145/2951913.2976746, 10.1145/3022670.2976746]
  • [2] [Anonymous], ARXIV170602390
  • [3] [Anonymous], B AM PHYS SOC US
  • [4] [Anonymous], ARXIV180300657 CORR
  • [5] [Anonymous], ARXIV170507215 CORR
  • [6] [Anonymous], ARXIV151106434 CORR
  • [7] [Anonymous], ARXIV191106671
  • [8] [Anonymous], 2005, Visualization and processing of tensor fields
  • [9] [Anonymous], 2017, AGU FALL M
  • [10] [Anonymous], ARXIV180804258