Versatile Multi-stage Graph Neural Network for Circuit Representation
Shuwen YangZhihao YangDong LiYingxue ZhangZhanguang ZhangGuojie SongJianye Hao
Proposes Circuit Graph and Circuit GNN to fuse topological and geometric chip design data into a unified representation, achieving state-of-the-art accuracy across multiple EDA stages while delivering a tenfold speedup in congestion prediction.
Modern integrated circuit design faces increasing complexity, making traditional design automation workflows time-consuming and costly. To accelerate development and prevent costly manufacturing defects, chip designers rely on predictive machine learning models to detect routing congestion and estimate wiring performance early in the design cycle. However, current artificial intelligence approaches are fragmented. Purely topological methods analyze structural connectivity in netlists but fail to utilize spatial data available after placement, while computer vision-based geometric methods rely on spatial grid images and cannot operate during earlier logic synthesis stages. This lack of a unified representation creates performance bottlenecks and restricts machine learning models from generalizing across multiple design stages and tasks.
To address this challenge, the article introduces Circuit Graph, a unified heterogeneous graph structure, and Circuit GNN, an efficient neural network framework designed to process it. The primary objective is to demonstrate a versatile circuit representation method that seamlessly integrates logical circuit topology and physical layout geometry, providing superior prediction accuracy and computational efficiency across both logic synthesis and placement stages.
The evaluated approach models circuits as bipartite graphs containing cell and net nodes connected by topological edges, while adding geometric edges between spatially adjacent cells when physical placement coordinates are available. The neural network applies distinct message-passing mechanisms over both edge types and fuses the resulting representations using pooling operations. To maintain linear computational complexity relative to circuit size, the authors use a shifted-window technique to constrain geometric edge connections. The authors evaluated the framework on standard benchmark datasets (ISPD2011 and DAC2012) using established placement tools and routers across multiple predictive tasks, comparing it against conventional graph neural networks, computer vision baselines, and industry-specific predictive models.
The findings show that the proposed framework consistently outperforms existing state-of-the-art methods while dramatically reducing computational overhead. In the logic synthesis stage, the model improves average grid-level congestion prediction accuracy by 16.7% over existing topological methods. In the placement stage, it achieves a 5.6% gain in congestion prediction accuracy while delivering a tenfold speedup compared to leading models. For net wirelength estimation, the model reduces prediction error by 16.9%. Furthermore, transfer learning experiments confirm that features learned during congestion prediction transfer effectively to wirelength estimation with minimal fine-tuning, outperforming competing architectures in cross-task adaptability.
These results demonstrate that unifying topological and spatial circuit data into a single neural framework significantly enhances predictive power without introducing exponential runtime penalties. Operationally, earlier and more accurate congestion and wirelength forecasting enables an industry shift toward earlier design optimization, commonly referred to as shifting left. This reduces iterative redesign cycles, cuts production timelines, lowers design costs, and prevents the fabrication of flawed semiconductor chips.
Based on these findings, engineering teams developing electronic design automation pipelines should consider adopting unified heterogeneous graph structures to standardize circuit feature representations across pre-placement and post-placement toolchains. Organizations should pursue pilot integrations into global placement and logic synthesis workflows to validate speedups in production environments. Further engineering work is needed to bridge the deployment gap between research algorithms and commercial electronic design automation tool suites. Additionally, researchers should explore extending the representation to earlier design representations, such as high-level data-flow graphs and logic-level graphs, where node and edge semantics diverge from standard netlists.
- Paper: Semi-Supervised Classification with Graph Convolutional Networks, Thomas N. Kipf et al. (2017). Establishes foundational spatial graph convolutional networks that the source adapts to process circuit netlist topologies.
- Paper: Graph Transformer Networks, Seongjun Yun et al. (2019). Introduces graph neural network architectures for heterogeneous graphs containing multiple node and relation types, providing the structural foundation for bipartite circuit representations.
- Paper: Dynamic Edge-Conditioned Filters in Convolutional Neural Networks on Graphs, Martin Simonovsky et al. (2017). Introduces edge-conditioned convolutions that inform how geometric and spatial attributes are incorporated into graph message-passing operations.
- Paper: The Graph Neural Network Model, Franco Scarselli et al. (2009). Provides the seminal theoretical formulations of graph neural networks and node-level message passing underpinning modern circuit learning architectures.
- Paper: How Powerful are Graph Neural Networks?, Keyulu Xu et al. (2019). Formalizes the expressive power and aggregation mechanics of graph neural networks, motivating the design of specialized bipartite and multi-stage circuit encoders.
- Paper: CktGNN: Circuit Graph Neural Network for Electronic Design Automation, Zehao Dong et al. (2023). Extends graph neural network representations of electronic circuits to automated topology generation and sizing in analog electronic design automation.
- Paper: BetterV: Controlled Verilog Generation with Discriminative Guidance, Zehua Pei et al. (2024). Applies machine learning to earlier electronic design automation stages by generating Verilog code optimized for downstream logic synthesis workflows.
