Deep Learning Paves Way for Better Diagnostics

By Tiffany Trader

September 19, 2016

Stanford researchers are leveraging GPU-based machines in the Amazon EC2 cloud to run deep learning workloads with the goal of improving diagnostics for a chronic eye disease, called diabetic retinopathy. The disease is a complication of diabetes that can lead to blindness if blood sugar is poorly controlled. It affects about 45 percent of diabetics and 100 million people worldwide, many in developing nations.

Final-year Stanford PhD students Apaar Sadhwani and Jason Su got involved in developing the diagnostic solution as part of a class project and corresponding Kaggle competition that was held last year. Sponsor Amazon provided AWS cloud credits in support of the research.

diabetic-retinopathy_5-classes_sadhwani-su_400x
Source: Automatic Grading of Eye Diseases Through Deep Learning, 2016

After Kaggle, the duo decided to turn their research project into a cloud-based platform that hospitals and clinics can use to guide the diagnosis of eye diseases. Their approach relies on a convolutional neural net (CNN) that grades the severity of diabetic retinopathy disease states into five categories: 0-4, with 0 being normal and 4 being the most severe.

The researchers have been training their model with a data set of 80,000 images from EYEPACS, a web-based application for exchanging eye-related clinical information, run by the California Health Foundation. “Getting data is the most constraining part of applying deep learning to a medical setting,” said Sadhwani, “but we are working closely with partners to get more data.”

They’ve also had to address a class imbalance in the data set. “We have a lot more 0’s and 1’s than 3’s and 4’s, for example,” said Sadhwani. As the disease progresses to stage four (known as proliferative diabetic retinopathy, or PDR), image data is more rare. A total of about 10,000 stage four images are required for optimal results.

The training problem is run on AWS Elastic Compute Cloud (EC2) with single-GPU and multi-GPU nodes. Some S3 storage and Elastic Block Store (EBS) services are also employed. The training takes about three days to a week for a given model.

Within EC2, the researchers are using Starcluster which lets them build custom clusters among the nodes and network them together. They used a master node to store all their training data and up to 28 different training nodes. All these separate training nodes would access the master node so they wouldn’t have to mirror the data onto each of the nodes.

“With Starcluster and AWS you can bring up different node types independently on demand,” said Su. “So we would run this experiment that would only need a single-GPU node and then after that finished we could shut down that node and save money. Then we would scale it up to a larger resolution image and we would need four-GPU nodes for that – so we’d spin that up, train on that, and come back three days later and shut that off. AWS provides this flexibility for scaling up and scaling down for cost and for trying out different ideas.”

The researchers relied on AWS spot instance pricing to further improve the economics. Their program saves a state every “epoch,” which relates to one pass through the data set, so losing a node did not incur a big setback. With 55 epochs in a run, the most they would lose is 1/55th of their training progress.

They used the g2.2xlarge instance type and the g2.8xlarge instance type for training their final models. They trained two kinds of models, one on low-res images and the final model on high-res images, for which they employed the larger multi-GPU nodes.

Amazon’s GPU instances are based on older Nvidia GRID K520 graphics cards, which at 4 GB per GPU do not have an ideal memory profile for training based on very high-resolution images.

“Typically in deep learning, you have a 256×256 image, or about one-sixteenth of a megapixel and we’re at four megapixels, so memory is a huge part of doing this problem,” said Sadhwani. “Our workaround was to scale to 4-GPU nodes, which effectively had 4 gigabytes of memory each [GPU], but we lose some to overhead because we have to have the model independently at each of the separate GPUs. It would be more advantageous to have a single GPU with a full 16 gigabytes.”

Because their model was dealing with these high-resolution images, they used Torch to split it across the 4-GPU node to fine-tune its parameters. Currently, they are moving to a distributed training model, which enables several different nodes to train essentially the same model but with independent data. This gives them the ability to train one model across many GPUs, rather than a single model on a single GPU node and thus accelerates the training.

The researchers are eyeing clouds with higher-memory GPUs, which could mean holding out for upgraded Amazon instances or moving to the Microsoft Azure cloud with its Tesla K80s.

