MLPerf Debuts HPC Training Benchmark with Small but Impressive List of Participants

By John Russell

November 18, 2020

The AI benchmarking organization MLPerf.org dipped a toe into HPC-centric waters today with release of results from its first HPC training run – MLPerf HPC Training v0.7. The new suite, which includes CosmoFlow and DeepCAM models, is intended to measure machine learning performance on large-scale high performance computing systems.

While the number of first-round entries was small, they represented impressive HPC systems – Fugaku (RIKEN), top performer on the last two Top500 Lists; Piz Daint (CSCS); Cori (Lawrence Berkeley National Laboratory); Frontera (Texas Advanced Computing Center); AI Bridging Cloud Infrastructure (ABCI, Fujitsu) and HAL cluster (National Center for Supercomputer Applications).

There have been a few significant rule changes from MLPerf ‘standard’ training benchmarks for the HPC exercise. Time-to-solution is still the metric of merit but MLPerf has made an effort to include other bottlenecks beyond (mostly) processor performance. The size and complexity of the HPC systems participating make it necessary to dig into the details of the results for a proper performance evaluation. That said, among the more interesting pieces of today’s MLPerf announcement are statements provided by the participants on their MLPerf work (see below). It’s interesting, for example, to see how batch size was a factor.

Steven Farrell, an engineer at NERSC and member of the MLPerf team spearheading development of the HPC benchmarking effort said, “The rules for the HPC differ a little bit from MLPerf training. For example, any data movement as part of the preprocessing from a general parallel file system to local storage or I/O accelerator type system [must] be included in the benchmark reported time. And actually we captured the time that’s spent in this staging process.”

“At the end of day, the way the MLPerf HPC results are presented doesn’t really give you a strict, ranking, like number one, number two, number three. You kind of have to parse the results with some sense of the scale of the system used [to make a judgement],” said Farrell. MLPerf provides easy access to the data for slicing and dicing.

The top performer on CosmoFlow in the closed division was ABCI at 34.2 minutes; its closest rival was Fugaku at 101.49 minutes. ABCI was also the top performer in the open division at 13.21 minutes. Closed division regulations are more restrictive while the open division permits more flexibility in how one runs the benchmark.

David Kanter, executive director of MLPerf noted, “HPC training is what I think of as a supercomputer and site specific benchmark. Whereas MLPerf training tends to be more vendor specific. There is a lot of analysis that shows that the interconnect matters tremendously on training for HPC and I sincerely hope that that aspect is reflected in the benchmark.”

Per MLPerf’s long-term preference, the HPC training suite uses real-world applications. The first two benchmarks measure the time it takes to train emerging scientific machine learning models to a standard quality target in tasks relevant to climate analytics and cosmology. Both benchmarks make use of large scientific simulations to generate training data:

  • CosmoFlow: A 3D convolutional architecture trained on N-body cosmological simulation data to predict four cosmological parameter targets.
  • DeepCAM: A convolutional encoder-decoder segmentation architecture trained on CAM5+TECA climate simulation data to identify extreme weather phenomena such as atmospheric rivers and tropical cyclones.

The models and data used by the HPC suite differ from the canonical MLPerf training benchmarks in significant ways. For instance, CosmoFlow is trained on volumetric (3D) data, rather than the 2D data commonly employed in training image classifiers.

Similarly, DeepCAM is trained on images with 768 x 1152 pixels and 16 channels, which is substantially larger than standard vision datasets like ImageNet. Both benchmarks have massive datasets – 8.8 TB in the case of DeepCAM and 5.1 TB for CosmoFlow – introducing significant I/O challenges that expose storage and interconnect performance.

More generally, MLPerf HPC v0.7 follows MLPerf Training v0.7 rules. One exception, as noted earlier, is the effort to capture the complexity of large-scale data movement experience in HPC systems; all data staging from parallel file systems into accelerated and/or on-node storage systems must be included in the measured runtime.

Noteworthy, only two participants submitted results for the DeepCAM benchmark, and although the total number of participants in this first MLPerf HPC Training exercise was small, they did “showcase the state-of-the-art capabilities of supercomputers for training large scale scientific problems, utilizing data-parallel and model-parallel training techniques on thousands to tens of thousands of processors,” said Farrell.

It was interesting that Nvidia did not participate with its Selene supercomputer. Kanter said the COVID-19 pandemic likely an issue in keeping participation down as so many HPC systems have been pressed into service for COVID-related research. MLPerf has high hopes for the new HPC metric but recognizes establishing it may take time. Plans call for adding models and perhaps following a twice-yearly cadence (perhaps around ISC and SC) though that is uncertain.

Kanter said, “Part of this is sort of based on demand. Prior to the advent of HPC training we had been approached by a couple of supercomputing centers that were interested in using MLPerf training for bids qualification and acceptance. I think at this point, over a billion dollars of bids have used MLPerf components in the bidding process. Hopefully that’ll be more going forward.

“Frankly one of the things we see as value that we can provide to the industry is sort of aligning sales, marketing, engineering, making sure that people are using the right metrics.”

CSCS

