arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.12840cs.LG
收藏

HGNAS:面向边缘设备的硬件感知图神经架构搜索

HGNAS: Hardware-Aware Graph Neural Architecture Search for Edge Devices

School of Computer Science and Engineering, Beihang University(北京航空航天大学计算机科学与工程学院);School of Integrated Circuits and Engineering, Beihang University(北京航空航天大学集成电路与工程学院)

Ao Zhou, Jianlei Yang, Yingjie Qi, Tong Qiao, Yumeng Shi, Cenlin Duan, Weisheng Zhao, Chunming Hu

AI总结:

提出面向边缘设备的硬件感知图神经架构搜索框架HGNAS,通过集成硬件性能预测器、峰值内存估计方法和多阶段分层搜索策略,在细粒度设计空间中高效探索架构,实现高达10.6倍加速和82.5%内存减少。

摘要:

图神经网络(GNNs)因其最先进(SOTA)的性能,在点云处理等基于图的学习任务中日益普及。然而,研究界主要集中于提升模型表达能力,缺乏对如何为具有实时要求和资源受限的边缘场景设计高效GNN模型的考虑。检查现有GNN模型发现,其在不同平台上执行情况各异且频繁出现内存溢出(OOM)问题,凸显了硬件感知GNN设计的必要性。为应对此挑战,本工作提出了一种专为资源受限边缘设备定制的硬件感知图神经架构搜索框架,即HGNAS。为实现硬件感知,HGNAS集成了高效的GNN硬件性能预测器,能在毫秒级评估GNN的延迟和峰值内存使用量。同时,我们研究了推理期间的GNN内存使用情况,并提供峰值内存估计方法,与预测器结果结合时增强了架构评估的鲁棒性。此外,HGNAS通过解耦GNN范式构建了细粒度设计空间,以实现对极致性能架构的探索。另外,利用多阶段分层搜索策略促进对庞大候选者的导航,可将单次搜索时间缩减至几个GPU小时。据我们所知,HGNAS是首个面向边缘设备的自动化GNN设计框架,也是首个实现GNN在不同平台上硬件感知的工作。跨各种应用和边缘设备的大量实验证明了HGNAS的优越性。与DGCNN在ModelNet40上相比,它可实现高达10.6倍的加速和82.5%的峰值内存减少,且精度损失可忽略不计。

英文摘要:

Graph Neural Networks (GNNs) are becoming increasingly popular for graph-based learning tasks such as point cloud processing due to their state-of-the-art (SOTA) performance. Nevertheless, the research community has primarily focused on improving model expressiveness, lacking consideration of how to design efficient GNN models for edge scenarios with real-time requirements and limited resources. Examining existing GNN models reveals varied execution across platforms and frequent Out-Of-Memory (OOM) problems, highlighting the need for hardware-aware GNN design. To address this challenge, this work proposes a novel hardware-aware graph neural architecture search framework tailored for resource constraint edge devices, namely HGNAS. To achieve hardware awareness, HGNAS integrates an efficient GNN hardware performance predictor that evaluates the latency and peak memory usage of GNNs in milliseconds. Meanwhile, we study GNN memory usage during inference and offer a peak memory estimation method, enhancing the robustness of architecture evaluations when combined with predictor outcomes. Furthermore, HGNAS constructs a fine-grained design space to enable the exploration of extreme performance architectures by decoupling the GNN paradigm. In addition, the multi-stage hierarchical search strategy is leveraged to facilitate the navigation of huge candidates, which can reduce the single search time to a few GPU hours. To the best of our knowledge, HGNAS is the first automated GNN design framework for edge devices, and also the first work to achieve hardware awareness of GNNs across different platforms. Extensive experiments across various applications and edge devices have proven the superiority of HGNAS. It can achieve up to a 10.6x speedup and an 82.5% peak memory reduction with negligible accuracy loss compared to DGCNN on ModelNet40.

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

↑