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!

Verifying the Universe with Exascale Computers

July 30, 2021

The ExaSky project, one of the critical Earth and Space Science applications being solved by the US Department of Energy’s (DOE’s) Exascale Computing Project (ECP), is preparing to use the nation’s forthcoming exas Read more…

What’s After Exascale? The Internet of Workflows Says HPE’s Nicolas Dubé

July 29, 2021

With the race to exascale computing in its final leg, it’s natural to wonder what the Post Exascale Era will look like. Nicolas Dubé, VP and chief technologist for HPE’s HPC business unit, agrees and shared his vision at Supercomputing Frontiers Europe 2021 held last week. The next big thing, he told the virtual audience at SFE21, is something that will connect HPC and (broadly) all of IT – into what Dubé calls The Internet of Workflows. Read more…

How UK Scientists Developed Transformative, HPC-Powered Coronavirus Sequencing System

July 29, 2021

In November 2020, the COVID-19 Genomics UK Consortium (COG-UK) won the HPCwire Readers’ Choice Award for Best HPC Collaboration for its CLIMB-COVID sequencing project. Launched in March 2020, CLIMB-COVID has now resulted in the sequencing of over 675,000 coronavirus genomes – an increasingly critical task as variants like Delta threaten the tenuous prospect of a return to normalcy in much of the world. Read more…

KAUST Leverages Mixed Precision for Geospatial Data

July 28, 2021

For many computationally intensive tasks, exacting precision is not necessary for every step of the entire task to obtain a suitably precise result. The alternative is mixed-precision computing: using high precision wher Read more…

Oak Ridge Supercomputer Enables Next-Gen Jet Turbine Research

July 27, 2021

Air travel is notoriously carbon-inefficient, with many airlines going as far as to offer purchasable carbon offsets to ease the guilt over large-footprint travel. But even over just the last decade, major aircraft model Read more…

AWS Solution Channel

Data compression with increased performance and lower costs

Many customers associate a performance cost with data compression, but that’s not the case with Amazon FSx for Lustre. With FSx for Lustre, data compression reduces storage costs and increases aggregate file system throughput. Read more…

IBM and University of Tokyo Roll Out Quantum System One in Japan

July 27, 2021

IBM and the University of Tokyo today unveiled an IBM Quantum System One as part of the IBM-Japan quantum program announced in 2019. The system is the second IBM Quantum System One assembled outside the U.S. and follows Read more…

What’s After Exascale? The Internet of Workflows Says HPE’s Nicolas Dubé

July 29, 2021

With the race to exascale computing in its final leg, it’s natural to wonder what the Post Exascale Era will look like. Nicolas Dubé, VP and chief technologist for HPE’s HPC business unit, agrees and shared his vision at Supercomputing Frontiers Europe 2021 held last week. The next big thing, he told the virtual audience at SFE21, is something that will connect HPC and (broadly) all of IT – into what Dubé calls The Internet of Workflows. Read more…

How UK Scientists Developed Transformative, HPC-Powered Coronavirus Sequencing System

July 29, 2021

In November 2020, the COVID-19 Genomics UK Consortium (COG-UK) won the HPCwire Readers’ Choice Award for Best HPC Collaboration for its CLIMB-COVID sequencing project. Launched in March 2020, CLIMB-COVID has now resulted in the sequencing of over 675,000 coronavirus genomes – an increasingly critical task as variants like Delta threaten the tenuous prospect of a return to normalcy in much of the world. Read more…

IBM and University of Tokyo Roll Out Quantum System One in Japan

July 27, 2021

IBM and the University of Tokyo today unveiled an IBM Quantum System One as part of the IBM-Japan quantum program announced in 2019. The system is the second IB 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 it seem like it's two nodes behind? For Intel, the response was to change how it refers to its nodes with the aim of better reflecting its positioning within the leadership semiconductor manufacturing space. Intel revealed its new node nomenclature, and... Read more…

Will Approximation Drive Post-Moore’s Law HPC Gains?

July 26, 2021

“Hardware-based improvements are going to get more and more difficult,” said Neil Thompson, an innovation scholar at MIT’s Computer Science and Artificial Intelligence Lab (CSAIL). “I think that’s something that this crowd will probably, actually, be already familiar with.” Thompson, speaking... Read more…

With New Owner and New Roadmap, an Independent Omni-Path Is Staging a Comeback