The Swiss National Supercomputing Centre (CSCS) participated in the first MLPerf HPC Training round as part of our benchmarking initiative to identify the needs of future systems to support ML workflows in science. We focused on two data-parallel submissions with CosmoFlow on Piz Daint with 128 and 256 GPUs, one GPU per node. By using Sarus, a container engine with near-native performance for Docker-compatible containers, we were able to rapidly test and tune fine-grained communication for distributed training with Horovod and NCCL for near optimal weak scaling in the range of 100-1000 nodes.

Curiously, execution time per epoch scaled 12% faster than ideal from 128 to 256 GPUs. This scaling is a result of being able to cache the data set in RAM with 256 GPUs, whereas with 128 GPUs parallel filesystem I/O becomes an overhead. This overhead could be alleviated using near-compute storage. Algorithmically, our submission demonstrates the limits of CosmoFlow’s data-parallel scalability under closed division rules. Specifically, the number of epochs to converge scales up by about 1.6X as the system scales from 128 to 256 GPUs, while scaling from 32 to 128 GPUs only increases the epoch count by about 1.3X. Additionally, the standard deviation increases by 7X, making the model harder to train. In summary, we have identified fine-grained communication together with the addition of near-compute storage as key optimizations for ML on HPC systems, and CSCS will continue working on alternative parallelization strategies to overcome the data-parallel scalability challenge found in this round.

Fujitsu

AI Bridging Cloud Infrastructure (ABCI) is the world’s first large-scale Open AI Computing Infrastructure, constructed and operated by National Institute of Advanced Industrial Science and Technology (AIST) [1]. The ABCI system is powered by 2,176 Intel Xeon Scalable Processors (Skylake-SP), 4,352 NVIDIA Tesla V100 GPUs, and dual-rail Infiniband EDR interconnects. Fujitsu in collaboration with AIST and Fujitsu Laboratories submitted CosmoFlow and DeepCAM results. For CosmoFlow, 128 nodes (512 GPUs) were used for closed division and 512 nodes (2,048 GPUs) were used for open division. The dataset was reformatted to tar.xz files to reduce data staging time, and the following performance optimizations were applied to improve training throughput: (1) improve data loader throughput using NVIDIA Data Loading Library (DALI), (2) apply mixed precision training, (3) increase validation batch size. For open division the following accuracy improvement techniques were also applied: (1) use linear learning rate decay scheduler, (2) apply data augmentation, (3) disable dropout layers. These techniques enabled increasing batch size from 512 to 2,048 and reduced run time by 2.61x.

For DeepCAM, 256 nodes (1,024GPUs) were used for closed and open divisions. The dataset was reformatted to tar files to reduce data staging time, and the distributed data shuffles were applied among intra-node multi GPUs, and hyper-parameters were tuned including warmup steps to reduce the number of epochs to convergence. For open division, the Gradient Skipping (GradSkip) technique, one of Content-Aware Computing (CAC) techniques developed by Fujitsu Laboratories, was also applied. GradSkip avoids updating weights in some layers in the training process, by finding layers which have little effect on accuracy, based on automatic analysis of the content of data during the training process.

Fujitsu + RIKEN

RIKEN and Fujitsu are jointly developing the world’s top-level supercomputer—the supercomputer Fugaku—capable of realizing high effective performance for a broad range of application software, with the goal of full operation in 2021. RIKEN and Fujitsu in collaboration with Fujitsu Laboratories submitted CosmoFlow results for closed division using 512 nodes and 8,192 nodes, and for open division using 16,384 nodes. The dataset was reformatted to tar.xz files for reducing data staging time and LLIO (Lightweight Layered IO Accelerator) was used to make use of temporary local file system in a process. Optimized oneAPI Deep Neural Network Library (oneDNN) was developed to exploit the performance of A64FX.

Since the accuracy could not reach the target using the batch size larger than 4,096, the model parallel was introduced to apply hybrid parallelism of both data and model. Model parallel in TensorFlow was extended based on Mesh TensorFlow (MTF) so that multi processes of both data and model parallelisms are enabled. Model parallel was applied in Conv3d layers by spatial partitioning in two dimensions. The hybrid parallelism enabled scaling the number of CPUs up to 8,192 for closed division and 16,384 for open division (about 1/10 of Fugaku).

LBNL

MLPerf HPC is an important opportunity for the National Energy Research and Scientific Computing (NERSC) center at Lawrence Berkeley National Laboratory as we prepare for a growing scientific AI workload in the coming years. Berkeley Lab co-led the published scientific applications that the current benchmarks are based on, DeepCAM and CosmoFlow. For this first round of results, we have submitted results measured on the Cori supercomputer at NERSC, demonstrating data-parallel training capabilities on both the KNL and GPU partitions up to 1024 nodes and 64 GPUs, respectively. Our participation in MLPerf HPC v0.7 marks an important step for us to standardize our AI benchmarking strategy in preparation for our announced next machine coming online in 2021: Perlmutter.

NCSA

