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 industy updates delivered to you every week!

Energy Exascale Earth System Model Version 2 Promises Twice the Speed

October 18, 2021

The Energy Exascale Earth System Model (E3SM) is an ongoing Department of Energy (DOE) earth system modeling, simulation and prediction project aiming to “assert and maintain an international scientific leadership posi Read more…

Intel Reorgs HPC Group, Creates Two ‘Super Compute’ Groups

October 15, 2021

Following on changes made in June that moved Intel’s HPC unit out of the Data Platform Group and into the newly created Accelerated Computing Systems and Graphics (AXG) business unit, led by Raja Koduri, Intel is making further updates to the HPC group and announcing... Read more…

Royalty-free stock illustration ID: 1938746143

MosaicML, Led by Naveen Rao, Comes Out of Stealth Aiming to Ease Model Training

October 15, 2021

With more and more enterprises turning to AI for a myriad of tasks, companies quickly find out that training AI models is expensive, difficult and time-consuming. Finding a new approach to deal with those cascading challenges is the aim of a new startup, MosaicML, that just came out of stealth... Read more…

NSF Awards $11M to SDSC, MIT and Univ. of Oregon to Secure the Internet

October 14, 2021

From a security standpoint, the internet is a problem. The infrastructure developed decades ago has cracked, leaked and been patched up innumerable times, leaving vulnerabilities that are difficult to address due to cost Read more…

SC21 Announces Science and Beyond Plenary: the Intersection of Ethics and HPC

October 13, 2021

The Intersection of Ethics and HPC will be the guiding topic of SC21's Science & Beyond plenary, inspired by the event tagline of the same name. The evening event will be moderated by Daniel Reed with panelists Crist Read more…

AWS Solution Channel

Cost optimizing Ansys LS-Dyna on AWS

Organizations migrate their high performance computing (HPC) workloads from on-premises infrastructure to Amazon Web Services (AWS) for advantages such as high availability, elastic capacity, latest processors, storage, and networking technologies; Read more…

Quantum Workforce – NSTC Report Highlights Need for International Talent

October 13, 2021

Attracting and training the needed quantum workforce to fuel the ongoing quantum information sciences (QIS) revolution is a hot topic these days. Last week, the U.S. National Science and Technology Council issued a report – The Role of International Talent in Quantum Information Science... Read more…

Intel Reorgs HPC Group, Creates Two ‘Super Compute’ Groups

October 15, 2021

Following on changes made in June that moved Intel’s HPC unit out of the Data Platform Group and into the newly created Accelerated Computing Systems and Graphics (AXG) business unit, led by Raja Koduri, Intel is making further updates to the HPC group and announcing... Read more…

Royalty-free stock illustration ID: 1938746143

MosaicML, Led by Naveen Rao, Comes Out of Stealth Aiming to Ease Model Training

October 15, 2021

With more and more enterprises turning to AI for a myriad of tasks, companies quickly find out that training AI models is expensive, difficult and time-consuming. Finding a new approach to deal with those cascading challenges is the aim of a new startup, MosaicML, that just came out of stealth... Read more…

Quantum Workforce – NSTC Report Highlights Need for International Talent

October 13, 2021

Attracting and training the needed quantum workforce to fuel the ongoing quantum information sciences (QIS) revolution is a hot topic these days. Last week, the U.S. National Science and Technology Council issued a report – The Role of International Talent in Quantum Information Science... Read more…

Eni Returns to HPE for ‘HPC4’ Refresh via GreenLake

October 13, 2021

