Up to two billion times acceleration of scientific simulations with deep neural architecture search

Computer simulations are invaluable tools for scientific discovery. However, accurate simulations are often slow to execute, which limits their applicability to extensive parameter exploration, large-scale data analysis, and uncertainty quantification. A promising route to accelerate simulations by building fast emulators with machine learning requires large training datasets, which can be prohibitively expensive to obtain with slow simulations. Here we present a method based on neural architecture search to build accurate emulators even with a limited number of training data. The method successfully accelerates simulations by up to 2 billion times in 10 scientific cases including astrophysics, climate science, biogeochemistry, high energy density physics, fusion energy, and seismology, using the same super-architecture, algorithm, and hyperparameters. Our approach also inherently provides emulator uncertainty estimation, adding further confidence in their use. We anticipate this work will accelerate research involving expensive simulations, allow more extensive parameters exploration, and enable new, previously unfeasible computational discovery.

[1]  T. Pütterich,et al.  Pedestal and Er profile evolution during an edge localized mode cycle at ASDEX Upgrade , 2017 .

[2]  Andreas Oschlies,et al.  MOPS-1.0: modelling the regulation of the global oceanic nitrogen budget by marine biogeochemical processes , 2015 .

[3]  Zoubin Ghahramani,et al.  Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning , 2015, ICML.

[4]  T. Pütterich,et al.  Ion heat transport dynamics during edge localized mode cycles at ASDEX Upgrade , 2018 .

[5]  Nitish Srivastava,et al.  Dropout: a simple way to prevent neural networks from overfitting , 2014, J. Mach. Learn. Res..

[6]  O. Landen,et al.  Theoretical model of x-ray scattering as a dense matter probe. , 2003, Physical review. E, Statistical, nonlinear, and soft matter physics.

[7]  Leena Aho-Mantila,et al.  Particle transport analysis of the density build-up after the L–H transition in ASDEX Upgrade , 2013 .

[8]  K. Müller,et al.  Fast and accurate modeling of molecular atomization energies with machine learning. , 2011, Physical review letters.

[9]  Andrea Vedaldi,et al.  Deep Image Prior , 2017, International Journal of Computer Vision.

[10]  Gilbert W. Collins,et al.  Hot-spot mix in ignition-scale inertial confinement fusion targets. , 2013, Physical review letters.

[11]  M.Vaccari,et al.  The galaxy–halo connection in the VIDEO survey at 0.5 < z < 1.7 , 2015, 1511.05476.

[12]  Francis F. Chen Plasma Physics and Controlled Fusion , 1984 .

[13]  Jonathan R Goodman,et al.  Ensemble samplers with affine invariance , 2010 .

[14]  Tom Schaul,et al.  Natural Evolution Strategies , 2008, 2008 IEEE Congress on Evolutionary Computation (IEEE World Congress on Computational Intelligence).

[15]  Ronald J. Williams,et al.  Simple Statistical Gradient-Following Algorithms for Connectionist Reinforcement Learning , 2004, Machine Learning.

[16]  Justin S. Wark,et al.  Simultaneous diagnosis of radial profiles and mix in NIF ignition-scale implosions via X-ray spectroscopy , 2017 .

[17]  Roger W. Falcone,et al.  Ultrafast X-ray Thomson Scattering of Shock-Compressed Matter , 2008, Science.

[18]  M Nocente,et al.  Beam-Ion Acceleration during Edge Localized Modes in the ASDEX Upgrade Tokamak. , 2018, Physical review letters.

[19]  Gaël Varoquaux,et al.  Scikit-learn: Machine Learning in Python , 2011, J. Mach. Learn. Res..

[20]  Anand Srivastav,et al.  Calibrating a global three-dimensional biogeochemical ocean model (MOPS-1.0) , 2016 .

[21]  D. Lamb,et al.  Laboratory evidence of dynamo amplification of magnetic fields in a turbulent plasma , 2017, Nature Communications.

[22]  J. Topp-Mugglestone,et al.  Inverse problem instabilities in large-scale modeling of matter in extreme conditions , 2019, Physics of Plasmas.

[23]  Li Li,et al.  Bypassing the Kohn-Sham equations with machine learning , 2016, Nature Communications.

[24]  Ven-Shion Shu,et al.  Robust estimation of a location parameter in the presence of outliers , 1978 .

[25]  Duncan Watson-Parris,et al.  The global aerosol–climate model ECHAM6.3–HAM2.3 – Part 1: Aerosol evaluation , 2019, Geoscientific Model Development.

[26]  Frank Hutter,et al.  CMA-ES for Hyperparameter Optimization of Deep Neural Networks , 2016, ArXiv.

[27]  William W. Sager,et al.  Seismic tomography of Shatsky Rise by adaptive importance sampling , 2012 .

[28]  Frederick R. Forst,et al.  On robust estimation of the location parameter , 1980 .

[29]  W. Treutterer,et al.  Novel free-boundary equilibrium and transport solver with theory-based models and its validation against ASDEX Upgrade current ramp scenarios , 2013 .

[30]  L. Perelman,et al.  A finite-volume, incompressible Navier Stokes model for studies of the ocean on parallel computers , 1997 .

[31]  J. Nørskov,et al.  Computational high-throughput screening of electrocatalytic materials for hydrogen evolution , 2006, Nature materials.

[32]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[33]  H. J. Lee,et al.  X-ray Thomson-scattering measurements of density and temperature in shock-compressed beryllium. , 2008, Physical review letters.

[34]  Hal Finkel,et al.  COSMIC EMULATION: FAST PREDICTIONS FOR THE GALAXY POWER SPECTRUM , 2013, 1311.6444.

[35]  Kelli Humbird,et al.  Zonal flow generation in inertial confinement fusion implosions , 2017 .

[36]  Samar Khatiwala,et al.  A computational framework for simulation of biogeochemical tracers in the ocean , 2007 .

[37]  Nikolaus Hansen,et al.  The CMA Evolution Strategy: A Tutorial , 2016, ArXiv.