Saved in:
Bibliographic Details
Main Authors: Shen, Shan, Zhang, Yibin, Rodriguez, Hector Rodriguez, Yu, Wenjian
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2507.06538
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918087199031296
author Shen, Shan
Zhang, Yibin
Rodriguez, Hector Rodriguez
Yu, Wenjian
author_facet Shen, Shan
Zhang, Yibin
Rodriguez, Hector Rodriguez
Yu, Wenjian
contents Graph representation learning is a powerful method to extract features from graph-structured data, such as analog/mixed-signal (AMS) circuits. However, training deep learning models for AMS designs is severely limited by the scarcity of integrated circuit design data. In this work, we present CircuitGPS, a few-shot learning method for parasitic effect prediction in AMS circuits. The circuit netlist is represented as a heterogeneous graph, with the coupling capacitance modeled as a link. CircuitGPS is pre-trained on link prediction and fine-tuned on edge regression. The proposed method starts with a small-hop sampling technique that converts a link or a node into a subgraph. Then, the subgraph embeddings are learned with a hybrid graph Transformer. Additionally, CircuitGPS integrates a low-cost positional encoding that summarizes the positional and structural information of the sampled subgraph. CircuitGPS improves the accuracy of coupling existence by at least 20\% and reduces the MAE of capacitance estimation by at least 0.067 compared to existing methods. Our method demonstrates strong inherent scalability, enabling direct application to diverse AMS circuit designs through zero-shot learning. Furthermore, the ablation studies provide valuable insights into graph models for representation learning.
format Preprint
id arxiv_https___arxiv_org_abs_2507_06538
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Few-shot Learning on AMS Circuits and Its Application to Parasitic Capacitance Prediction
Shen, Shan
Zhang, Yibin
Rodriguez, Hector Rodriguez
Yu, Wenjian
Machine Learning
Systems and Control
Graph representation learning is a powerful method to extract features from graph-structured data, such as analog/mixed-signal (AMS) circuits. However, training deep learning models for AMS designs is severely limited by the scarcity of integrated circuit design data. In this work, we present CircuitGPS, a few-shot learning method for parasitic effect prediction in AMS circuits. The circuit netlist is represented as a heterogeneous graph, with the coupling capacitance modeled as a link. CircuitGPS is pre-trained on link prediction and fine-tuned on edge regression. The proposed method starts with a small-hop sampling technique that converts a link or a node into a subgraph. Then, the subgraph embeddings are learned with a hybrid graph Transformer. Additionally, CircuitGPS integrates a low-cost positional encoding that summarizes the positional and structural information of the sampled subgraph. CircuitGPS improves the accuracy of coupling existence by at least 20\% and reduces the MAE of capacitance estimation by at least 0.067 compared to existing methods. Our method demonstrates strong inherent scalability, enabling direct application to diverse AMS circuit designs through zero-shot learning. Furthermore, the ablation studies provide valuable insights into graph models for representation learning.
title Few-shot Learning on AMS Circuits and Its Application to Parasitic Capacitance Prediction
topic Machine Learning
Systems and Control
url https://arxiv.org/abs/2507.06538