Deep Learning-Based Surrogate Models Outperform Simulators and Could Hasten Scientific Discoveries

June 18, 2020

June 18, 2020 — Surrogate models supported by neural networks can perform as well, and in some ways better, than computationally expensive simulators and could lead to new insights in complicated physics problems such as inertial confinement fusion (ICF), Lawrence Livermore National Laboratory (LLNL) scientists reported.

Lawrence Livermore researchers are integrating technologies such as the Sierra supercomputer (left) and the National Ignition Facility (NIF) (right) to understand complex problems like fusion in energy and the aging effects in nuclear weapons. Data from NIF experiments (inset, right) and simulation (inset, left) are being combined with deep learning methods to improve areas important to national security and our future energy sector. Illustration by Tanya Quijalvo/LLNL.

In a paper published by the Proceedings of the National Academy of Sciences (PNAS), LLNL researchers describe the development of a deep learning-driven Manifold & Cyclically Consistent (MaCC) surrogate model incorporating a multi-modal neural network capable of quickly and accurately emulating complex scientific processes, including the high-energy density physics involved in ICF.

The research team applied the model to ICF implosions performed at the National Ignition Facility (NIF), in which a computationally expensive numerical simulator is used to predict the energy yield of a target imploded by shock waves produced by the facility’s high-energy laser. Comparing the results of the neural network-backed surrogate to the existing simulator, the researchers found the surrogate could adequately replicate the simulator, and significantly outperformed the current state-of-the-art in surrogate models across a wide range of metrics.

“One major question we were dealing with was ‘how do we start using machine learning when you have a lot of different kinds of data?’ ” said LLNL computer scientist and lead author Rushil Anirudh. “What we proposed was making the problem simpler by finding a common space where all these modalities, such as high pressure or temperature, live and do the analysis within that space. We’re saying that deep learning can capture the important relationships between all these different data sources and give us a compact representation for all of them.

“The nice thing about doing all this is not only that it makes the analysis easier, because now you have a common space for all these modalities, but we also showed that doing it this way actually gives you better models, better analysis and objectively better results than with baseline approaches,” Anirudh added.

Simulations that would normally take a numerical simulator a half-hour to run could be done equally as well within a fraction of a second using neural networks, Anirudh explained. Perhaps even more valuable than saving compute time, explained computer scientist and co-author Timo Bremer, is the demonstrated ability of the deep learning surrogate model to analyze a large volume of complex, high-dimensional data in the ICF test case, which has implications for stockpile modernization efforts. The results indicate the approach could lead to new scientific discoveries and a completely novel class of techniques for performing and analyzing simulations, Bremer said.

This is particularly important at NIF, Bremer explained, where scientists do not yet fully understand why discrepancies exist between simulations and experiments. In the future, deep learning models could elicit capabilities that didn’t exist before and provide a way for scientists to analyze the massive amounts of X-ray images, sensor data and other information collected from diagnostics of each NIF shot, including data that has not been incorporated because there is too much of it to be analyzed by humans alone, Bremer said.

“This tool is providing us with a fundamentally different way of connecting simulations to experiments,” Bremer said. “By building these deep learning models, it allows us to directly predict the full complexity of the simulation data. Using this common latent space to correlate all these different modalities and different diagnostics, and using that space to connect experiments to simulations, is going to be extremely valuable, not just for this particular piece of science, but everything that tries to combine computational sciences with experimental sciences. This is something that could potentially lead to new insights in a way that’s just unfeasible right now.”

Comparing the results of predictions made by the surrogate model to the simulator typically used for ICF experiments, the researchers found the MaCC surrogate was nearly indistinguishable from the simulator in errors and expected quantities of energy yield and more accurate than other types of surrogate models. Researchers said the key to the MaCC model’s success was the coupling of forward and inverse models and training them on data together. The surrogate model used data inputs to make predictions, and those predictions were run through an inverse model to estimate, from the outputs, what the inputs might have been. During training, the surrogate’s neural networks learned to be compatible with the inverse models, meaning that errors did not accumulate as much as they would have before, Anirudh said.

