Data Management at NERSC in the Era of Petascale Deep Learning

By Rob Farber

May 9, 2018

Now that computer scientists at Lawrence Berkeley National Laboratory’s National Energy Research Scientific Computing Center (NERSC) have demonstrated 15 petaflops deep-learning training performance on the Cray Cori supercomputer, the NERSC staff is working to address the data management issues that arise when running production deep-learning codes at such scale. The existing deep learning tools were not designed to efficiently ingest or manage the terabyte- to petabyte-sized deep-learning training sets that scientists can now use on this leadership class supercomputer. “Enabling the NERSC user community to perform deep learning at scale on Cori,” Quincey Koziol (Staff, Berkeley Lab) observes, “means scientists can use deep learning as part of their leading-edge scientific efforts.”

Thus NERSC staff are working to break new ground in adapting existing deep-learning frameworks to run efficiently at scale on thousands of nodes while giving researchers the ability to create and manage training sets containing tens to hundreds of terabytes of data in a portable fashion. For these datasets, it is imperative that they are formatted so Cori can ingest them efficiently at runtime.

Appreciating the magnitude of the petascale data management problem

To appreciate the magnitude of the petascale data management problem, consider that the 9,600 Intel Xeon Phi nodes used in the 15 petaflops deep learning training performance contained over a petabyte of main memory. (Specifically, 921.6 terabytes of DDR4 RAM and 153.6 terabytes of high-bandwidth 3D stacked memory.)

The first petascale training runs on the Cray XC40 Cori supercomputer focused on scalability, which left lots of room for groundbreaking research in training on really big datasets. Kurth, et.al. noted in their paper “Deep Learning at 15PF: Supervised and Semi-Supervised Classification for Scientific Data” that the climate dataset contained 15 TB of data and the HEP (High Energy Physics) data contained 10 million images. With more than a petabyte of RAM contained in 9,600 nodes, Cori can obviously utilize much larger data sets.

Not so obvious are the asynchronous data management issues that crop up after the data has been ingested and the training run has started. These asynchronous methods use prefetching and lots of communications, so per-node memory usage and network performance are critical to running at the petascale.

Without getting too technical, the 15 petaflops deep learning performance was achieved using a hybrid, asynchronous implementation of the SGD (stochastic gradient descent) numerical optimization method. SGD is a common numerical method used by popular packages such as Caffe (used in the 15 petaflops Cori runs) and TensorFlow.

Thorsten Kurth (Application Performance Specialist, NERSC) observes that, “Tensorflow is the most widely used framework and is therefore a primary optimization target at the moment, but the deep learning software world changes rapidly so that sustainable implementations are necessary. Thus it makes sense to create libraries of optimized kernels that can be used by many deep learning frameworks. This same idea can be used to create methods for the data feeding/IO operations.” These optimized libraries can then be rapidly adopted to new upcoming frameworks such as pytorch and mxnet, Kurth observes.

Addressing the challenges

Given the popularity of TensorFlow, the NERSC team is working to adapt TensorFlow to run at scale on Cori. The main challenges, Koziol observes, are threefold:

  • TensorFlow uses text or binary images for input rather than HDF5 or another data format typically used by HPC scientists. Koziol and NERSC are currently integrating HDF5 with TensorFlow.
  • TensorFlow uses a client-server model rather than MPI, which is the typical communications package for scientific applications that run on HPC systems. This means that there are no collective operations inside TensorFlow, which can cause performance issues.
  • TensorFlow uses an asynchronous training that is very loosely coupled, which means data prefetching is critical to prevent performance from suffering due to data starvation. Conversely, prefetching increases the per-node memory consumption, so an appropriate balance must be struck to prefetch “just enough and no more.” Finding that ideal balance without overburdening any node or set of nodes with data in a large (think hundred- or thousand-node) training run is a fertile research area as NERSC brings TensorFlow into a new scaling realm.

HDF5 integration

The data management aspects of deep learning are often overlooked as researchers work to speed training and find the right ANN (Artificial Neural Network) architecture(s) to solve complex problems.

In reality, much of a data scientist’s time is spent creating a clean, representative dataset for training. The data challenge becomes that much larger and unwieldly when creating data for a petascale, deep-learning-capable, leadership-class supercomputer like Cori. Data management is sometimes referred to as the Victorian Era Child of the 21st Century – to be seen and not heard. Unfortunately, the challenges associated with Cori-sized datasets simply cannot be ignored.