July 23, 2021

Put on a shelf by Intel in 2019, Omni-Path faced a uncertain future, but under new custodian Cornelis Networks, OmniPath is looking to make a comeback as an independent high-performance interconnect solution. A "significant refresh" – called Omni-Path Express – is coming later this year according to the company. Cornelis Networks formed last September as a spinout of Intel's Omni-Path division. Read more…

Chameleon’s HPC Testbed Sharpens Its Edge, Presses ‘Replay’

July 22, 2021

“One way of saying what I do for a living is to say that I develop scientific instruments,” said Kate Keahey, a senior fellow at the University of Chicago a Read more…

Summer Reading: “High-Performance Computing Is at an Inflection Point”

July 21, 2021

At last month’s 11th International Symposium on Highly Efficient Accelerators and Reconfigurable Technologies (HEART), a group of researchers led by Martin Schulz of the Leibniz Supercomputing Center (Munich) presented a “position paper” in which they argue HPC architectural landscape... Read more…

AMD Chipmaker TSMC to Use AMD Chips for Chipmaking

May 8, 2021

TSMC has tapped AMD to support its major manufacturing and R&D workloads. AMD will provide its Epyc Rome 7702P CPUs – with 64 cores operating at a base cl Read more…

Intel Launches 10nm ‘Ice Lake’ Datacenter CPU with Up to 40 Cores

April 6, 2021

The wait is over. Today Intel officially launched its 10nm datacenter CPU, the third-generation Intel Xeon Scalable processor, codenamed Ice Lake. With up to 40 Read more…

Berkeley Lab Debuts Perlmutter, World’s Fastest AI Supercomputer

May 27, 2021

A ribbon-cutting ceremony held virtually at Berkeley Lab's National Energy Research Scientific Computing Center (NERSC) today marked the official launch of Perlmutter – aka NERSC-9 – the GPU-accelerated supercomputer built by HPE in partnership with Nvidia and AMD. 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 called Dojo to process truly vast amounts of video data. It’s a beast! … A truly useful exaflop at de facto FP32.” Read more…

Google Launches TPU v4 AI Chips

May 20, 2021

Google CEO Sundar Pichai spoke for only one minute and 42 seconds about the company’s latest TPU v4 Tensor Processing Units during his keynote at the Google I 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 after Red Hat deprecated its support for the widely popular, free CentOS server operating system. The Rocky Linux development effort... Read more…

Iran Gains HPC Capabilities with Launch of ‘Simorgh’ Supercomputer

May 18, 2021

Iran is said to be developing domestic supercomputing technology to advance the processing of scientific, economic, political and military data, and to strengthen the nation’s position in the age of AI and big data. On Sunday, Iran unveiled the Simorgh supercomputer, which will deliver.... Read more…

HPE Launches Storage Line Loaded with IBM’s Spectrum Scale File System

April 6, 2021

HPE today launched a new family of storage solutions bundled with IBM’s Spectrum Scale Erasure Code Edition parallel file system (description below) and featu Read more…

Leading Solution Providers

Contributors

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…

GTC21: Nvidia Launches cuQuantum; Dips a Toe in Quantum Computing

April 13, 2021

Yesterday Nvidia officially dipped a toe into quantum computing with the launch of cuQuantum SDK, a development platform for simulating quantum circuits on GPU-accelerated systems. As Nvidia CEO Jensen Huang emphasized in his keynote, Nvidia doesn’t plan to build... Read more…

Microsoft to Provide World’s Most Powerful Weather & Climate Supercomputer for UK’s Met Office

April 22, 2021

More than 14 months ago, the UK government announced plans to invest £1.2 billion ($1.56 billion) into weather and climate supercomputing, including procuremen 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…

Q&A with Jim Keller, CTO of Tenstorrent, and an HPCwire Person to Watch in 2021

April 22, 2021

As part of our HPCwire Person to Watch series, we are happy to present our interview with Jim Keller, president and chief technology officer of Tenstorrent. One of the top chip architects of our time, Keller has had an impactful career. Read more…

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…

Senate Debate on Bill to Remake NSF – the Endless Frontier Act – Begins

May 18, 2021

The U.S. Senate today opened floor debate on the Endless Frontier Act which seeks to remake and expand the National Science Foundation by creating a technology Read more…

  • arrow
  • Click Here for More Headlines
  • arrow
HPCwire