CERN Project Sees Orders-of-Magnitude Speedup with AI Approach

By Rob Farber

August 14, 2018

An award-winning effort at CERN has demonstrated potential to significantly change how the physics based modeling and simulation communities view machine learning. The CERN team demonstrated that AI-based models have the potential to act as orders-of-magnitude-faster replacements for computationally expensive tasks in simulation, while maintaining a remarkable level of accuracy.

Dr. Federico Carminati (Project Coordinator, CERN) points out, “This work demonstrates the potential of ‘black box’ machine-learning models in physics-based simulations.”

A poster describing this work was awarded the prize for best poster in the category ‘programming models and systems software’ at ISC’18. This recognizes the importance of the work, which was carried out by Dr. Federico Carminati, Gul Rukh Khattak, and Dr. Sofia Vallecorsa at CERN, as well as Jean-Roch Vlimant at Caltech. The work is part of a CERN openlab project in collaboration with Intel Corporation, who partially funded the endeavor through the Intel Parallel Computing Center (IPCC) program.

Widespread potential impact for simulation

The world-wide impact for High-Energy Physics (HEP) scientists could be substantial, as outlined by the CERN poster, which points out that ”Currently, most of the LHC’s worldwide distributed CPU budget — in the range of half a million CPU-years equivalent — is dedicated to simulation.” Speeding up the most time-consuming simulation tasks (e.g., high-granularity calorimeters, which are components in a detector that measure the energy of particles[i]) will help scientists better utilize these allocations. The following are comparative results obtained by the CERN team in the time to create an electron shower, once the AI model has been fully trained:

Figure 1: Comparative runtime to create an electron shower of the machine-learning method (e.g. 3d GAN) vs. the full Monte-Carlo simulation (Image courtesy CERN)

Dr. Sofia Vallecorsa points out that the CPU based runtime is important as nearly all of the Geant user base runs on CPUs. Vallecorsa is a CERN physicist who was also highlighted in the CERN article Coding has no gender.

As scientists consider future CERN experiments, Vallecorsa observes, “Given future plans to upgrade CERN’s Large Hadron Collider, dramatically increasing particle collision rates, frameworks like this have the potential to play an important role in ensuring data rates remain manageable.”

This kind of approach could help to realize similar orders-of-magnitude-faster speedups for computationally expensive simulation tasks used in a range of fields.

Vallecorsa explains that the data distributions coming from the trained machine-learning model are remarkably close to the real and simulated data.

A big change in thinking

The team demonstrated that “energy showers” detected by calorimeters can be interpreted as a 3D image[ii]. The process is illustrated in the following figure. The team adopted this approach from the machine-learning community as deep-learning convolutional neural networks are heavily utilized when working with images.

Figure 2: Schematic from the poster showing how a single particle creates an electron shower that can be viewed as an image (Courtesy CERN)

Use of GANS

The CERN team decided to train Generative Adversarial Networks (GANs) on the calorimeter images. GANs are particularly suited to act as a replacement for the expensive Monte Carlo methods used in HEP simulations as they generate realistic samples for complicated probability distributions, allow multi-modal output, can do interpolation, and are robust against missing data.

The basic idea is easy to understand: train a Generator (G) to create the calorimeter image with sufficient accuracy to trick a discriminator (D) which tries to identify artificial samples from the generator compared to real samples from the Monte Carlo simulation. G reproduces the data distribution starting from random noise. D estimates the probability that a sample came from the training data rather than G. The training procedure for G is to maximize the probability of D making a mistake. A high-level illustration of the GAN is provided below.

