University of Maryland Team Wins DOE Award to Advance AI Using Supercomputers

December 4, 2023

Dec. 4, 2023 — University of Maryland researchers have won a competitive award from the U.S. Department of Energy (DOE) that will provide them access to some of the world’s most powerful computational platforms.

The award—from the DOE’s Innovative and Novel Computational Impact on Theory and Experiment (INCITE) program—will enable UMD experts in high-performance computing and machine learning to scale distributed AI training and develop new AI vision and language models used in popular applications like DALL-E and ChatGPT.

The researchers will use federal facilities to develop and test their novel methods, with the INCITE award granting them significant run-time on several government supercomputers. This includes 600,000 node-hours on the Frontier supercomputer located at the Oak Ridge National Laboratory in Tennessee; 100,000 node-hours on the Polaris supercomputer located at the Argonne National Laboratory in Illinois; and 50,000 node-hours on the recently unveiled Aurora supercomputer that is also located in Illinois.

Training AI models that do not fit in the memory of a single graphics processing unit (GPU) requires large-scale distributed training on GPU-based supercomputers, said Abhinav Bhatele, an associate professor of computer science who is the principal investigator of the project.

With large language models (LLMs) and vision-based diffusion models now used for journalism, understanding protein structures, writing software code, and much more, training the largest of these AI models—which can reach hundreds of billions of parameters of information—can often overwhelm the computing capabilities of many academic labs doing work in these areas.

The UMD team seeks to “democratize” this process by using innovative parallel computing algorithms and other methods that enable platforms powered by NVIDIA and Advanced Micro Devices’ GPUs to efficiently train extremely large models, explained Bhatele. The researchers will also open source their work, Bhatele added, allowing input and feedback from a diverse group of users.

Bhatele is joined on the project by Tom Goldstein, a professor of computer science and director of the University of Maryland Center for Machine Learning, and Harshitha Menon, a computer scientist at the Lawrence Livermore National Laboratory in California.

Both Bhatele and Goldstein have joint appointments in the University of Maryland Institute for Advanced Computer Studies.

The UMD project was one of only 75 proposals nationwide chosen to receive a 2024 INCITE award. The program, established in 2003, incentivizes researchers from academia, government laboratories and industry to accelerate scientific discoveries and technological innovations involving large-scale, computationally intensive projects that address “grand challenges” in science and engineering.

“We are grateful to receive the INCITE award, both to advance the science and to allow our graduate students access to computational power not readily available at most universities,” said Bhatele.

In identifying the scope of work involved, the UMD researchers said that the “scaling” of large neural networks—allowing them to be processed on large-scale computing platforms—is not a trivial matter. It requires parallelizing and optimizing different computational and communication motifs such as dense and sparse tensor computations, irregular communication patterns, load imbalance issues, and fast filesystem access.

They plan on using a framework called AxoNN developed in Bhatele’s Parallel Software and Systems Group to analyze and optimize the performance and portability of training large models. This includes fine-tuning the refined models and greatly improving the system’s “inference”—which refers to the process wherein a trained model applies its learned knowledge to new, unseen data.

They also plan to explore efficient alternatives for transformer models for language modeling. Many of the modern LLMs currently use transformer models to perform a variety of natural language processing (NLP) tasks like identifying context—and thus meaning—by tracking relationships in sequential data like the words in this sentence.

“We intend to train variants of modern language model architectures that are directly aimed at usability constraints in smaller academic laboratories,” said Goldstein. “We want to focus on variants with smaller memory footprints and adaptive compute capabilities at deployment—traits that will enable more robust outcomes in the fields of machine learning and NLP.”

The researchers will also work on fine-tuning trained models for downstream tasks that have important real-world applications, Goldstein said.

One example is improving performance “explainability,” he said, which involves the concept of being able to understand and utilize the output from a machine learning model. This could be useful in instances where a healthcare model is predicting whether a patient is suffering from a particular disease or not.

Ultimately, the UMD team hopes their research will help scientists, physicians and policymakers better utilize the power of AI systems, whether that power is borne from one of the industry giants like Microsoft or NVIDIA, or from a three- or four-person laboratory at a college or university.

“We hope our work will lead to new discoveries from many of the smaller academic labs that don’t have 24/7 access to supercomputers—this includes our own labs on the University of Maryland campus that are filled to the brim with outstanding talent,” said Bhatele.