“We were exploring this notion of self-consistency,” Anirudh explained. “We found that including the inverse problem into the surrogate modeling process is actually essential. It makes the problem more data-efficient and slightly more robust. When you put these two pieces together, the inverse model and the common space for all the modalities, you get this grand surrogate model that has all these other desirable properties — it is more efficient and better with less amount of data, and it’s also resilient to sampling artifacts.”

The team said the benefit of machine learning-based surrogates is that they can speed up extremely complex calculations and compare varied data sources efficiently without requiring a scientist to scan tremendous amounts of data. As simulators become increasingly complex, producing even more data, such surrogate models will become a fundamental complementary tool for scientific discovery, researchers said.

“The tools we built will be useful even as the simulation becomes more complex,” said computer scientist and co-author Jayaraman Thiagarajan. “Tomorrow we will get new computing power, bigger supercomputers and more accurate calculations, and these techniques will still hold true. We are surprisingly finding that you can produce very powerful emulators for the underlying complex simulations, and that’s where this becomes very important.

‘As long as you can approximate the underlying science using a mathematical model, the speed at which we can explore the space becomes really, really fast,” Thiagarajan continued. “That will hopefully help us in the future to make scientific discoveries even quicker and more effectively. We believe that even though we used it for this particular application, this approach is broadly applicable to the general umbrella of science.”

Researchers said the MaCC surrogate model could be adapted for any future change in modality, new types of sensors or imaging techniques. Because of its flexibility and accuracy, the model and its deep learning approach, referred to at LLNL as “cognitive simulation” or simply CogSim, is being applied to a number of other projects within the Laboratory and is transitioning over to programmatic work, including efforts in uncertainty quantification, weapons physics design, magnetic confinement fusion and other laser projects.

MaCC is a key product of the Lab’s broader Cognitive Simulation Director’s Initiative, led by principal investigator and LLNL physicist Brian Spears and funded through the Laboratory Directed Research and Development (LDRD) program. The initiative aims to advance a wide range of AI technologies and computational platforms specifically designed to improve scientific predictions by more effectively coupling precision simulation with experimental data. By focusing on both the needs in critical mission spaces and the opportunities presented by AI and compute advances, the initiative has helped further LLNL’s lead in using AI for science.

“MaCC’s ability to combine multiple, scientifically relevant data streams opens the door for a wide range of new analyses,” Spears said. “It will allow us to extract information from our most valuable and mission-critical experimental and simulation data sets that has been inaccessible until now. Fully exploiting this information in concert with a new suite of related CogSim tools will lead quickly and directly to improved predictive models.”

The research team has made their data publicly available on the web.


Source: Lawrence Livermore National Laboratory

Subscribe to HPCwire's Weekly Update!

Be the most informed person in the room! Stay ahead of the tech trends with industy updates delivered to you every week!

ISC21 Cluster Competition Bracketology

June 18, 2021

For the first time ever, cluster competition experts have gathered together for an actual seeding reveal for the ISC21 Student Cluster Competition. What’s this, you ask? It’s where bona fide student cluster competi Read more…

OSC Enables On-Demand HPC for Automotive Engineering Firm

June 18, 2021

In motorsports, vehicle designers are constantly looking for the tiniest sliver of time to shave off through some clever piece of engineering – but as the low-hanging fruit gets snatched up, those advances are getting Read more…

PNNL Researchers Unveil Tool to Accelerate CGRA Development

June 18, 2021

Moore’s law is in decline due to the physical limits of transistor chips, putting an expiration date on a hitherto-perennial exponential trend in computing power – and leaving hardware developers scrambling to contin Read more…

TU Wien Announces VSC-5, Austria’s Most Powerful Supercomputer

June 17, 2021

Austria is getting a new top supercomputer: VSC-5, the latest iteration of the Vienna Scientific Cluster. The news was announced by VSC-5’s soon-to-be home, TU Wien (also known as the Vienna University of Technology). Read more…

