2025-11-13T07:49:11.035604

Transferable Parasitic Estimation via Graph Contrastive Learning and Label Rebalancing in AMS Circuits

Shen, Hua, Zou et al.
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.
academic

Stima Parassitaria Trasferibile tramite Apprendimento Contrastivo su Grafi e Ribilanciamento delle Etichette nei Circuiti AMS

Informazioni Fondamentali

  • ID Articolo: 2507.06535
  • Titolo: Transferable Parasitic Estimation via Graph Contrastive Learning and Label Rebalancing in AMS Circuits
  • Autori: Shan Shen, Shenglu Hua, Jiajun Zou, Jiawei Liu, Jianwang Zhai, Chuan Shi, Wenjian Yu
  • Classificazione: cs.LG cs.SY eess.SY
  • Data di Pubblicazione: 10 ottobre 2025 (sottomissione arXiv)
  • Link Articolo: https://arxiv.org/abs/2507.06535

Riassunto

Questo articolo propone il framework CircuitGCL per affrontare il problema della stima dei parametri parassitari nei circuiti analogici-misti (AMS). Il framework risolve le sfide della scarsità di dati di progettazione circuitale, dello squilibrio nella distribuzione delle etichette e della diversità nell'implementazione circuitale attraverso l'apprendimento contrastivo su grafi e tecniche di ribilanciamento delle etichette. CircuitGCL adotta una strategia autosupervisionata, apprendendo incorporamenti di nodi topologicamente invarianti attraverso la dispersione di rappresentazioni su sfere unitarie, introducendo al contempo funzioni di perdita con errore quadratico medio bilanciato (BMSE) e entropia incrociata softmax bilanciata (BSCE) per mitigare le differenze nella distribuzione delle etichette tra circuiti. Gli esperimenti su progettazioni AMS TSMC 28nm dimostrano che il metodo migliora R² del 33,64%~44,20% nei compiti di regressione dei bordi e il punteggio F1 di 0,9×~2,1× nei compiti di classificazione dei nodi.

Contesto di Ricerca e Motivazione

Definizione del Problema

I moderni circuiti AMS integrano moduli analogici (come amplificatori, oscillatori) e sottosistemi digitali (come controllori, array SRAM), richiedendo numerose iterazioni manuali durante il processo di progettazione. Con la continua riduzione dei nodi tecnologici, gli effetti parassitari diventano sempre più importanti, in particolare le capacità parassite che influenzano significativamente le prestazioni circuitali, causando aumento della latenza di propagazione, incremento dei consumi e degradazione dell'integrità del segnale.

Importanza della Ricerca

Il flusso di progettazione tradizionale dipende dalla simulazione post-layout per verificare gli effetti parassitari, un metodo dispendioso in termini di tempo e costi. La previsione dei parametri parassitari nella fase pre-layout può ridurre significativamente il numero di iterazioni di progettazione, migliorando l'efficienza progettuale. Le reti neurali su grafi (GNN) forniscono una soluzione efficace, consentendo di modellare i circuiti come strutture grafiche per la previsione dei parametri parassitari.

Limitazioni dei Metodi Esistenti

  1. Scarsità di Dati: I dati circuitali AMS di alta qualità (inclusi netlist SPICE, parametri parassitari di layout, ecc.) sono generalmente proprietari e costosi da generare
  2. Diversità Circuitale: I circuiti AMS si estendono su domini analogico, digitale e misto-segnale, con principi di progettazione e requisiti di prestazione differenti
  3. Squilibrio delle Etichette: La distribuzione della capacità parassita presenta caratteristiche di coda lunga, con grave insufficienza di campioni con valori di capacità elevati
  4. Scarsa Trasferibilità: I metodi esistenti addestrati su tipi di circuiti specifici hanno difficoltà a generalizzarsi ad altre topologie circuitali

Contributi Principali

  1. Proposta del Framework CircuitGCL: Adatta il meccanismo di dispersione di rappresentazioni (RSM) all'apprendimento contrastivo su grafi, generando rappresentazioni trasferibili direttamente applicabili a progettazioni AMS non viste, senza necessità di fine-tuning specifico per il compito
  2. Tecniche di Ribilanciamento delle Etichette: Risolve i problemi di squilibrio nei dataset circuitali attraverso funzioni di perdita MSE bilanciato e BSCE bilanciato, migliorando la capacità di trasferimento cross-dominio del modello
  3. Framework Teorico Unificato: Fornisce una base teorica unificata per regressione e classificazione non bilanciate basata su principi di allineamento distributivo
  4. Applicabilità Diffusa: Il metodo può essere direttamente esteso a compiti di previsione di resistenza/induttanza, analisi di diafonia, stima della caduta IR e trasferimento cross-processo

Dettagli del Metodo

Definizione del Compito

Input: Netlist dello schema circuitale AMS, modellato come grafo eterogeneo G = (V, E), dove i nodi V rappresentano reti, dispositivi transistor e pin, e gli archi E codificano le relazioni di connessione Output:

  • Compito di regressione dei bordi: previsione dei valori di capacità di accoppiamento
  • Compito di classificazione dei nodi: classificazione della capacità verso terra in intervalli discreti (piccola/media/grande)

