Congestion Prediction in IC Design using Graph Neural Network

UC San Diego, Halıcıoğlu Data Science Institute

gnn_graph.gif

Authors: Michelle Tong, Yuyang Pang, Hang Liu, Vivian Chen

Mentors: Lindsey Kostas, Guillaume Shippee, Lea Hagen


Abstract

Congestion prediction is a crucial part of the design process of Integrated Circuits (IC). It allows chip designers to discover areas that have excessive routing demand and optimize the circuit accordingly. This paper presents insights into the use of Graph Neural Networks for early congestion prediction in IC designs. Utilizing the NCSU-DigIC-GraphData dataset, which includes 13 small netlists with distinct congestion characteristics, our methodology includes node feature engineering, multi-graph train-test split, and a comparison between model architectures, which features Graph Convolutional Networks (GCNs) and Graph Attention Networks (GATs). Results show that Graph Neural Networks do not scale well on small netlists that only contain a few thousand nodes. GATs can perform better than GCNs but have significantly longer training time. Moreover, GRC-based prediction models are inferior to node-based prediction models.