用图对比学习提升模拟电路寄生参数估计的跨设计迁移能力
Transferable Parasitic Estimation via Graph Contrastive Learning and Label Rebalancing in AMS Circuits
- 通过超球面散射实现拓扑不变的节点嵌入,减少对大规模数据依赖
- 引入平衡损失函数,在标签分布不均下仍保持高精度预测
- 在28nm模拟电路中显著优于现有方法,适合芯片设计自动化场景
模拟-混合信号(AMS)电路的图表示学习对寄生参数估计等下游任务至关重要。然而,设计数据稀缺、标签分布不均以及电路实现多样性等问题严重制约了鲁棒且可迁移的电路表征学习。为此,我们提出CircuitGCL框架,结合图对比学习与标签重平衡策略,增强异构电路图间的迁移能力。CircuitGCL采用自监督策略,通过超球面表示散射学习拓扑不变的节点嵌入,无需依赖大规模数据;同时引入平衡均方误差(BMSE)和平衡软最大交叉熵(BSCE)损失,缓解不同电路间标签分布差异,实现稳健且可迁移的寄生参数估计。在TSMC 28nm AMS设计上评估,针对寄生电容估计(边级任务)的R²提升33.64%~44.20%,针对地电容分类(节点级任务)的F1分数提高0.9倍~2.1倍。代码已开源。
原文摘要 · Abstract (English)
Graph representation learning on Analog-Mixed Signal (AMS) circuits is crucial for various downstream tasks, e.g., parasitic estimation. However, the scarcity of design data, the unbalanced distribution of labels, and the inherent diversity of circuit implementations pose significant challenges to learning robust and transferable circuit representations. To address these limitations, we propose CircuitGCL, a novel graph contrastive learning framework that integrates representation scattering and label rebalancing to enhance transferability across heterogeneous circuit graphs. CircuitGCL employs a self-supervised strategy to learn topology-invariant node embeddings through hyperspherical representation scattering, eliminating dependency on large-scale data. Simultaneously, balanced mean squared error (BMSE) and balanced softmax cross-entropy (BSCE) losses are introduced to mitigate label distribution disparities between circuits, enabling robust and transferable parasitic estimation. Evaluated on parasitic capacitance estimation (edge-level task) and ground capacitance classification (node-level task) across TSMC 28nm AMS designs, CircuitGCL outperforms all state-of-the-art (SOTA) methods, with the $R^2$ improvement of $33.64\% \sim 44.20\%$ for edge regression and F1-score gain of $0.9\times \sim 2.1\times$ for node classification. Our code is available at https://github.com/ShenShan123/CircuitGCL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。