公开(公告)号：US20240127031A1

公开(公告)日：2024-04-18

申请号：US18394307

申请日：2023-12-22

Applicant: Intel Corporation

Inventor： Hamza Yous ,  Ian Hunter ,  Alessandro Palla

IPC: G06N3/042 ,  G06N3/08

CPC classification number: G06N3/042 ,  G06N3/08

Abstract: A graph neural network (GNN) model is used in a scheduling process for compiling a deep neural network (DNN). The DNN, and parameter options for scheduling the DNN, are represented as a graph, and the GNN predicts a set of parameters that is expected to have a low cost. Using the GNN-based model, a compiler can produce a schedule for compiling the DNN in a relatively short and predictable amount of time, even for DNNs with many layers and/or many parameter options. For example, the GNN-based model reduces the overhead of exploring every parameter combination and does not exclude combinations from consideration like prior heuristic-based approaches.