PDNNet: PDN-Aware GNN-CNN Heterogeneous Network for Dynamic IR Drop Prediction

Yuxiang Zhao,Zhuomin Chai,Xun Jiang,Yibo Lin,Runsheng Wang,Ru Huang
2024-03-27
Abstract:IR drop on the power delivery network (PDN) is closely related to PDN's configuration and cell current consumption. As the integrated circuit (IC) design is growing larger, dynamic IR drop simulation becomes computationally unaffordable and machine learning based IR drop prediction has been explored as a promising solution. Although CNN-based methods have been adapted to IR drop prediction task in several works, the shortcomings of overlooking PDN configuration is non-negligible. In this paper, we consider not only how to properly represent cell-PDN relation, but also how to model IR drop following its physical nature in the feature aggregation procedure. Thus, we propose a novel graph structure, PDNGraph, to unify the representations of the PDN structure and the fine-grained cell-PDN relation. We further propose a dual-branch heterogeneous network, PDNNet, incorporating two parallel GNN-CNN branches to favorably capture the above features during the learning process. Several key designs are presented to make the dynamic IR drop prediction highly effective and interpretable. We are the first work to apply graph structure to deep-learning based dynamic IR drop prediction method. Experiments show that PDNNet outperforms the state-of-the-art CNN-based methods by up to 39.3% reduction in prediction error and achieves 545x speedup compared to the commercial tool, which demonstrates the superiority of our method.
Machine Learning,Artificial Intelligence
What problem does this paper attempt to address?
### Problems Addressed by the Paper This paper aims to address the issue of dynamic IR drop prediction in integrated circuit (IC) design. Specifically: 1. **Problem Background**: - Dynamic IR drop refers to the voltage drop phenomenon in the power delivery network (PDN) caused by current flow. - As the scale of IC design increases, traditional simulation-based methods become computationally infeasible, necessitating a new, efficient method for predicting dynamic IR drop. 2. **Limitations of Existing Methods**: - Current methods based on convolutional neural networks (CNN) have achieved some success but overlook the importance of PDN structure. - Existing methods mainly focus on current load patterns and fail to fully explore the PDN structure and its relationship with the cells. 3. **Contributions of the Paper**: - Proposes a new graph structure—PDNGraph, which comprehensively represents the PDN structure and the relationship between the cells and the PDN. - Designs a dual-branch heterogeneous network, PDNNet, combining the advantages of graph neural networks (GNN) and CNN to capture PDN architecture and dynamic IR drop variations. - Experimental results show that PDNNet reduces prediction error by up to 39.3% compared to state-of-the-art CNN methods and improves speed by 545 times. In summary, this paper improves existing dynamic IR drop prediction methods by introducing a graph structure and a dual-branch network, enhancing prediction accuracy and efficiency.