Architettura del Modello

1. Modulo di Trasformazione Grafica

Converte il grafo AMS eterogeneo in un grafo omogeneo, distinguendo i nodi di rete, dispositivo e pin attraverso attributi di tipo di nodo X ∈ {0,1,2}^{N×1}. La matrice di caratteristiche aumentata X_C ∈ R^{N×d_C} cattura parametri di progettazione dettagliati e informazioni statistiche di connessione.

2. Meccanismo di Dispersione di Rappresentazioni (RSM)

Definizione: Nello spazio di incorporamento D-dimensionale R^D, RSM applica due vincoli:

  • Vincolo di allontanamento dal centro: massima separazione degli incorporamenti dei nodi dal centro di dispersione c
  • Vincolo di uniformità: gli incorporamenti dei nodi sono uniformemente distribuiti sul sottospazio S_k

Implementazione:

h̃_i = h_i / max(||h_i||_2, ε)  # Normalizzazione L2
L_scattering = -1/N ∑||h̃_i - c||²_2  # Perdita di dispersione
c = 1/N ∑h̃_i  # Centro di dispersione

3. Codificatore Online

Il codificatore target genera rappresentazioni disperse H_target = f_φ(A,X), il codificatore online produce rappresentazioni intermedie H_online, ottenendo rappresentazioni predette z_online attraverso il predittore q_θ(·). La perdita di allineamento è:

L_alignment = -1/N ∑(z_i^T h_i)/(||z_i||_2 ||h_i||_2)

I parametri del codificatore target vengono aggiornati tramite media mobile esponenziale: φ ← τφ + (1-τ)θ

Punti di Innovazione Tecnica

1. Vantaggi rispetto a DSPD

CircuitGPS utilizza la distanza del percorso più breve a doppio ancoraggio (DSPD) come codifica posizionale, ma i costi computazionali e di memoria crescono quadraticamente con la dimensione del grafo. Il preaddestramento GCL di CircuitGCL presenta elevata parallelizzabilità e buona scalabilità del modello, superando significativamente DSPD su circuiti su larga scala.

2. Base Teorica del Ribilanciamento delle Etichette

Basato sul teorema di Bayes, la relazione tra la distribuzione di addestramento p_train(y|x) e la distribuzione bilanciata p_bal(y|x) è:

p_train(y|x)/p_bal(y|x) ∝ p_train(y)/p_bal(y)

Configurazione Sperimentale

Dataset

Sei progettazioni circuitali AMS implementate utilizzando la tecnologia CMOS TSMC 28nm:

  • Set di Addestramento: SSRAM (87K nodi, 134K archi)
  • Set di Test: DIGITAL CLK GEN, TIMING CTRL, ARRAY 128 32, ULTRA8T, SANDWICH-RAM
  • La progettazione più grande contiene 4,3M nodi e 13,3M archi

Metriche di Valutazione

  • Compiti di Regressione: MAE, MSE, R²
  • Compiti di Classificazione: Accuracy, Precision, Recall, F1-score

Metodi di Confronto

  1. ParaGraph: Modello ensemble basato su MPNN
  2. DLPL-Cap: Regressore GNN multi-esperto
  3. CircuitGPS: Metodo di apprendimento con pochi campioni utilizzando codifica posizionale

Dettagli di Implementazione

  • Codificatore: 4 strati ClusterGCN, 256 dimensioni nascoste, attivazione Tanh, dropout 0,3
  • GNN downstream: 5 strati GraphSAGE, 144 dimensioni nascoste, attivazione PReLU
  • σ_noise = 0,001, τ = 0,99

Risultati Sperimentali

Risultati Principali

Compito di Regressione dei Bordi (Stima della Capacità di Accoppiamento)

Set di TestMetodo MiglioreMiglioramento R²
TIMING CTRLCircuitGCL(GAI)41,08%
ARRAY 128 32CircuitGCL(GAI)44,20%
ULTRA8TCircuitGCL(GAI)37,44%
SANDWICH-RAMCircuitGCL(GAI)33,64%

Compito di Classificazione dei Nodi (Classificazione della Capacità verso Terra)

Set di TestMetodo MiglioreMiglioramento F1
DIGITAL CLK GENCircuitGCL(BSCE)0,9×
ARRAY 128 32CircuitGCL(BSCE)2,1×
ULTRA8TCircuitGCL(BSCE)1,2×
SANDWICH-RAMCircuitGCL(BSCE)1,5×

Esperimenti di Ablazione

Analisi dell'Effetto RSM

RSM porta miglioramenti delle prestazioni su tutti i dataset:

  • Miglioramento massimo di R² del 26,9% (ARRAY 128 32)
  • Miglioramento massimo di F1 del 20,0% (ULTRA8T)
  • Miglioramento minimo di 4,1% (F1) e 6,56% (R²)

Effetto del Ribilanciamento delle Etichette