Italian energy company Eni is upgrading its HPC4 system with new gear from HPE that will be installed in Eni’s Green Data Center in Ferrera Erbognone (a provi Read more…

The Blueprint for the National Strategic Computing Reserve

October 12, 2021

Over the last year, the HPC community has been buzzing with the possibility of a National Strategic Computing Reserve (NSCR). An in-utero brainchild of the COVID-19 High-Performance Computing Consortium, an NSCR would serve as a Merchant Marine for urgent computing... Read more…

UCLA Researchers Report Largest Chiplet Design and Early Prototyping

October 12, 2021

What’s the best path forward for large-scale chip/system integration? Good question. Cerebras has set a high bar with its wafer scale engine 2 (WSE-2); it has 2.6 trillion transistors, including 850,000 cores, and was fabricated using TSMC’s 7nm process on a roughly 8” x 8” silicon footprint. Read more…

What’s Next for EuroHPC: an Interview with EuroHPC Exec. Dir. Anders Dam Jensen

October 7, 2021

One year after taking the post as executive director of the EuroHPC JU, Anders Dam Jensen reviews the project's accomplishments and details what's ahead as EuroHPC's operating period has now been extended out to the year 2027. Read more…

University of Bath Unveils Janus, an Azure-Based Cloud HPC Environment

October 6, 2021

The University of Bath is upgrading its HPC infrastructure, which it says “supports a growing and wide range of research activities across the University.” Read more…

Ahead of ‘Dojo,’ Tesla Reveals Its Massive Precursor Supercomputer

June 22, 2021

In spring 2019, Tesla made cryptic reference to a project called Dojo, a “super-powerful training computer” for video data processing. Then, in summer 2020, Tesla CEO Elon Musk tweeted: “Tesla is developing a [neural network] training computer... Read more…

Enter Dojo: Tesla Reveals Design for Modular Supercomputer & D1 Chip

August 20, 2021

Two months ago, Tesla revealed a massive GPU cluster that it said was “roughly the number five supercomputer in the world,” and which was just a precursor to Tesla’s real supercomputing moonshot: the long-rumored, little-detailed Dojo system. Read more…

Esperanto, Silicon in Hand, Champions the Efficiency of Its 1,092-Core RISC-V Chip

August 27, 2021

Esperanto Technologies made waves last December when it announced ET-SoC-1, a new RISC-V-based chip aimed at machine learning that packed nearly 1,100 cores onto a package small enough to fit six times over on a single PCIe card. Now, Esperanto is back, silicon in-hand and taking aim... Read more…

CentOS Replacement Rocky Linux Is Now in GA and Under Independent Control

June 21, 2021

The Rocky Enterprise Software Foundation (RESF) is announcing the general availability of Rocky Linux, release 8.4, designed as a drop-in replacement for the soon-to-be discontinued CentOS. The GA release is launching six-and-a-half months... Read more…

US Closes in on Exascale: Frontier Installation Is Underway

September 29, 2021

At the Advanced Scientific Computing Advisory Committee (ASCAC) meeting, held by Zoom this week (Sept. 29-30), it was revealed that the Frontier supercomputer is currently being installed at Oak Ridge National Laboratory in Oak Ridge, Tenn. The staff at the Oak Ridge Leadership... Read more…

Intel Completes LLVM Adoption; Will End Updates to Classic C/C++ Compilers in Future

August 10, 2021

Intel reported in a blog this week that its adoption of the open source LLVM architecture for Intel’s C/C++ compiler is complete. The transition is part of In Read more…

Intel Reorgs HPC Group, Creates Two ‘Super Compute’ Groups

October 15, 2021

Following on changes made in June that moved Intel’s HPC unit out of the Data Platform Group and into the newly created Accelerated Computing Systems and Graphics (AXG) business unit, led by Raja Koduri, Intel is making further updates to the HPC group and announcing... Read more…

Hot Chips: Here Come the DPUs and IPUs from Arm, Nvidia and Intel

August 25, 2021

The emergence of data processing units (DPU) and infrastructure processing units (IPU) as potentially important pieces in cloud and datacenter architectures was Read more…

Leading Solution Providers

Contributors

AMD-Xilinx Deal Gains UK, EU Approvals — China’s Decision Still Pending

July 1, 2021

AMD’s planned acquisition of FPGA maker Xilinx is now in the hands of Chinese regulators after needed antitrust approvals for the $35 billion deal were receiv Read more…

HPE Wins $2B GreenLake HPC-as-a-Service Deal with NSA

September 1, 2021

In the heated, oft-contentious, government IT space, HPE has won a massive $2 billion contract to provide HPC and AI services to the United States’ National Security Agency (NSA). Following on the heels of the now-canceled $10 billion JEDI contract (reissued as JWCC) and a $10 billion... 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…

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…

Quantum Roundup: IBM, Rigetti, Phasecraft, Oxford QC, China, and More

July 13, 2021

IBM yesterday announced a proof for a quantum ML algorithm. A week ago, it unveiled a new topology for its quantum processors. Last Friday, the Technical Univer Read more…

The Latest MLPerf Inference Results: Nvidia GPUs Hold Sway but Here Come CPUs and Intel

September 22, 2021

The latest round of MLPerf inference benchmark (v 1.1) results was released today and Nvidia again dominated, sweeping the top spots in the closed (apples-to-ap Read more…

Frontier to Meet 20MW Exascale Power Target Set by DARPA in 2008

July 14, 2021

After more than a decade of planning, the United States’ first exascale computer, Frontier, is set to arrive at Oak Ridge National Laboratory (ORNL) later this year. Crossing this “1,000x” horizon required overcoming four major challenges: power demand, reliability, extreme parallelism and data movement. Read more…

Intel Unveils New Node Names; Sapphire Rapids Is Now an ‘Intel 7’ CPU

July 27, 2021

What's a preeminent chip company to do when its process node technology lags the competition by (roughly) one generation, but outmoded naming conventions make i Read more…

  • arrow
  • Click Here for More Headlines
  • arrow
HPCwire