SymDetect

This repository contains the implementation of Layout Symmetry Annotation for Analog Circuits with GraphNeural Networks, ASPDAC 2021.

Overview

Graph structure is widely adopted in EDA. The netlists of analog circuits can be represented with hypergraphs and symmetry constraints of analog circuit layouts can be seen as vertice pair on those graphs. So it is intuitive to introduce recent graph neural network methodology to symmetry constraint detection problem.

This repository implements a GraphSage-based model to predict device-level symmetry constraints for analog circuits. The detailed methodology is described in the paper:

Requirements

We develop in Python 3.7.5 with below packages:

numpy 1.17.3
networkx 2.4 
scipy 1.3.2
matplotlib 3.1.1
torch 1.3.1 
torchvision 0.4.2
scikit-learn 0.21.3 

Datasets

We do experiments on extracted leaf-level subcircuits from the two datasets: MAGICAL, ALIGN.

Running the codes

• prepare the data:

  $ bash util/prepare_dataset.sh  
  $ python graphsage/readgraph.py
  

• train and test:

  $ python test.py
  

• run baseline:

  $ python s3det/s3det.py 
  

There are also some options:

• with cuda
• modify hyperparameters

Citation

@inproceedings{gao2021layout,
  title={Layout symmetry annotation for analog circuits with graph neural networks},
  author={Gao, Xiaohan and Deng, Chenhui and Liu, Mingjie and Zhang, Zhiru and Pan, David Z and Lin, Yibo},
  booktitle={Proceedings of the 26th Asia and South Pacific Design Automation Conference},
  pages={152--157},
  year={2021}
}

Acknowledgements

The orginal version of our project is forked from https://github.com/Chenhui1016/directed_graphsage, which is based on https://github.com/williamleif/graphsage-simple/.