L'applicazione di BSCE a tutti i metodi baseline porta miglioramenti significativi dell'accuratezza, con effetti ancora più evidenti su progettazioni su larga scala. L'MSE bilanciato migliora significativamente le prestazioni del modello nelle regioni di dati scarsi.

Lavori Correlati

Previsione dei Parametri Parassitari

  • ParaGraph: Converte schemi circuitali in grafi, utilizza MPNN per prevedere capacità di rete e parametri di layout
  • DLPL-Cap: Combina router GNN e cinque regressori esperti per gestire la distribuzione non bilanciata dei dati nei circuiti SRAM
  • CircuitGPS: Utilizza apprendimento con pochi campioni e codifica posizionale a basso costo per la previsione dei parametri parassitari

Apprendimento Contrastivo su Grafi

I principali framework GCL eseguono implicitamente la dispersione di rappresentazioni, che gioca un ruolo cruciale nel loro successo. Questo articolo adotta SGRL come base GCL, incorporando rappresentazioni di nodi all'interno di sfere unitarie.

Apprendimento Non Bilanciato

  • Classificazione: Ricampionamento dei dati, ripeseramento delle perdite, aggiustamento logit e altri metodi
  • Regressione: Relativamente sottesplorata, i metodi esistenti si concentrano principalmente su adattamenti SMOTE e ripeseramento delle perdite

Conclusioni e Discussione

Conclusioni Principali

  1. CircuitGCL risolve con successo i problemi di scarsità di dati e squilibrio delle etichette nella progettazione di circuiti AMS attraverso l'apprendimento di rappresentazioni autosupervisionato e funzioni di perdita consapevoli della distribuzione
  2. Gli incorporamenti topologicamente invarianti generati da RSM possiedono eccellenti capacità di trasferimento cross-dominio
  3. Le funzioni di perdita bilanciate unificano teoricamente i problemi di regressione e classificazione non bilanciate
  4. Raggiungono prestazioni SOTA su molteplici progettazioni TSMC 28nm

Limitazioni

  1. Validato solo su processi TSMC 28nm, la capacità di trasferimento cross-nodo tecnologico rimane da verificare
  2. Attualmente focalizzato sulla previsione della capacità, l'applicabilità alla previsione di resistenza/induttanza richiede ulteriore verifica
  3. L'efficienza computazionale su circuiti su larga scala ha ancora spazio per ottimizzazione

Direzioni Future

  1. Estensione a tipi di circuiti AMS più ampi
  2. Adattamento alla stima della resistenza parassitaria
  3. Integrazione in strumenti di instradamento consapevoli di RC
  4. Ricerca sul trasferimento di apprendimento cross-nodo tecnologico

Valutazione Approfondita

Punti di Forza

  1. Importanza del Problema: Affronta i punti critici effettivi nel campo dell'EDA, con significativo valore ingegneristico
  2. Innovazione del Metodo: Prima applicazione dell'apprendimento contrastivo su grafi alla previsione dei parametri parassitari circuitali, adattamento RSM ingegnoso
  3. Contributo Teorico: Fornisce un framework teorico unificato per regressione e classificazione non bilanciate
  4. Esperimenti Completi: Validazione su molteplici progettazioni circuitali reali, risultati convincenti
  5. Riproducibilità: Fornisce codice open-source, facilitando riproduzione e applicazione

Insufficienze

  1. Limitazioni del Dataset: Utilizza solo dati di un singolo nodo tecnologico, la capacità di generalizzazione rimane da verificare
  2. Sovraccarico Computazionale: Sebbene più efficiente di DSPD, il preaddestramento GCL richiede comunque risorse computazionali aggiuntive
  3. Ingegneria delle Caratteristiche: L'estrazione e la rappresentazione delle caratteristiche circuitali dipendono ancora dalla progettazione manuale
  4. Analisi Teorica: Manca un'analisi teorica approfondita del perché RSM sia particolarmente adatto ai grafi circuitali

Impatto

  1. Valore Accademico: Apre l'applicazione dell'apprendimento contrastivo su grafi nel campo dell'EDA, potrebbe ispirare ricerche correlate
  2. Valore Ingegneristico: Direttamente applicabile agli strumenti EDA commerciali, migliorando l'efficienza progettuale
  3. Generalità del Metodo: Il framework è estendibile ad altri tipi di previsione di parametri parassitari e compiti di analisi circuitale

Scenari Applicabili

  1. Fase Pre-Layout: Valutazione rapida degli effetti parassitari, riduzione delle iterazioni di progettazione
  2. Esplorazione dello Spazio di Progettazione: Screening rapido di candidati in spazi di progettazione su larga scala
  3. Trasferimento Cross-Design: Applicazione di modelli addestrati su un circuito a nuove progettazioni circuitali
  4. Integrazione in Strumenti EDA: Come modulo algoritmo centrale negli strumenti EDA commerciali

Valutazione Complessiva: Questo è un articolo di ricerca di alta qualità interdisciplinare che applica con successo tecnologie all'avanguardia dell'apprendimento automatico a problemi pratici nel campo dell'EDA. Il metodo è innovativo, gli esperimenti completi, i risultati significativi, con importante valore teorico e pratico.