Source: UMIACS communications group

Subscribe to HPCwire's Weekly Update!

Be the most informed person in the room! Stay ahead of the tech trends with industry updates delivered to you every week!

A Big Memory Nvidia GH200 Next to Your Desk: Closer Than You Think

February 22, 2024

Students of the microprocessor may recall that the original 8086/8088 processors did not have floating point units. The motherboard often had an extra socket for an optional 8087 math coprocessor. The math coprocessor ma Read more…

IonQ Reports Advance on Path to Networked Quantum Computing

February 22, 2024

IonQ reported reaching a milestone in its efforts to use entangled photon-ion connectivity to scale its quantum computers. IonQ’s quantum computers are based on trapped ions which feature long coherence times and qubit Read more…

Apple Rolls out Post Quantum Security for iOS

February 21, 2024

Think implementing so-called Post Quantum Cryptography (PQC) isn't important because quantum computers able to decrypt current RSA codes don’t yet exist? Not Apple. Today the consumer electronics giant started rolling Read more…

GenAI Having Major Impact on Data Culture, Survey Says

February 21, 2024

While 2023 was the year of GenAI, the adoption rates for GenAI did not match expectations. Most organizations are continuing to invest in GenAI but are yet to derive any substantial value from it. However, the GenAI hyp Read more…

QED-C Issues New Quantum Benchmarking Paper

February 20, 2024

The Quantum Economic Development Consortium last week released a new paper on benchmarking – Quantum Algorithm Exploration using Application-Oriented Performance Benchmarks – that builds on earlier work and is an eff Read more…

AWS Solution Channel

Shutterstock 2283618597

Deep-dive into Ansys Fluent performance on Ansys Gateway powered by AWS

Today, we’re going to deep-dive into the performance and associated cost of running computational fluid dynamics (CFD) simulations on AWS using Ansys Fluent through the Ansys Gateway powered by AWS (or just “Ansys Gateway” for the rest of this post). Read more…

Atom Computing Reports Advance in Scaling Up Neutral Atom Qubit Arrays

February 15, 2024

The scale-up challenge facing quantum computing (QC) is daunting and varied. It’s commonly held that 1 million qubits (or more) will be needed to deliver practical fault tolerant QC. It’s also a varied challenge beca Read more…

A Big Memory Nvidia GH200 Next to Your Desk: Closer Than You Think

February 22, 2024

Students of the microprocessor may recall that the original 8086/8088 processors did not have floating point units. The motherboard often had an extra socket fo Read more…

Apple Rolls out Post Quantum Security for iOS

February 21, 2024

Think implementing so-called Post Quantum Cryptography (PQC) isn't important because quantum computers able to decrypt current RSA codes don’t yet exist? Not Read more…

QED-C Issues New Quantum Benchmarking Paper

February 20, 2024

The Quantum Economic Development Consortium last week released a new paper on benchmarking – Quantum Algorithm Exploration using Application-Oriented Performa Read more…

The Pulse of HPC: Tracking 4.5 Million Heartbeats of 3D Coronary Flow

February 15, 2024

Working in Duke University's Randles Lab, Cyrus Tanade, a National Science Foundation graduate student fellow and Ph.D. candidate in biomedical engineering, is Read more…

It Doesn’t Get Much SWEETER: The Winter HPC Computing Festival in Corpus Christi

February 14, 2024

(Main Photo by Visit Corpus Christi CrowdRiff) Texas A&M University's High-Performance Research Computing (HPRC) team hosted the "SWEETER Winter Comput Read more…

Q-Roundup: Diraq’s War Chest, DARPA’s Bet on Topological Qubits, Citi/Classiq Explore Optimization, WEF’s Quantum Blueprint

February 13, 2024

Yesterday, Australian start-up Diraq added $15 million to its war chest (now $120 million) to build a fault tolerant computer based on quantum dots. Last week D Read more…

2024 Winter Classic: Razor Thin Margins in HPL/HPCG

February 12, 2024

The first task for the 11 teams in the 2024 Winter Classic student cluster competition was to run and optimize the LINPACK and HPCG benchmarks. As usual, the Read more…

2024 Winter Classic: We’re Back!

February 9, 2024

