Accurately predicting the power consumption and latency of software applications deployed on multicore platforms is a critical bottleneck for early-stage performance optimization, as it often relies on slow and costly simulations. To address this challenge, we introduce a graph-based methodology that models the mappings of software applications onto multicore architectures as heterogeneous graphs. We then investigate Graph Neural Networks (GNNs) to predict power consumption and latency for these mappings. Four classes of state-of-the-art GNNs were trained under various conditions on 11 datasets, each containing multiple mappings of a given Neural-Network application. The two best-performing GNN classes achieve mean absolute percentage errors of around 2% for power-consumption prediction and 15% for latency. Inference requires only a few tens of milliseconds per prediction. This work demonstrates that GNNs offer a fast and promising approach to performance prediction, opening the door to AI-assisted optimization of software mapping on multicore platforms. The relevance of our work extends beyond the use case: we introduce an operational framework for GNNs to support heterogeneous graph processing and multi-output regression at the graph level.

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

Graph Neural Networks for Graph-Level Regression on Heterogeneous Network Data: Use Case in Early-Stage Optimization of Software Mapping on Multicore Platforms

  • Oscar Roussel,
  • Zainab Ghrayeb,
  • Sébastien Le Nours,
  • Christine Sinoquet

摘要

Accurately predicting the power consumption and latency of software applications deployed on multicore platforms is a critical bottleneck for early-stage performance optimization, as it often relies on slow and costly simulations. To address this challenge, we introduce a graph-based methodology that models the mappings of software applications onto multicore architectures as heterogeneous graphs. We then investigate Graph Neural Networks (GNNs) to predict power consumption and latency for these mappings. Four classes of state-of-the-art GNNs were trained under various conditions on 11 datasets, each containing multiple mappings of a given Neural-Network application. The two best-performing GNN classes achieve mean absolute percentage errors of around 2% for power-consumption prediction and 15% for latency. Inference requires only a few tens of milliseconds per prediction. This work demonstrates that GNNs offer a fast and promising approach to performance prediction, opening the door to AI-assisted optimization of software mapping on multicore platforms. The relevance of our work extends beyond the use case: we introduce an operational framework for GNNs to support heterogeneous graph processing and multi-output regression at the graph level.