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

• 发表时间: 2022-01
• 期刊: ArXiv
• 作者: Chenhui Deng;Xiuyu Li;Zhuobo Feng;Zhiru Zhang

An Intermediate Language for General Sparse Format Customization

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