They are not interested in CPUs. “It would take significantly longer, at least a factor of 50,” said Sadhwani. “The kind of neural networks we are using [convolutional neural nets] harness parallelization a lot. Even if we were not using this special class of network, there is at least a 10x speedup going from CPUs to GPUs, but for this particular variety that speedup is magnified a lot more, in the neighborhood of 100x.”

Diabetic retinopathy is a disease of the blood vessels in the eye. As the sugar level in the blood rises, it causes the walls of the blood vessels to thin and eventually they’ll crack and bleed. The most important thing to look for is tiny dot bleeds, called hemorrhages. They are very small and difficult to locate even with advanced algorithms. The deep learning model must also be trained to ignore or flag likely camera artifacts, which appear in approximately 40 percent of the images, and can obscure identification of disease traits.

To address these challenges, the Stanford team’s approach uses two networks, a lesion detector and a main network. The lesion detector looks at a small part of the image and outputs a number between 0 and 1, a probability. The lesion detector has so far achieved an accuracy of 99 percent for negatives and 76 percent for positives. The purpose of the main network is to characterize details about where the disease-related features are with respect to the important parts of the eye.

deep-learning-fused-architecture_sadhwani-su_800x
Source: Automatic Grading of Eye Diseases Through Deep Learning, 2016

The outputs of these two pipelines are then fused together. This provides a way to combine low-level details about where there are dot hemorrhages with high-level information like which parts of the image should actually be ignored because they are corrupted by artifacts. The fuse network is responsible for integrating all these signals together to deliver a final probability for the disease class.

Right now the team has been working with five classes, but they say that in the clinical setting, these grades are not tracked with such granularity. In terms of intervention, there are really three stages: 0) no action is required; 1) monitor the progress of the disease; and 2) medical intervention such as surgery is required.

“Moving to three-classes would increase the accuracy of our models because it’s a simpler problem and easier to solve,” said Su.

The ultimate goal here is to deliver a digital assistant to radiologists, opthamologists and other clinicians, so they can screen more patients, more frequently.

“Using an automated tool to augment human resources, you can more closely monitor the changes in the disease state as they progress to more effectively treat the disease,” said Su.

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!

AMD’s EPYC Road to Redemption in Six Slides

June 21, 2018

A year ago AMD returned to the server market with its EPYC processor line. The earth didn’t tremble but folks took notice. People remember the Opteron fondly but later versions of the Bulldozer line not so much. Fast f Read more…

By John Russell

Why Student Cluster Competitions are Better than World Cup

June 21, 2018

My last article about the ISC18 Student Cluster Competition, titled “World Cup is Lame Compared to This Competition”, may have implied that I believe Student Cluster Competitions are better than World Cup soccer in s Read more…

By Dan Olds

European HPC Summit Week and PRACEdays 2018: Slaying Dragons and SHAPEing Futures One SME at a Time

June 20, 2018