Prior to joining NERSC, Koziol was director of core software and high-performance computing at the HDF Group, where he spent 11 years developing the HDF5 I/O middleware package and overseeing the group’s HPC development efforts. This makes Koziol a natural to incorporate the versatile HDF5 data model into TensorFlow. HDF5 is a Hierarchical Data Format that can represent very large, complex numerical datasets along with their metadata in a portable format that can be moved between machines. HDF5 1.10.2 is the current, latest version. The specification is open, and the tools are open source. Development of HDF5 is done by the HDF Group, a nonprofit corporation.

The benefits of HDF5 integration into TensorFlow means that scientists can use tools and a data format that have been developed over decades to enable scientists to portably manage even the largest scientific datasets. Portability means the data preprocessing and data cleaning can happen on remote systems using familiar open-source tools and frameworks. Once ready, the data can be moved onto Cori and ingested into TensorFlow. According to Koziol, this helps address the challenge of “How do we get data into the system fast enough?”

Those who are interested can find the scripts and one example of HDF5 integration in the NERSC cori-tf-distributed-examples repository on github. Specifically, https://github.com/NERSC/cori-tf-distributed-examples.

Other work in progress

NERSC is also working to address TensorFlow’s memory consumption issue and speed the collective operations. However, these are non-trivial problems that will take time. As Koziol observes, “The MPI community has been thinking about collectives for about 20 years. TensorFlow is currently only about two years old.”

Along with the per-node memory consumption challenges that must be addressed when using asynchronous training methods, researchers are also rapidly increasing the complexity of the ANNs they use to solve complex problems. Deeper and more complex ANNs utilize more parameters, which further exacerbates the memory consumed per node problem. For example, calculating the gradient for SGD in TensorFlow is becoming an issue even when running on small systems.

The NERSC team has to contend with those issues as well as prefetching and buffering of data used to support the asynchronous operations during training, so the CPU is used as effectively as possible. The large memory of the Intel Xeon Phi nodes helps, as does the fact that the data extraction and training both occur on the CPU, but finding the right configuration can be challenging, Koziol notes. “Sometimes it helps to have a small number of fat nodes,” he observes.

Steps to the future

Koziol emphasizes that deep learning workloads stress the data ingest capabilities of current supercomputers. He hopes future supercomputer designs will incorporate more features to speed data ingest for data-intensive workloads like deep learning.

Current supercomputer designs have focused on burst buffers for checkpoint/restart, a common write-optimized I/O operation used in modeling and simulation software in which the state of the simulation is quickly saved (the checkpoint operation) so that thousands of hours of compute time won’t be lost in the event of a failure. In the unlikely event that something bad does happen, the supercomputer simply reloads the last checkpoint from storage (a restart operation) and continues with the calculation once the problem is fixed. The frequency of the checkpoint operation dictates how much supercomputer runtime will be lost in the event of a failure.

As deep learning becomes an ever more common workload on supercomputers, Koziol envisions a future where supercomputers are specifically designed to support faster data ingest for deep learning and other data-intensive workloads.

Summary

The NERSC Cori supercomputer has made the training of deep-learning ANNs a member of the petascale application club. Now the NERSC data management team is working to make this petascale capability available to its users to facilitate their ability to perform leading-edge science. Incorporating HDF5 into TensorFlow is an excellent beginning to making TensorFlow a petascale-capable platform for deep learning.

Rob Farber is a global technology consultant and author with an extensive background in HPC and advanced computational technology that he applies at national labs and commercial organizations. He can be reached at [email protected]

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!

What’s New in Computing vs. COVID-19: AMD, Remdesivir, Fab Spending & More

September 29, 2020

Supercomputing, big data and artificial intelligence are crucial tools in the fight against the coronavirus pandemic. Around the world, researchers, corporations and governments are urgently devoting their computing reso Read more…

By Oliver Peckham

Global QC Market Projected to Grow to More Than $800 million by 2024

September 28, 2020

The Quantum Economic Development Consortium (QED-C) and Hyperion Research are projecting that the global quantum computing (QC) market - worth an estimated $320 million in 2020 - will grow at an anticipated 27% CAGR betw Read more…

By Staff Reports

DoE’s ASCAC Backs AI for Science Program that Emulates the Exascale Initiative

September 28, 2020

Roughly a year after beginning formal efforts to explore an AI for Science initiative the Department of Energy’s Advanced Scientific Computing Advisory Committee last week accepted a subcommittee report calling for a t Read more…

By John Russell

Supercomputer Research Aims to Supercharge COVID-19 Antiviral Remdesivir

September 25, 2020

