OAC Core: Scalable Graph ML on Distributed Heterogeneous Systems

OAC 核心：分布式异构系统上的可扩展图 ML

基本信息

批准号：
2209563
负责人：
Viktor Prasanna
金额：
$ 59.97万
依托单位：
University of Southern California
依托单位国家：
美国
项目类别：
Standard Grant
财政年份：
2022
资助国家：
美国
起止时间：
2022-08-15 至 2025-07-31
项目状态：
未结题

来源：
https://www.nsf.gov/awardsearch/showAward?AWD_ID=2209563&HistoricalAwards=false
关键词：
OAC Core Scalable Graph ML

项目摘要

Methods that employ graph machine learning (Graph ML), which is a sub-discipline within machine learning that deals with graph data, are becoming important in many key science and engineering domains. For example, the predictive power of graph embedding has been effectively utilized in domains such as social media, biology, pharmacology, and knowledge understanding. However, such methods typically come with an expensive computational footprint, as the computations often need to be performed in real-time on very large and highly heterogeneous static and dynamic graphs with billions of vertices and edges of different types. This project aims at conducting multi-pronged research to enable creation of a cyberinfrastructure (CI) toolkit to run such complex Graph ML applications on emerging heterogeneous distributed systems. The objective of this project is to develop high-performance Graph ML algorithms for key graph workflows spanning multiple scientific and engineering domains targeting distributed heterogeneous systems composed of multi-core processors, Graphics Processing Units (GPUs), Field Programmable Gate Arrays (FPGAs), accelerators and high bandwidth memory interconnected with cache coherent interfaces. The project develops a scalable, deployable, and robust CI toolkit consisting of: (1) novel graph sampling algorithms and efficient Graph ML models for low complexity training and inference computation on static and dynamic graphs; (2) a heterogeneity-aware hardware mapping methodology to accelerate these algorithms and models; and (3) software and hardware libraries for automatic design generation. The project develops proof of concept software for the ML and Data Science communities to facilitate end-to-end deployment of various large-scale applications. Given that graph neural networks are increasingly becoming an important tool for analyzing data in many diverse domains, the outcomes of this project will have a strong impact across a broad range of disciplines, including domains that rely on edge computing, such as autonomous vehicles and smart cities. The project has a robust plan to integrate research into education programs and focuses on activities that promote involvement of students from minority and economically disadvantaged backgrounds into the research.This award reflects NSF's statutory mission and has been deemed worthy of support through evaluation using the Foundation's intellectual merit and broader impacts review criteria.

采用图形机器学习（图ML）的方法是机器学习中涉及图数据的一个子学科，在许多关键的科学和工程领域中变得越来越重要。例如，图形嵌入的预测能力已被有效地用于社交媒体，生物学，药理学和知识理解等领域。但是，这种方法通常具有昂贵的计算足迹，因为这些计算通常需要在非常大且高度异构的静态和动态图上实时执行，并具有数十亿个不同类型的顶点和边缘。该项目旨在进行多管齐下的研究，以实现网络基础结构（CI）工具包，以在新兴的异质分布系统上运行如此复杂的ML应用程序。该项目的目的是为关键图形工作流程开发高性能图ML算法，这些算法涵盖了多个科学和工程领域，以分布式异质系统组成，由多核处理器，图形处理单元（GPU），现场程序可编程栅极阵列（FPGAS），ACCELERATORS，ACCELERATORS，ACCELERERATOR，ACCELERERATER和高频段互联网组成。该项目开发了一个可扩展，可部署和鲁棒的CI工具包，该工具包由以下方式组成：（1）用于低复杂性训练和静态和动态图上的低复杂性训练和推理计算的新型图形采样算法和有效的图形ML模型；（2）一种异质性意识性的硬件映射方法，以加速这些算法和模型；（3）自动设计生成的软件和硬件库。该项目为ML和数据科学社区开发了概念软件证明，以促进各种大规模应用程序的端到端部署。鉴于图形神经网络越来越成为分析许多不同领域数据的重要工具，因此该项目的结果将在广泛的学科中产生很大的影响，包括依靠边缘计算的领域，例如自动驾驶汽车和智能城市。该项目制定了一项强大的计划，可以将研究整合到教育计划中，并着重于促进少数群体和经济不利的背景的学生参与研究的活动。该奖项反映了NSF的法定任务，并被认为是通过基金会的知识分子优点和更广泛的影响审查标准通过评估来通过评估来支持的。

项目成果

期刊论文数量（11）

专著数量（0）

科研奖励数量（0）

会议论文数量（0）

专利数量（0）

Performance Modeling Sparse MTTKRP Using Optical Static Random Access Memory on FPGA

使用 FPGA 上的光学静态随机存取存储器对稀疏 MTTKRP 进行性能建模

DOI：
10.1109/hpec55821.2022.9926407
发表时间：
2022
期刊：
IEEE High Performance Extreme Computing Conference
影响因子：
0
作者：
Wijeratne, Sasindu;Jaiswal, Akhilesh;Jacob, Ajey P.;Zhang, Bingyi;Prasanna, Viktor
通讯作者：
Prasanna, Viktor

Accelerating Sparse MTTKRP for Tensor Decomposition on FPGA

加速 FPGA 上张量分解的稀疏 MTTKRP

DOI：
10.1145/3543622.3573179
发表时间：
2023
期刊：
ACM/SIGDA International Symposium on Field Programmable Gate Arrays
影响因子：
0
作者：
Wijeratne, Sasindu;Wang, Ta-Yang;Kannan, Rajgopal;Prasanna, Viktor
通讯作者：
Prasanna, Viktor

HTNet: Dynamic WLAN Performance Prediction using Heterogenous Temporal GNN

DOI：
10.1109/infocom53939.2023.10229047
发表时间：
2023-04
期刊：
IEEE INFOCOM 2023 - IEEE Conference on Computer Communications
影响因子：
0
作者：
Hongkuan Zhou;R. Kannan;A. Swami;V. Prasanna
通讯作者：
Hongkuan Zhou;R. Kannan;A. Swami;V. Prasanna

Modeling the Energy Efficiency of GEMM using Optical Random Access Memory

DOI：
10.1109/hpec55821.2022.9926291
发表时间：
2022-09
期刊：
2022 IEEE High Performance Extreme Computing Conference (HPEC)
影响因子：
0
作者：
Bingyi Zhang;Akhilesh R. Jaiswal;Clynn Mathew;R. T. Lakkireddy;Ajey P. Jacob;Sasindu Wijeratne;V. Prasanna
通讯作者：
Bingyi Zhang;Akhilesh R. Jaiswal;Clynn Mathew;R. T. Lakkireddy;Ajey P. Jacob;Sasindu Wijeratne;V. Prasanna

Graph Neural Network for Accurate and Low-complexity SAR ATR