arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2212.03475cs.LG

PyGFI:分析并增强图神经网络针对硬件错误的鲁棒性

PyGFI: Analyzing and Enhancing Robustness of Graph Neural Networks Against Hardware Errors

Villanova University(维拉诺瓦大学);Meta Platforms Inc.(元平台公司)

Ruixuan Wang, Fred Lin, Daniel Moore, Sriram Sankar, Xun Jiao

AI总结:

本文首次通过基于 PyTorch 的定制故障注入工具大规模评估 GNN 的硬件错误弹性,揭示模型与数据集间的显著差异,并提出低成本缓解机制以指导加速器设计。

摘要:

图神经网络(GNN)近来已成为学习图结构数据的一种很有前景的学习范式,并在推荐系统、社交网络和电子设计自动化(EDA)等多个领域取得了广泛成功。与其他深度学习(DL)方法一样,GNN 正被部署到复杂的现代硬件系统以及专用加速器中。然而,尽管 GNN 广受欢迎,而且近期已有许多将 GNN 引入硬件的努力,但 GNN 的容错性和弹性总体上仍被忽视。受 DL 方法固有算法弹性的启发,本文首次开展了一项大规模、实证性的 GNN 弹性研究,旨在理解硬件故障与 GNN 准确率之间的关系。通过在 PyTorch 之上开发定制化故障注入工具,我们在多种 GNN 模型和应用数据集上进行了大量故障注入实验。我们观察到,对于不同模型和应用数据集,GNN 模型的错误弹性会相差数个数量级。此外,我们探索了一种面向 GNN 的低成本错误缓解机制,以增强其弹性。这项 GNN 弹性研究旨在为未来的 GNN 加速器设计和架构优化开辟新的方向与机遇。

英文摘要:

Graph neural networks (GNNs) have recently emerged as a promising learning paradigm in learning graph-structured data and have demonstrated wide success across various domains such as recommendation systems, social networks, and electronic design automation (EDA). Like other deep learning (DL) methods, GNNs are being deployed in sophisticated modern hardware systems, as well as dedicated accelerators. However, despite the popularity of GNNs and the recent efforts of bringing GNNs to hardware, the fault tolerance and resilience of GNNs have generally been overlooked. Inspired by the inherent algorithmic resilience of DL methods, this paper conducts, for the first time, a large-scale and empirical study of GNN resilience, aiming to understand the relationship between hardware faults and GNN accuracy. By developing a customized fault injection tool on top of PyTorch, we perform extensive fault injection experiments on various GNN models and application datasets. We observe that the error resilience of GNN models varies by orders of magnitude with respect to different models and application datasets. Further, we explore a low-cost error mitigation mechanism for GNN to enhance its resilience. This GNN resilience study aims to open up new directions and opportunities for future GNN accelerator design and architectural optimization.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