Remdesivir is one of a handful of therapeutic antiviral drugs that have been proven to improve outcomes for COVID-19 patients, and as such, is a crucial weapon in the fight against the pandemic – especially in the abse Read more…

By Oliver Peckham

NOAA Announces Major Upgrade to Ensemble Forecast Model, Extends Range to 35 Days

September 23, 2020

A bit over a year ago, the United States’ Global Forecast System (GFS) received a major upgrade: a new dynamical core – its first in 40 years – called the finite-volume cubed-sphere, or FV3. Now, the National Oceanic and Atmospheric Administration (NOAA) is bringing the FV3 dynamical core to... Read more…

By Oliver Peckham

AWS Solution Channel

The Water Institute of the Gulf runs compute-heavy storm surge and wave simulations on AWS

The Water Institute of the Gulf (Water Institute) runs its storm surge and wave analysis models on Amazon Web Services (AWS)—a task that sometimes requires large bursts of compute power. Read more…

Intel® HPC + AI Pavilion

Berlin Institute of Health: Putting HPC to Work for the World

Researchers from the Center for Digital Health at the Berlin Institute of Health (BIH) are using science to understand the pathophysiology of COVID-19, which can help to inform the development of targeted treatments. Read more…

AI Silicon Startup Graphcore Launches Channel Partner Program

September 23, 2020

AI compute platform vendor Graphcore has launched its first formal global channel partner program to promote and boost the sales of its AI processors and blade computing products. The formalized, all-new Graphcore Elite Partner Program follows the company’s past history of working with several... Read more…

By Todd R. Weiss

DoE’s ASCAC Backs AI for Science Program that Emulates the Exascale Initiative

September 28, 2020

Roughly a year after beginning formal efforts to explore an AI for Science initiative the Department of Energy’s Advanced Scientific Computing Advisory Commit Read more…

By John Russell

NOAA Announces Major Upgrade to Ensemble Forecast Model, Extends Range to 35 Days

September 23, 2020

A bit over a year ago, the United States’ Global Forecast System (GFS) received a major upgrade: a new dynamical core – its first in 40 years – called the finite-volume cubed-sphere, or FV3. Now, the National Oceanic and Atmospheric Administration (NOAA) is bringing the FV3 dynamical core to... Read more…

By Oliver Peckham

Arm Targets HPC with New Neoverse Platforms

September 22, 2020

UK-based semiconductor design company Arm today teased details of its Neoverse roadmap, introducing V1 (codenamed Zeus) and N2 (codenamed Perseus), Arm’s second generation N-series platform. The chip IP vendor said the new platforms will deliver 50 percent and 40 percent more... Read more…

By Tiffany Trader

Oracle Cloud Deepens HPC Embrace with Launch of A100 Instances, Plans for Arm, More 

September 22, 2020

Oracle Cloud Infrastructure (OCI) continued its steady ramp-up of HPC capabilities today with a flurry of announcements. Topping the list is general availabilit Read more…

By John Russell

European Commission Declares €8 Billion Investment in Supercomputing

September 18, 2020