The University of Ljubljana in Slovenia hosted the third annual EHPCSW18 and fifth annual PRACEdays18 events which opened May 29, 2018. The conference was chaired by PRACE Council Vice-Chair Sergi Girona (Barcelona Super Read more…

By Elizabeth Leake (STEM-Trek for HPCwire)

HPE Extreme Performance Solutions

HPC and AI Convergence is Accelerating New Levels of Intelligence

Data analytics is the most valuable tool in the digital marketplace – so much so that organizations are employing high performance computing (HPC) capabilities to rapidly collect, share, and analyze endless streams of data. Read more…

IBM Accelerated Insights

Preview the World’s Smartest Supercomputer at ISC 2018

Introducing an accelerated IT infrastructure for HPC & AI workloads Read more…

An Overview of ‘OpenACC for Programmers’ from the Book’s Editors

June 20, 2018

In an era of multicore processors coupled with manycore accelerators in all kinds of devices from smartphones all the way to supercomputers, it is important to train current and future computational scientists of all dom Read more…

By Sunita Chandrasekaran and Guido Juckeland

AMD’s EPYC Road to Redemption in Six Slides

June 21, 2018

A year ago AMD returned to the server market with its EPYC processor line. The earth didn’t tremble but folks took notice. People remember the Opteron fondly Read more…

By John Russell

European HPC Summit Week and PRACEdays 2018: Slaying Dragons and SHAPEing Futures One SME at a Time

June 20, 2018

The University of Ljubljana in Slovenia hosted the third annual EHPCSW18 and fifth annual PRACEdays18 events which opened May 29, 2018. The conference was chair Read more…

By Elizabeth Leake (STEM-Trek for HPCwire)

Cray Introduces All Flash Lustre Storage Solution Targeting HPC

June 19, 2018

Citing the rise of IOPS-intensive workflows and more affordable flash technology, Cray today introduced the L300F, a scalable all-flash storage solution whose p Read more…

By John Russell

Sandia to Take Delivery of World’s Largest Arm System

June 18, 2018

While the enterprise remains circumspect on prospects for Arm servers in the datacenter, the leadership HPC community is taking a bolder, brighter view of the x86 server CPU alternative. Amongst current and planned Arm HPC installations – i.e., the innovative Mont-Blanc project, led by Bull/Atos, the 'Isambard’ Cray XC50 going into the University of Bristol, and commitments from both Japan and France among others -- HPE is announcing that it will be supply the United States National Nuclear Security Administration (NNSA) with a 2.3 petaflops peak Arm-based system, named Astra. Read more…

By Tiffany Trader

The Machine Learning Hype Cycle and HPC

June 14, 2018

Like many other HPC professionals I’m following the hype cycle around machine learning/deep learning with interest. I subscribe to the view that we’re probably approaching the ‘peak of inflated expectation’ but not quite yet starting the descent into the ‘trough of disillusionment. This still raises the probability that... Read more…

By Dairsie Latimer

Xiaoxiang Zhu Receives the 2018 PRACE Ada Lovelace Award for HPC

June 13, 2018

Xiaoxiang Zhu, who works for the German Aerospace Center (DLR) and Technical University of Munich (TUM), was awarded the 2018 PRACE Ada Lovelace Award for HPC for her outstanding contributions in the field of high performance computing (HPC) in Europe. Read more…

By Elizabeth Leake

U.S Considering Launch of National Quantum Initiative

June 11, 2018

Sometime this month the U.S. House Science Committee will introduce legislation to launch a 10-year National Quantum Initiative, according to a recent report by Read more…

By John Russell

ORNL Summit Supercomputer Is Officially Here

June 8, 2018

Oak Ridge National Laboratory (ORNL) together with IBM and Nvidia celebrated the official unveiling of the Department of Energy (DOE) Summit supercomputer toda Read more…

By Tiffany Trader

MLPerf – Will New Machine Learning Benchmark Help Propel AI Forward?

May 2, 2018

Let the AI benchmarking wars begin. Today, a diverse group from academia and industry – Google, Baidu, Intel, AMD, Harvard, and Stanford among them – releas Read more…

By John Russell

How the Cloud Is Falling Short for HPC

March 15, 2018

The last couple of years have seen cloud computing gradually build some legitimacy within the HPC world, but still the HPC industry lies far behind enterprise I Read more…

By Chris Downing

US Plans $1.8 Billion Spend on DOE Exascale Supercomputing

April 11, 2018

On Monday, the United States Department of Energy announced its intention to procure up to three exascale supercomputers at a cost of up to $1.8 billion with th Read more…

By Tiffany Trader

Deep Learning at 15 PFlops Enables Training for Extreme Weather Identification at Scale

March 19, 2018

Petaflop per second deep learning training performance on the NERSC (National Energy Research Scientific Computing Center) Cori supercomputer has given climate Read more…

By Rob Farber

Lenovo Unveils Warm Water Cooled ThinkSystem SD650 in Rampup to LRZ Install

February 22, 2018

This week Lenovo took the wraps off the ThinkSystem SD650 high-density server with third-generation direct water cooling technology developed in tandem with par Read more…

By Tiffany Trader

ORNL Summit Supercomputer Is Officially Here

June 8, 2018

Oak Ridge National Laboratory (ORNL) together with IBM and Nvidia celebrated the official unveiling of the Department of Energy (DOE) Summit supercomputer toda Read more…

By Tiffany Trader

Nvidia Responds to Google TPU Benchmarking

April 10, 2017

Nvidia highlights strengths of its newest GPU silicon in response to Google's report on the performance and energy advantages of its custom tensor processor. Read more…

By Tiffany Trader

Hennessy & Patterson: A New Golden Age for Computer Architecture

April 17, 2018

On Monday June 4, 2018, 2017 A.M. Turing Award Winners John L. Hennessy and David A. Patterson will deliver the Turing Lecture at the 45th International Sympo Read more…

By Staff

Leading Solution Providers

SC17 Booth Video Tours Playlist

Altair @ SC17

Altair

AMD @ SC17

AMD

ASRock Rack @ SC17

ASRock Rack

CEJN @ SC17

CEJN

DDN Storage @ SC17

DDN Storage

Huawei @ SC17

Huawei

IBM @ SC17

IBM

IBM Power Systems @ SC17

IBM Power Systems

Intel @ SC17

Intel

Lenovo @ SC17

Lenovo

Mellanox Technologies @ SC17

Mellanox Technologies

Microsoft @ SC17

Microsoft

Penguin Computing @ SC17

Penguin Computing

Pure Storage @ SC17

Pure Storage

Supericro @ SC17

Supericro

Tyan @ SC17

Tyan

Univa @ SC17

Univa

Google Chases Quantum Supremacy with 72-Qubit Processor

March 7, 2018

Google pulled ahead of the pack this week in the race toward "quantum supremacy," with the introduction of a new 72-qubit quantum processor called Bristlecone. Read more…

By Tiffany Trader

Google I/O 2018: AI Everywhere; TPU 3.0 Delivers 100+ Petaflops but Requires Liquid Cooling

May 9, 2018

All things AI dominated discussion at yesterday’s opening of Google’s I/O 2018 developers meeting covering much of Google's near-term product roadmap. The e Read more…

By John Russell

Nvidia Ups Hardware Game with 16-GPU DGX-2 Server and 18-Port NVSwitch

March 27, 2018

Nvidia unveiled a raft of new products from its annual technology conference in San Jose today, and despite not offering up a new chip architecture, there were still a few surprises in store for HPC hardware aficionados. Read more…

By Tiffany Trader

Pattern Computer – Startup Claims Breakthrough in ‘Pattern Discovery’ Technology

May 23, 2018

If it weren’t for the heavy-hitter technology team behind start-up Pattern Computer, which emerged from stealth today in a live-streamed event from San Franci Read more…

By John Russell

Part One: Deep Dive into 2018 Trends in Life Sciences HPC

March 1, 2018

Life sciences is an interesting lens through which to see HPC. It is perhaps not an obvious choice, given life sciences’ relative newness as a heavy user of H Read more…

By John Russell

Intel Pledges First Commercial Nervana Product ‘Spring Crest’ in 2019

May 24, 2018

At its AI developer conference in San Francisco yesterday, Intel embraced a holistic approach to AI and showed off a broad AI portfolio that includes Xeon processors, Movidius technologies, FPGAs and Intel’s Nervana Neural Network Processors (NNPs), based on the technology it acquired in 2016. Read more…

By Tiffany Trader

Google Charts Two-Dimensional Quantum Course

April 26, 2018

Quantum error correction, essential for achieving universal fault-tolerant quantum computation, is one of the main challenges of the quantum computing field and it’s top of mind for Google’s John Martinis. At a presentation last week at the HPC User Forum in Tucson, Martinis, one of the world's foremost experts in quantum computing, emphasized... Read more…

By Tiffany Trader

Cray Rolls Out AMD-Based CS500; More to Follow?

April 18, 2018

Cray was the latest OEM to bring AMD back into the fold with introduction today of a CS500 option based on AMD’s Epyc processor line. The move follows Cray’ Read more…

By John Russell

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