The fourth edition of the Winter Classic Invitational Student Cluster Competition is up and running. This year, we have 11 teams of eager students representin Read more…

CORNELL I-WAY DEMONSTRATION PITS PARASITE AGAINST VICTIM

October 6, 1995

Ithaca, NY --Visitors to this year's Supercomputing '95 (SC'95) conference will witness a life-and-death struggle between parasite and victim, using virtual Read more…

SGI POWERS VIRTUAL OPERATING ROOM USED IN SURGEON TRAINING

October 6, 1995

Surgery simulations to date have largely been created through the development of dedicated applications requiring considerable programming and computer graphi Read more…

U.S. Will Relax Export Restrictions on Supercomputers

October 6, 1995

New York, NY -- U.S. President Bill Clinton has announced that he will definitely relax restrictions on exports of high-performance computers, giving a boost Read more…

Dutch HPC Center Will Have 20 GFlop, 76-Node SP2 Online by 1996

October 6, 1995

Amsterdam, the Netherlands -- SARA, (Stichting Academisch Rekencentrum Amsterdam), Academic Computing Services of Amsterdam recently announced that it has pur Read more…

Cray Delivers J916 Compact Supercomputer to Solvay Chemical

October 6, 1995

Eagan, Minn. -- Cray Research Inc. has delivered a Cray J916 low-cost compact supercomputer and Cray's UniChem client/server computational chemistry software Read more…

NEC Laboratory Reviews First Year of Cooperative Projects

October 6, 1995

Sankt Augustin, Germany -- NEC C&C (Computers and Communication) Research Laboratory at the GMD Technopark has wrapped up its first year of operation. Read more…

Sun and Sybase Say SQL Server 11 Benchmarks at 4544.60 tpmC

October 6, 1995

Mountain View, Calif. -- Sun Microsystems, Inc. and Sybase, Inc. recently announced the first benchmark results for SQL Server 11. The result represents a n Read more…

New Study Says Parallel Processing Market Will Reach $14B in 1999

October 6, 1995

Mountain View, Calif. -- A study by the Palo Alto Management Group (PAMG) indicates the market for parallel processing systems will increase at more than 4 Read more…

Leading Solution Providers

Contributors

CORNELL I-WAY DEMONSTRATION PITS PARASITE AGAINST VICTIM

October 6, 1995

Ithaca, NY --Visitors to this year's Supercomputing '95 (SC'95) conference will witness a life-and-death struggle between parasite and victim, using virtual Read more…

SGI POWERS VIRTUAL OPERATING ROOM USED IN SURGEON TRAINING

October 6, 1995

Surgery simulations to date have largely been created through the development of dedicated applications requiring considerable programming and computer graphi Read more…

U.S. Will Relax Export Restrictions on Supercomputers

October 6, 1995

New York, NY -- U.S. President Bill Clinton has announced that he will definitely relax restrictions on exports of high-performance computers, giving a boost Read more…

Dutch HPC Center Will Have 20 GFlop, 76-Node SP2 Online by 1996

October 6, 1995

Amsterdam, the Netherlands -- SARA, (Stichting Academisch Rekencentrum Amsterdam), Academic Computing Services of Amsterdam recently announced that it has pur Read more…

Cray Delivers J916 Compact Supercomputer to Solvay Chemical

October 6, 1995

Eagan, Minn. -- Cray Research Inc. has delivered a Cray J916 low-cost compact supercomputer and Cray's UniChem client/server computational chemistry software Read more…

NEC Laboratory Reviews First Year of Cooperative Projects

October 6, 1995

Sankt Augustin, Germany -- NEC C&C (Computers and Communication) Research Laboratory at the GMD Technopark has wrapped up its first year of operation. Read more…

Sun and Sybase Say SQL Server 11 Benchmarks at 4544.60 tpmC

October 6, 1995

Mountain View, Calif. -- Sun Microsystems, Inc. and Sybase, Inc. recently announced the first benchmark results for SQL Server 11. The result represents a n Read more…

New Study Says Parallel Processing Market Will Reach $14B in 1999

October 6, 1995

Mountain View, Calif. -- A study by the Palo Alto Management Group (PAMG) indicates the market for parallel processing systems will increase at more than 4 Read more…

  • arrow
  • Click Here for More Headlines
  • arrow
HPCwire