Figure 3: High-level view of training a GAN (image from https://medium.com/@devnag/generative-adversearial-networks-in-50-lines-of-code-pytorch-e81b79659e3f)

Even though the description is simple, 3D GANs are unfortunately not “out-of-the-box” networks, which meant the training of the model was non-trivial.

Results

After detailed validation of the trained GAN, there was “remarkable” agreement between the images from the generator and the Monte-Carlo images. This type of approach could potentially be beneficial in other fields where Monte Carlo simulation is used.

More specifically, the CERN team compared high level quantities (e.g., energy shower shapes) and detailed calorimeter response (e.g., single cell response) between the trained generator and the standard Monte Carlo. The CERN team describes the agreement, which is within a few percent, as “remarkable” in their poster.

Visually this agreement can be seen by how closely the blue (real data) and red lines (GAN generated data) overlap in the following results reported in the poster.

Figure 4: Transverse shower shape for 100-500 GeV pions. Red is the GAN data while blue represents the real data. (Image courtesy CERN)

 

Figure 5: Longitudinal shower shape for 400 GeV electron (Image courtesy CERN)

 

Figure 6: Longitudinal shower shape for 100 GeV electron (Image courtesy CERN)

Vallecorsa summarizes these results by stating, “The agreement between the images generated by our model and the Monte Carlo images has been beyond our expectations. This demonstrates that this is a promising avenue for further investigation.”

CERN openlab

The CERN team plans to test performance using FPGAs and other integrated accelerator technologies. FPGAs are known to deliver lower latency and higher inferencing performance than both CPUs and GPUs[iii]. The CERN group also intends to test several deep learning techniques in the hope of achieving a yet greater speedup with respect to Monte Carlo techniques, and ensuring this approach covers a range of detector types, which CERN believes is key to future projects.

This research is being carried out through a CERN openlab project. CERN openlab is a public-private partnership through which CERN collaborates with leading ICT companies to drive innovation in cutting-edge ICT solutions for its research community. Intel has been a partner in CERN openlab since it was first established in 2001. Dr. Alberto Di Meglio (Head of CERN openlab) observes, “At CERN, we’re always interested in exploring upcoming technologies that can help researchers to make new ground-breaking discoveries about our universe. We support this through joint R&D projects with our collaborators from industry, and by making cutting-edge technologies available for evaluation by researchers at CERN.”

Summary

The HPC modeling and simulation community now has a promising path forward to exploit the benefits of machine learning. The key, as demonstrated by CERN, is that the machine-learning-generated distribution needs to be indistinguishable from other high-fidelity methods in physics-based simulations.

The motivation is straightforward: (1) orders of magnitude faster performance, (2) efficient CPU implementations, and (3) this approach could enable the use of other new technologies such as FPGAs that may significantly improve performance.

Additional References

Rob Farber is a global technology consultant and author with an extensive background in HPC and in machine learning technology that he applies at national labs and commercial organizations on a variety of problems including challenges in high energy physics. Rob can be reached at [email protected]

[i] http://cds.cern.ch/record/2254048#

[ii] ibid

[iii] https://medium.com/syncedreview/deep-learning-in-real-time-inference-acceleration-and-continuous-training-17dac9438b0b

Subscribe to HPCwire's Weekly Update!

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

MLPerf Training 4.0 – Nvidia Still King; Power and LLM Fine Tuning Added

June 12, 2024

There are really two stories packaged in the most recent MLPerf  Training 4.0 results, released today. The first, of course, is the results. Nvidia (currently king of accelerated computing) wins again, sweeping all nine Read more…

Highlights from GlobusWorld 2024: The Conference for Reimagining Research IT

June 11, 2024

The Globus user conference, now in its 22nd year, brought together over 180 researchers, system administrators, developers, and IT leaders from 55 top research computing centers, national labs, federal agencies, and univ Read more…

Nvidia Shipped 3.76 Million Data-center GPUs in 2023, According to Study

June 10, 2024

Nvidia had an explosive 2023 in data-center GPU shipments, which totaled roughly 3.76 million units, according to a study conducted by semiconductor analyst firm TechInsights. Nvidia's GPU shipments in 2023 grew by more Read more…

Weekly Wire Roundup: June 2-June 7, 2024

June 8, 2024

Computex (and Jensen Huang) gave us an extra day of news this week, compensating for last week's shorter, holiday-driven news cycle. On Sunday ahead of the official start of Computex, Nvidia's CEO Jensen Huang deliver Read more…

ASC24 Expert Perspective: Dongarra, Hoefler, Yong Lin

June 7, 2024

One of the great things about being at an ASC (Asia Supercomputer Community) cluster competition is getting the chance to interview various industry experts and learning more about the various challenges the students are Read more…

HPC and Climate: Coastal Hurricanes Around the World Are Intensifying Faster

June 6, 2024

Hurricanes are among the world's most destructive natural hazards. Their environment shapes their ability to deliver damage; conditions like warm ocean waters, guiding winds, and atmospheric moisture can all dictate stor Read more…

MLPerf Training 4.0 – Nvidia Still King; Power and LLM Fine Tuning Added

June 12, 2024

There are really two stories packaged in the most recent MLPerf  Training 4.0 results, released today. The first, of course, is the results. Nvidia (currently Read more…

Highlights from GlobusWorld 2024: The Conference for Reimagining Research IT

June 11, 2024

The Globus user conference, now in its 22nd year, brought together over 180 researchers, system administrators, developers, and IT leaders from 55 top research Read more…

Nvidia Shipped 3.76 Million Data-center GPUs in 2023, According to Study

June 10, 2024

Nvidia had an explosive 2023 in data-center GPU shipments, which totaled roughly 3.76 million units, according to a study conducted by semiconductor analyst fir Read more…

ASC24 Expert Perspective: Dongarra, Hoefler, Yong Lin

June 7, 2024

One of the great things about being at an ASC (Asia Supercomputer Community) cluster competition is getting the chance to interview various industry experts and Read more…

HPC and Climate: Coastal Hurricanes Around the World Are Intensifying Faster

June 6, 2024

Hurricanes are among the world's most destructive natural hazards. Their environment shapes their ability to deliver damage; conditions like warm ocean waters, Read more…

ASC24: The Battle, The Apps, and The Competitors

June 5, 2024

The ASC24 (Asia Supercomputer Community) Student Cluster Competition was one for the ages. More than 350 university teams worked for months in the preliminary competition to earn one of the 25 final competition slots. The winning teams... Read more…

Computex 2024: Nvidia, AMD Push GPUs; Intel Revs Up x86 Power Efficiency

June 5, 2024

"The days of millions of GPU data centers are coming," said Nvidia CEO Jensen Huang during a keynote at Computex. Huang's predictions are becoming bolder and bo Read more…

Using AI and Robots to Advance Science

June 4, 2024

Even though we invented it, humans can be pretty bad at science. We need to eat and sleep, we sometimes let our emotions regulate our behavior, and our bodies a Read more…

Atos Outlines Plans to Get Acquired, and a Path Forward

May 21, 2024

Atos – via its subsidiary Eviden – is the second major supercomputer maker outside of HPE, while others have largely dropped out. The lack of integrators and Atos' financial turmoil have the HPC market worried. If Atos goes under, HPE will be the only major option for building large-scale systems. Read more…

Comparing NVIDIA A100 and NVIDIA L40S: Which GPU is Ideal for AI and Graphics-Intensive Workloads?

October 30, 2023

With long lead times for the NVIDIA H100 and A100 GPUs, many organizations are looking at the new NVIDIA L40S GPU, which it’s a new GPU optimized for AI and g Read more…

Nvidia H100: Are 550,000 GPUs Enough for This Year?

August 17, 2023

The GPU Squeeze continues to place a premium on Nvidia H100 GPUs. In a recent Financial Times article, Nvidia reports that it expects to ship 550,000 of its lat Read more…

Choosing the Right GPU for LLM Inference and Training

December 11, 2023

Accelerating the training and inference processes of deep learning models is crucial for unleashing their true potential and NVIDIA GPUs have emerged as a game- Read more…

Everyone Except Nvidia Forms Ultra Accelerator Link (UALink) Consortium

May 30, 2024

Consider the GPU. An island of SIMD greatness that makes light work of matrix math. Originally designed to rapidly paint dots on a computer monitor, it was then Read more…

Nvidia’s New Blackwell GPU Can Train AI Models with Trillions of Parameters

March 18, 2024

Nvidia's latest and fastest GPU, codenamed Blackwell, is here and will underpin the company's AI plans this year. The chip offers performance improvements from Read more…

Synopsys Eats Ansys: Does HPC Get Indigestion?

February 8, 2024

Recently, it was announced that Synopsys is buying HPC tool developer Ansys. Started in Pittsburgh, Pa., in 1970 as Swanson Analysis Systems, Inc. (SASI) by John Swanson (and eventually renamed), Ansys serves the CAE (Computer Aided Engineering)/multiphysics engineering simulation market. Read more…

Some Reasons Why Aurora Didn’t Take First Place in the Top500 List

May 15, 2024

The makers of the Aurora supercomputer, which is housed at the Argonne National Laboratory, gave some reasons why the system didn't make the top spot on the Top Read more…

Leading Solution Providers

Contributors

AMD MI3000A

How AMD May Get Across the CUDA Moat

October 5, 2023

When discussing GenAI, the term "GPU" almost always enters the conversation and the topic often moves toward performance and access. Interestingly, the word "GPU" is assumed to mean "Nvidia" products. (As an aside, the popular Nvidia hardware used in GenAI are not technically... Read more…

The NASA Black Hole Plunge

May 7, 2024

We have all thought about it. No one has done it, but now, thanks to HPC, we see what it looks like. Hold on to your feet because NASA has released videos of wh Read more…

GenAI Having Major Impact on Data Culture, Survey Says

February 21, 2024

While 2023 was the year of GenAI, the adoption rates for GenAI did not match expectations. Most organizations are continuing to invest in GenAI but are yet to Read more…

Google Announces Sixth-generation AI Chip, a TPU Called Trillium

May 17, 2024

On Tuesday May 14th, Google announced its sixth-generation TPU (tensor processing unit) called Trillium.  The chip, essentially a TPU v6, is the company's l Read more…

Q&A with Nvidia’s Chief of DGX Systems on the DGX-GB200 Rack-scale System

March 27, 2024

Pictures of Nvidia's new flagship mega-server, the DGX GB200, on the GTC show floor got favorable reactions on social media for the sheer amount of computing po Read more…

Intel’s Next-gen Falcon Shores Coming Out in Late 2025 

April 30, 2024

It's a long wait for customers hanging on for Intel's next-generation GPU, Falcon Shores, which will be released in late 2025.  "Then we have a rich, a very Read more…

Intel Plans Falcon Shores 2 GPU Supercomputing Chip for 2026  

August 8, 2023

Intel is planning to onboard a new version of the Falcon Shores chip in 2026, which is code-named Falcon Shores 2. The new product was announced by CEO Pat Gel Read more…

Shutterstock 1285747942

AMD’s Horsepower-packed MI300X GPU Beats Nvidia’s Upcoming H200

December 7, 2023

AMD and Nvidia are locked in an AI performance battle – much like the gaming GPU performance clash the companies have waged for decades. AMD has claimed it Read more…

  • arrow
  • Click Here for More Headlines
  • arrow
HPCwire