Just under two years ago, the European Commission formalized the EuroHPC Joint Undertaking (JU): a concerted HPC effort (comprising 32 participating states at c Read more…

By Oliver Peckham

Google Hires Longtime Intel Exec Bill Magro to Lead HPC Strategy

September 18, 2020

In a sign of the times, another prominent HPCer has made a move to a hyperscaler. Longtime Intel executive Bill Magro joined Google as chief technologist for hi Read more…

By Tiffany Trader

Future of Fintech on Display at HPC + AI Wall Street

September 17, 2020

Those who tuned in for Tuesday's HPC + AI Wall Street event got a peak at the future of fintech and lively discussion of topics like blockchain, AI for risk man Read more…

By Alex Woodie, Tiffany Trader and Todd R. Weiss

IBM’s Quantum Race to One Million Qubits

September 15, 2020

IBM today outlined its ambitious quantum computing technology roadmap at its virtual Quantum Summit. The eye-popping million qubit number is still far out, agrees IBM, but perhaps not that far out. Just as eye-popping is IBM’s nearer-term plan for a 1,000-plus qubit system named Condor... Read more…

By John Russell

Supercomputer-Powered Research Uncovers Signs of ‘Bradykinin Storm’ That May Explain COVID-19 Symptoms

July 28, 2020

Doctors and medical researchers have struggled to pinpoint – let alone explain – the deluge of symptoms induced by COVID-19 infections in patients, and what Read more…

By Oliver Peckham

Nvidia Said to Be Close on Arm Deal

August 3, 2020

GPU leader Nvidia Corp. is in talks to buy U.K. chip designer Arm from parent company Softbank, according to several reports over the weekend. If consummated Read more…

By George Leopold

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…

By Doug Black

Intel’s 7nm Slip Raises Questions About Ponte Vecchio GPU, Aurora Supercomputer

July 30, 2020

During its second-quarter earnings call, Intel announced a one-year delay of its 7nm process technology, which it says it will create an approximate six-month shift for its CPU product timing relative to prior expectations. The primary issue is a defect mode in the 7nm process that resulted in yield degradation... Read more…

By Tiffany Trader

Google Hires Longtime Intel Exec Bill Magro to Lead HPC Strategy

September 18, 2020

In a sign of the times, another prominent HPCer has made a move to a hyperscaler. Longtime Intel executive Bill Magro joined Google as chief technologist for hi Read more…

By Tiffany Trader

HPE Keeps Cray Brand Promise, Reveals HPE Cray Supercomputing Line

August 4, 2020

The HPC community, ever-affectionate toward Cray and its eponymous founder, can breathe a (virtual) sigh of relief. The Cray brand will live on, encompassing th Read more…

By Tiffany Trader

Neocortex Will Be First-of-Its-Kind 800,000-Core AI Supercomputer

June 9, 2020

Pittsburgh Supercomputing Center (PSC - a joint research organization of Carnegie Mellon University and the University of Pittsburgh) has won a $5 million award Read more…

By Tiffany Trader

European Commission Declares €8 Billion Investment in Supercomputing

September 18, 2020

Just under two years ago, the European Commission formalized the EuroHPC Joint Undertaking (JU): a concerted HPC effort (comprising 32 participating states at c Read more…

By Oliver Peckham

Leading Solution Providers

Contributors

Oracle Cloud Infrastructure Powers Fugaku’s Storage, Scores IO500 Win

August 28, 2020

In June, RIKEN shook the supercomputing world with its Arm-based, Fujitsu-built juggernaut: Fugaku. The system, which weighs in at 415.5 Linpack petaflops, topp Read more…

By Oliver Peckham

Google Cloud Debuts 16-GPU Ampere A100 Instances

July 7, 2020

On the heels of the Nvidia’s Ampere A100 GPU launch in May, Google Cloud is announcing alpha availability of the A100 “Accelerator Optimized” VM A2 instance family on Google Compute Engine. The instances are powered by the HGX A100 16-GPU platform, which combines two HGX A100 8-GPU baseboards using... Read more…

By Tiffany Trader

DOD Orders Two AI-Focused Supercomputers from Liqid

August 24, 2020

The U.S. Department of Defense is making a big investment in data analytics and AI computing with the procurement of two HPC systems that will provide the High Read more…

By Tiffany Trader

Supercomputer Modeling Tests How COVID-19 Spreads in Grocery Stores

April 8, 2020

In the COVID-19 era, many people are treating simple activities like getting gas or groceries with caution as they try to heed social distancing mandates and protect their own health. Still, significant uncertainty surrounds the relative risk of different activities, and conflicting information is prevalent. A team of Finnish researchers set out to address some of these uncertainties by... Read more…

By Oliver Peckham

Microsoft Azure Adds A100 GPU Instances for ‘Supercomputer-Class AI’ in the Cloud

August 19, 2020

Microsoft Azure continues to infuse its cloud platform with HPC- and AI-directed technologies. Today the cloud services purveyor announced a new virtual machine Read more…

By Tiffany Trader

Japan’s Fugaku Tops Global Supercomputing Rankings

June 22, 2020

A new Top500 champ was unveiled today. Supercomputer Fugaku, the pride of Japan and the namesake of Mount Fuji, vaulted to the top of the 55th edition of the To Read more…

By Tiffany Trader

Joliot-Curie Supercomputer Used to Build First Full, High-Fidelity Aircraft Engine Simulation

July 14, 2020

When industrial designers plan the design of a new element of a vehicle’s propulsion or exterior, they typically use fluid dynamics to optimize airflow and in Read more…

By Oliver Peckham

Intel Speeds NAMD by 1.8x: Saves Xeon Processor Users Millions of Compute Hours

August 12, 2020

Potentially saving datacenters millions of CPU node hours, Intel and the University of Illinois at Urbana–Champaign (UIUC) have collaborated to develop AVX-512 optimizations for the NAMD scalable molecular dynamics code. These optimizations will be incorporated into release 2.15 with patches available for earlier versions. Read more…

By Rob Farber

  • arrow
  • Click Here for More Headlines
  • arrow
Do NOT follow this link or you will be banned from the site!
Share This