Graph Neural Networks (GNNs) have demonstrated transformative potential across domains, driving the development of specialized frameworks like Deep Graph Library (DGL) and PyTorch Geometric (PyG) that employ emerging techniques to overcome computational bottlenecks in large-scale graph learning. However, due to the inherent sparsity of GNN models and the complexity of heterogeneous computing systems, optimizing GNN performance remains a significant challenge. Existing profiling tools, such as Nsight Systems, primarily focus on visualizing resource utilization over time, helping users identify inefficient execution patterns. While this approach provides insights into hardware-level performance, it lacks higher-level, code-centric analysis, making it difficult for developers to pinpoint and resolve performance bottlenecks in GNN training. To address these limitations, we propose GNNProf, an automated performance analysis tool designed to detect and diagnose potential inefficiencies in GNN training. GNNProf collects and restructures CPU function-level performance data into an analyzable format, and applies machine learning and unsupervised learning techniques to identify potential performance anomalies. By automatically recognizing inefficient functions and highlighting performance-critical regions, GNNProf enables developers to gain deeper insights into the execution behavior of GNN training. Additionally, it provides intuitive visualizations that facilitate performance debugging and optimization, ultimately improving training efficiency on heterogeneous systems.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Identifying Potential Anomalous Operations in Graph Neural Network Training

  • Zhibo Xuan,
  • Hailong Yang,
  • Xin You,
  • Zhongzhi Luan,
  • Yi Liu,
  • Depei Qian

摘要

Graph Neural Networks (GNNs) have demonstrated transformative potential across domains, driving the development of specialized frameworks like Deep Graph Library (DGL) and PyTorch Geometric (PyG) that employ emerging techniques to overcome computational bottlenecks in large-scale graph learning. However, due to the inherent sparsity of GNN models and the complexity of heterogeneous computing systems, optimizing GNN performance remains a significant challenge. Existing profiling tools, such as Nsight Systems, primarily focus on visualizing resource utilization over time, helping users identify inefficient execution patterns. While this approach provides insights into hardware-level performance, it lacks higher-level, code-centric analysis, making it difficult for developers to pinpoint and resolve performance bottlenecks in GNN training. To address these limitations, we propose GNNProf, an automated performance analysis tool designed to detect and diagnose potential inefficiencies in GNN training. GNNProf collects and restructures CPU function-level performance data into an analyzable format, and applies machine learning and unsupervised learning techniques to identify potential performance anomalies. By automatically recognizing inefficient functions and highlighting performance-critical regions, GNNProf enables developers to gain deeper insights into the execution behavior of GNN training. Additionally, it provides intuitive visualizations that facilitate performance debugging and optimization, ultimately improving training efficiency on heterogeneous systems.