Supercomputing Helps Advance Hydrogen Energy Research

June 16, 2021

Hydrogen energy has long remained an elusive target of the renewable energy industry, promising clean, carbon-free energy that would allow for rapid refueling, unlike current battery-based electric vehicles. Hydrogen-bas Read more…

AWS Solution Channel

Accelerating research and development for new medical treatments

Today, more than 290,000 researchers in France are working to provide better support and care for patients through modern medical treatment. To fulfill their mission, these researchers must be equipped with powerful tools. Read more…

FF4EuroHPC Initiative Highlights Results of First Open Call

June 16, 2021

EuroHPC is kicking into high gear, with seven of its first eight systems detailed – and one of them already operational. While the systems are, perhaps, the flashiest endeavor of the European Commission’s HPC effort, Read more…

TU Wien Announces VSC-5, Austria’s Most Powerful Supercomputer

June 17, 2021

Austria is getting a new top supercomputer: VSC-5, the latest iteration of the Vienna Scientific Cluster. The news was announced by VSC-5’s soon-to-be home, T Read more…

Catching up with ISC 2021 Digital Program Chair Martin Schulz

June 16, 2021

Leibniz Research Centre (LRZ)’s content creator Susanne Vieser interviews ISC 2021 Digital Program Chair, Prof. Martin Schulz to gain an understanding of his ISC affiliation, which is outside his usual scope of work at the research center and the Technical University of Munich. Read more…

Intel Debuts ‘Infrastructure Processing Unit’ as Part of Broader XPU Strategy

June 15, 2021

To boost the performance of busy CPUs hosted by cloud service providers, Intel Corp. has launched a new line of Infrastructure Processing Units (IPUs) that take Read more…

ISC Keynote: Glimpse into Microsoft’s View of the Quantum Computing Landscape

June 15, 2021

Looking for a dose of reality and realistic optimism about quantum computing? Matthias Troyer, Microsoft distinguished scientist, plans to do just that in his ISC2021 keynote in two weeks – Quantum Computing: From Academic Research to Real-world Applications. He notes wryly that classical... Read more…

A Carbon Crisis Looms Over Supercomputing. How Do We Stop It?

June 11, 2021

Supercomputing is extraordinarily power-hungry, with many of the top systems measuring their peak demand in the megawatts due to powerful processors and their c Read more…

Honeywell Quantum and Cambridge Quantum Plan to Merge; More to Follow?

June 10, 2021

Earlier this week, Honeywell announced plans to merge its quantum computing business, Honeywell Quantum Solutions (HQS), which focuses on trapped ion hardware, Read more…

ISC21 Keynoter Xiaoxiang Zhu to Deliver a Bird’s-Eye View of a Changing World

June 10, 2021

ISC High Performance 2021 – once again virtual due to the ongoing pandemic – is swiftly approaching. In contrast to last year’s conference, which canceled Read more…

Xilinx Expands Versal Chip Family With 7 New Versal AI Edge Chips

June 10, 2021

FPGA chip vendor Xilinx has been busy over the last several years cranking out its Versal AI Core, Versal Premium and Versal Prime chip families to fill customer compute needs in the cloud, datacenters, networks and more. Now Xilinx is expanding its reach to the booming edge... Read more…

AMD Chipmaker TSMC to Use AMD Chips for Chipmaking

May 8, 2021

TSMC has tapped AMD to support its major manufacturing and R&D workloads. AMD will provide its Epyc Rome 7702P CPUs – with 64 cores operating at a base cl Read more…

Intel Launches 10nm ‘Ice Lake’ Datacenter CPU with Up to 40 Cores

April 6, 2021

The wait is over. Today Intel officially launched its 10nm datacenter CPU, the third-generation Intel Xeon Scalable processor, codenamed Ice Lake. With up to 40 Read more…

Berkeley Lab Debuts Perlmutter, World’s Fastest AI Supercomputer

May 27, 2021

