Collaborative Research: SHF: Medium: Co-optimizing Spectral Algorithms and Systems for High-Performance Graph Learning

合作研究：SHF：中：协同优化高性能图学习的谱算法和系统

• 批准号: 2212371
• 负责人: Zhiru Zhang
• 金额: $ 40万
• 依托单位: Cornell University
• 依托单位国家: 美国
• 项目类别: Continuing Grant
• 财政年份: 2022
• 资助国家: 美国
• 起止时间: 2022-07-01 至 2026-06-30
• 项目状态: 未结题
• 来源: https://www.nsf.gov/awardsearch/showAward?AWD_ID=2212371&HistoricalAwards=false
• 关键词: Collaborative; Research; SHF; Medium; Co

Emerging graph learning techniques have shown promising results for various important applications such as community detection, drug discovery, and electronic design automation (EDA). However, even the state-of-the-art graph learning methods cannot scale to large data sets due to their high algorithm complexity. For example, the latest graph neural network (GNN) algorithms collectively aggregate feature information from the neighborhood of each node, which not only drastically increases the number of computations among nodes but also leads to high memory usage for storing the intermediate results. Hence most graph learning algorithms cannot efficiently handle large-scale problems due to their high computation and storage costs, not to mention the real-world graphs that may involve billions of edges.This project aims at addressing the most pressing challenges in modern graph learning tasks by investigating high-performance spectral graph algorithms and systems based on the latest theoretical breakthroughs. Unlike prior theoretical studies on spectral graph theory that put less focus on practical algorithm implementations and applications, the investigators of this project will develop practically-efficient spectral graph compression algorithms to boost the efficiency and solution quality of existing graph learning methods through algorithm and system co-optimizations by taking advantage of the latest heterogeneous computing devices, such as GPUs, FPGAs, and computational storage devices. The outcome of this project will potentially advance the state of the art in spectral graph theory, machine learning, data analytics, EDA, as well as high-performance computing. This project is also likely to spark new research in other related computer science and engineering fields such as complex system/network modeling, computational biology, precision medicine, and transportation networks. The investigators will partner with the STEM ambassador program at Stevens, and the Diversity Programs in Engineering office at Cornell to recruit highly diversified undergraduate and graduate students to participate in this project, while the latest research results will be integrated into several graduate and upper-division undergraduate courses to prepare a new generation of researchers and practitioners in high-performance machine learning.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.

新兴的图学习技术在社区检测、药物发现和电子设计自动化 (EDA) 等各种重要应用中显示出了有希望的结果。然而，即使是最先进的图学习方法也无法扩展到大型数据集，因为其算法复杂度很高。例如，最新的图神经网络（GNN）算法集体聚合每个节点邻域的特征信息，这不仅大大增加了节点之间的计算量，而且导致存储中间结果的内存使用率很高。因此，大多数图学习算法由于计算和存储成本较高而无法有效处理大规模问题，更不用说可能涉及数十亿条边的现实世界图。该项目旨在解决现代图学习任务中最紧迫的挑战通过研究基于最新理论突破的高性能谱图算法和系统。与之前的谱图理论研究较少关注实际算法实现和应用不同，该项目的研究人员将开发实用高效的谱图压缩算法，通过算法和系统合作来提高现有图学习方法的效率和解质量。 -利用最新的异构计算设备（例如 GPU、FPGA 和计算存储设备）进行优化。该项目的成果将有可能推动谱图理论、机器学习、数据分析、EDA 以及高性能计算领域的最新技术发展。该项目还可能引发其他相关计算机科学和工程领域的新研究，例如复杂系统/网络建模、计算生物学、精准医学和交通网络。研究人员将与史蒂文斯大学的STEM大使项目和康奈尔大学工程办公室多样性项目合作，招募高度多元化的本科生和研究生参与该项目，同时最新的研究成果将被整合到多个研究生和高年级学生中。该奖项反映了 NSF 的法定使命，并通过使用基金会的智力价值和更广泛的影响审查标准进行评估，被认为值得支持。

项目成果

GARNET: Reduced-Rank Topology Learning for Robust and Scalable Graph Neural Networks

GARNET：稳健且可扩展的图神经网络的降阶拓扑学习

• 发表时间: 2022-12
• 期刊: Learning on Graphs Conference
• 作者: Chenhui Deng;Xiuyu Li;Zhuo Feng;Zhiru Zhang
• 通讯作者: Zhiru Zhang

An Intermediate Language for General Sparse Format Customization

通用稀疏格式定制的中间语言

• DOI: 10.1109/lca.2023.3262610
• 发表时间: 2023-01
• 期刊: IEEE Computer Architecture Letters
• 影响因子: 2.3
• 作者: Liu, Jie;Zhao, Zhongyuan;Ding, Zijian;Brock, Benjamin;Rong, Hongbo;Zhang, Zhiru
• 通讯作者: Zhang, Zhiru