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
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.
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.
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.
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
Diversità Circuitale: I circuiti AMS si estendono su domini analogico, digitale e misto-segnale, con principi di progettazione e requisiti di prestazione differenti
Squilibrio delle Etichette: La distribuzione della capacità parassita presenta caratteristiche di coda lunga, con grave insufficienza di campioni con valori di capacità elevati
Scarsa Trasferibilità: I metodi esistenti addestrati su tipi di circuiti specifici hanno difficoltà a generalizzarsi ad altre topologie circuitali
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
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
Framework Teorico Unificato: Fornisce una base teorica unificata per regressione e classificazione non bilanciate basata su principi di allineamento distributivo
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
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)
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.
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 è:
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.
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.
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.
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
Gli incorporamenti topologicamente invarianti generati da RSM possiedono eccellenti capacità di trasferimento cross-dominio
Le funzioni di perdita bilanciate unificano teoricamente i problemi di regressione e classificazione non bilanciate
Raggiungono prestazioni SOTA su molteplici progettazioni TSMC 28nm
Importanza del Problema: Affronta i punti critici effettivi nel campo dell'EDA, con significativo valore ingegneristico
Innovazione del Metodo: Prima applicazione dell'apprendimento contrastivo su grafi alla previsione dei parametri parassitari circuitali, adattamento RSM ingegnoso
Contributo Teorico: Fornisce un framework teorico unificato per regressione e classificazione non bilanciate
Esperimenti Completi: Validazione su molteplici progettazioni circuitali reali, risultati convincenti
Riproducibilità: Fornisce codice open-source, facilitando riproduzione e applicazione
Fase Pre-Layout: Valutazione rapida degli effetti parassitari, riduzione delle iterazioni di progettazione
Esplorazione dello Spazio di Progettazione: Screening rapido di candidati in spazi di progettazione su larga scala
Trasferimento Cross-Design: Applicazione di modelli addestrati su un circuito a nuove progettazioni circuitali
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.