A team of researchers from the Department of Energy’s Oak Ridge National Laboratory has married artificial intelligence and high-performance computing to achieve a peak speed of 20 petaflops in the generation and training of deep learning networks on the laboratory’s Titan supercomputer.
Deep learning is a burgeoning field of artificial intelligence that uses networks modeled after the human brain to “learn” how to distinguish features and patterns in vast datasets. Such networks hold great promise in the realization of numerous technologies, from self-driving cars to intelligent robots.
Due to its ability to make sense of massive amounts of data, researchers across the scientific spectrum are eager to refine deep learning and apply it to some of today’s most challenging science problems. One such effort is ORNL’s Advances in Machine Learning to Improve Scientific Discovery at Exascale and Beyond (ASCEND) project, which aims to use deep learning to make sense of the massive datasets produced by the world’s most sophisticated scientific experiments, such as those located at ORNL.
Analysis of such datasets generally requires existing neural networks to be modified, or novel networks designed and then “trained” so that they know precisely what to look for and can produce valid results.
This is a time-consuming and difficult task, but one that an ORNL team led by Robert Patton and including Steven Young and Travis Johnston recently demonstrated can be dramatically expedited with a capable computing system such as ORNL’s Titan, the nation’s fastest supercomputer for science.
To efficiently design neural networks capable of tackling scientific datasets and expediting breakthroughs, Patton’s team developed two codes for evolving (MENNDL) and fine-tuning (RAvENNA) deep neural network architectures.
Both codes can generate and train as many as 18,600 neural networks simultaneously. Peak performance can be estimated by randomly sampling, and then carefully profiling, several hundred of these independently trained networks.
Both codes achieved a peak performance of 20 petaflops, or 20 thousand trillion calculations per second, on Titan (or just under half of Titan’s single precision total peak performance). In practical terms, that translates to training 40-50,000 networks per hour.
“The real measure of success in the deep learning community is time-to-solution,” said Johnston. “And with a machine like Titan we are able to train an unparalleled number of highly accurate networks.”
Titan is a Cray hybrid system, meaning that it uses both traditional CPUs and graphics processing units (GPUs) to tackle complex calculations for big science problems efficiently; the GPUs also happen to be the processor of choice for training deep learning networks.
The team’s work demonstrates that with the right high-performance computing system researchers can efficiently train large numbers of networks, which can then be used to help them tackle today’s increasingly data-heavy experiments and simulations.
This efficient design of deep neural networks will enable researchers to deploy highly accurate, custom-designed models, saving both time and money by freeing the scientist from the task of designing a network from the ground up.
And because the OLCF’s next leadership computing system, Summit, features a deep-learning friendly architecture with enhanced GPUs and complementary Tensor cores, the team is confident both codes will only get faster.
“Out of the box, without tuning to Summit’s unique architecture, we are expecting an increase in performance up to 50 times,” said Johnston.
With that sort of network training capability, Summit could be indispensable to researchers across the scientific spectrum looking to deep learning to help them tackle some of science’s most immense challenges.
Patton’s team is not waiting for the improved hardware to start tackling current scientific data challenges; they have already deployed their codes to assist domain scientists at the Department of Energy’s Fermilab in Batavia, Illinois.
Researchers at Fermilab used MENNDL to better understand how neutrinos interact with ordinary matter by producing a classification network to support their Main Injector Experiment for v-A (MINERvA), a neutrino scattering experiment. The task, known as vertex reconstruction, required a network to analyze images and precisely identify the location where neutrinos interact with one of many targets—a task akin to finding the aerial source of a starburst of fireworks.
In only 24 hours, MENNDL produced optimized networks that outperformed any previously handcrafted network—an achievement that could easily have taken scientists months to accomplish. To identify the high-performing network, MENNDL evaluated approximately 500,000 neural networks, training them on a data set consisting of 800,000 images of neutrino events, steadily using 18,000 of Titan’s nodes.
“You need something like MENNDL to explore this effectively infinite space of possible networks, but you want to do it efficiently,” Young said. “What Titan does is bring the time to solution down to something practical.”
And with Summit to come online this year, the future of deep learning in big science looks bright indeed.
The Latest on: Deep learning networks
Video: Exascale Deep Learning for Climate Analytics
on May 20, 2019 at 7:49 am
We’ll discuss how we scaled the training of a single deep learning model to 27,360 V100 GPUs (4,560 nodes) on the OLCF Summit HPC System using the high-productivity TensorFlow framework. We discuss ... […]
Framework improves continual learning for artificial intelligence
on May 19, 2019 at 12:13 am
“To date, AI systems using deep neural networks have not been very good at this.” “Deep neural network AI systems are designed for learning narrow tasks,” said Xilai Li, a Ph.D. candidate at NC State. ... […]
CosmoGAN Neural Network to Study Dark Matter
on May 18, 2019 at 9:51 am
In a paper published May 6, 2019 in Computational Astrophysics and Cosmology, they discuss their new deep learning network, dubbed CosmoGAN, and its ability to create high-fidelity, weak gravitational ... […]
Tech Data Partners with Deep Instinct to Provide Deep Learning that Thwarts Cyberattacks
on May 15, 2019 at 2:21 pm
Deep learning methods are based on the layers used in artificial neural networks and are modeled after how the human brain learns. Deep Instinct’s deep neural network learns to predict and ... […]
Deep Learning Tool Identifies Disease Markers of Alzheimer’s
on May 15, 2019 at 10:13 am
Deep learning and other forms of artificial intelligence will ... and accuracy of the resulting neural network models and support training of more sophisticated model architectures,” the researchers ... […]
Deep Learning Competitors Stalk Nvidia
on May 14, 2019 at 12:06 pm
First, Intel researchers claimed a new deep learning record for image classification on the ResNet-50 convolutional neural network. Separately, Israeli AI chip startup Hailo.ai released a new deep ... […]
AI Chipmaker Hailo Releases Industry-leading Deep Learning Processor
on May 14, 2019 at 10:08 am
By designing an architecture that relies on the core properties of neural networks, edge devices can now run deep learning applications at full scale more efficiently, effectively, and sustainably ... […]
An Intro to Deep Learning
on May 14, 2019 at 6:38 am
I wanted to mention a timely new book, Deep Learning for the Life Sciences ... if you would like to know in brief about (say) what a neural network is, the general scheme by which it processes inputs ... […]
AI chip startup Hailo announces specialized deep learning processor
on May 14, 2019 at 6:14 am
The company, which is focused on the automotive sector, says its chip enables devices to run “sophisticated” deep learning applications at the edge of the network that could previously be ... […]
Hailo launches its newest deep learning chip
on May 14, 2019 at 6:06 am
“By designing an architecture that relies on the core properties of neural networks, edge devices can now run deep learning applications at full scale more efficiently, effectively, and ... […]
via Google News and Bing News