A ribbon-cutting ceremony held virtually at Berkeley Lab's National Energy Research Scientific Computing Center (NERSC) today marked the official launch of Perlmutter – aka NERSC-9 – the GPU-accelerated supercomputer built by HPE in partnership with Nvidia and AMD. Read more…

Google Launches TPU v4 AI Chips

May 20, 2021

Google CEO Sundar Pichai spoke for only one minute and 42 seconds about the company’s latest TPU v4 Tensor Processing Units during his keynote at the Google I Read more…

CERN Is Betting Big on Exascale

April 1, 2021

The European Organization for Nuclear Research (CERN) involves 23 countries, 15,000 researchers, billions of dollars a year, and the biggest machine in the worl Read more…

Iran Gains HPC Capabilities with Launch of ‘Simorgh’ Supercomputer

May 18, 2021

Iran is said to be developing domestic supercomputing technology to advance the processing of scientific, economic, political and military data, and to strengthen the nation’s position in the age of AI and big data. On Sunday, Iran unveiled the Simorgh supercomputer, which will deliver.... Read more…

HPE Launches Storage Line Loaded with IBM’s Spectrum Scale File System

April 6, 2021

HPE today launched a new family of storage solutions bundled with IBM’s Spectrum Scale Erasure Code Edition parallel file system (description below) and featu Read more…

Quantum Computer Start-up IonQ Plans IPO via SPAC

March 8, 2021

IonQ, a Maryland-based quantum computing start-up working with ion trap technology, plans to go public via a Special Purpose Acquisition Company (SPAC) merger a Read more…

Leading Solution Providers

Contributors

10nm, 7nm, 5nm…. Should the Chip Nanometer Metric Be Replaced?

June 1, 2020

The biggest cool factor in server chips is the nanometer. AMD beating Intel to a CPU built on a 7nm process node* – with 5nm and 3nm on the way – has been i Read more…

AMD Launches Epyc ‘Milan’ with 19 SKUs for HPC, Enterprise and Hyperscale

March 15, 2021

At a virtual launch event held today (Monday), AMD revealed its third-generation Epyc “Milan” CPU lineup: a set of 19 SKUs -- including the flagship 64-core, 280-watt 7763 part --  aimed at HPC, enterprise and cloud workloads. Notably, the third-gen Epyc Milan chips achieve 19 percent... Read more…

Julia Update: Adoption Keeps Climbing; Is It a Python Challenger?

January 13, 2021

The rapid adoption of Julia, the open source, high level programing language with roots at MIT, shows no sign of slowing according to data from Julialang.org. I Read more…

Can Deep Learning Replace Numerical Weather Prediction?

March 3, 2021

Numerical weather prediction (NWP) is a mainstay of supercomputing. Some of the first applications of the first supercomputers dealt with climate modeling, and Read more…

GTC21: Nvidia Launches cuQuantum; Dips a Toe in Quantum Computing

April 13, 2021

Yesterday Nvidia officially dipped a toe into quantum computing with the launch of cuQuantum SDK, a development platform for simulating quantum circuits on GPU-accelerated systems. As Nvidia CEO Jensen Huang emphasized in his keynote, Nvidia doesn’t plan to build... Read more…

Microsoft to Provide World’s Most Powerful Weather & Climate Supercomputer for UK’s Met Office

April 22, 2021

More than 14 months ago, the UK government announced plans to invest £1.2 billion ($1.56 billion) into weather and climate supercomputing, including procuremen Read more…

African Supercomputing Center Inaugurates ‘Toubkal,’ Most Powerful Supercomputer on the Continent

February 25, 2021

Historically, Africa hasn’t exactly been synonymous with supercomputing. There are only a handful of supercomputers on the continent, with few ranking on the Read more…

The History of Supercomputing vs. COVID-19

March 9, 2021

The COVID-19 pandemic poses a greater challenge to the high-performance computing community than any before. HPCwire's coverage of the supercomputing response t Read more…

  • arrow
  • Click Here for More Headlines
  • arrow
HPCwire