One of the goals of the Innovative Systems Lab (ISL) at the National Center for Supercomputing Applications (NCSA) is to evaluate emerging hardware and software systems of interest to the AI research community. MLPerf HPC provides a great tool to conduct such evaluations. For this round of benchmarks, we have submitted the results obtained on our Hardware-Accelerated Learning (HAL) cluster based on IBM POWER9 CPUs and NVIDIA V100 GPUs.  The system consists of 16 IBM AC922 nodes backed by an all-flash DDN storage array and EDR InfiniBand interconnect and shows great distributed training capabilities across the entire cluster.  We have developed significant experience while participating in the MLPerf HPC v0.7 project, which will benefit us in our future system designs.

TACC

The Texas Advanced Computing Center (TACC) designs and operates some of the world’s most powerful computing resources. The center’s mission is to enable discoveries that advance science and society through the application of advanced computing technologies. MLPerf HPC applications like CosmoFlow provide an invaluable opportunity to understand next-generation ML and DL applications’ requirements. TACC participated in MLPerf HPC v0.7 and submitted the performance for the Cosmoflow application at 64 GPUs on the Frontera RTX partition. The lessons learned will be used to derive the architecture of future TACC systems for the benefit of the vast growing AI community.

Link to MLPerf results: https://mlperf.org/training-results-0-7

Link to MLPerf announcement: https://mlperf.org/press#mlperf-hpc-v0.7-results

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!

HPE and NVIDIA Join Forces and Plan Conquest of Enterprise AI Frontier

June 20, 2024

The HPE Discover 2024 conference is currently in full swing, and the keynote address from Hewlett-Packard Enterprise (HPE) CEO Antonio Neri on Tuesday, June 18, was an unforgettable event. Other than being the first busi Read more…

Slide Shows Samsung May be Developing a RISC-V CPU for In-memory AI Chip

June 19, 2024

Samsung may have unintentionally revealed its intent to develop a RISC-V CPU, which a presentation slide showed may be used in an AI chip. The company plans to release an AI accelerator with heavy in-memory processing, b Read more…

ASC24 Student Cluster Competition: Who Won and Why?

June 18, 2024

As is our tradition, we’re going to take a detailed look back at the recently concluded the ASC24 Student Cluster Competition (Asia Supercomputer Community) to see not only who won the various awards, but to figure out Read more…

Qubits 2024: D-Wave’s Steady March to Quantum Success

June 18, 2024

In his opening keynote at D-Wave’s annual Qubits 2024 user meeting, being held in Boston, yesterday and today, CEO Alan Baratz again made the compelling pitch that D-Wave’s brand of analog quantum computing (quantum Read more…

Apple Using Google Cloud Infrastructure to Train and Serve AI

June 18, 2024

Apple has built a new AI infrastructure to deliver AI features introduced in its devices and is utilizing resources available in Google's cloud infrastructure.  Apple's new AI backend includes: A homegrown foun Read more…

Argonne’s Rick Stevens on Energy, AI, and a New Kind of Science

June 17, 2024

The world is currently experiencing two of the largest societal upheavals since the beginning of the Industrial Revolution. One is the rapid improvement and implementation of artificial intelligence (AI) tools, while the Read more…

HPE and NVIDIA Join Forces and Plan Conquest of Enterprise AI Frontier

June 20, 2024

The HPE Discover 2024 conference is currently in full swing, and the keynote address from Hewlett-Packard Enterprise (HPE) CEO Antonio Neri on Tuesday, June 18, Read more…

Slide Shows Samsung May be Developing a RISC-V CPU for In-memory AI Chip

June 19, 2024

Samsung may have unintentionally revealed its intent to develop a RISC-V CPU, which a presentation slide showed may be used in an AI chip. The company plans to Read more…

Qubits 2024: D-Wave’s Steady March to Quantum Success

June 18, 2024

In his opening keynote at D-Wave’s annual Qubits 2024 user meeting, being held in Boston, yesterday and today, CEO Alan Baratz again made the compelling pitch Read more…

Shutterstock_666139696

Argonne’s Rick Stevens on Energy, AI, and a New Kind of Science

June 17, 2024

The world is currently experiencing two of the largest societal upheavals since the beginning of the Industrial Revolution. One is the rapid improvement and imp Read more…

Under The Wire: Nearly HPC News (June 13, 2024)

June 13, 2024

As managing editor of the major global HPC news source, the term "news fire hose" is often mentioned. The analogy is quite correct. In any given week, there are Read more…

Labs Keep Supercomputers Alive for Ten Years as Vendors Pull Support Early

June 12, 2024

Laboratories are running supercomputers for much longer, beyond the typical lifespan, as vendors prematurely deprecate the hardware and stop providing support. 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…

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…

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…

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…

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…

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…

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…

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…

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…

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…

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…

AMD Clears Up Messy GPU Roadmap, Upgrades Chips Annually

June 3, 2024

In the world of AI, there's a desperate search for an alternative to Nvidia's GPUs, and AMD is stepping up to the plate. AMD detailed its updated GPU roadmap, w Read more…

  • arrow
  • Click Here for More Headlines
  • arrow
HPCwire