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

Übertragbare Parasitäre Schätzung mittels Graph-Kontrastivem Lernen und Label-Rebalancierung in AMS-Schaltkreisen

Grundinformationen

  • Paper-ID: 2507.06535
  • Titel: Transferable Parasitic Estimation via Graph Contrastive Learning and Label Rebalancing in AMS Circuits
  • Autoren: Shan Shen, Shenglu Hua, Jiajun Zou, Jiawei Liu, Jianwang Zhai, Chuan Shi, Wenjian Yu
  • Klassifizierung: cs.LG cs.SY eess.SY
  • Veröffentlichungsdatum: 10. Oktober 2025 (arXiv-Einreichung)
  • Paper-Link: https://arxiv.org/abs/2507.06535

Zusammenfassung

Diese Arbeit präsentiert das CircuitGCL-Framework zur Schätzung parasitärer Parameter in analogen Mischsignal-(AMS-)Schaltkreisen. Das Framework adressiert durch Graph-Kontrastives Lernen und Label-Rebalancierungstechniken die Herausforderungen der Datensparsamkeit in der Schaltkreisentwicklung, der Unausgeglichenheit der Labelverteilung sowie der Vielfalt der Schaltkreisimplementierungen. CircuitGCL nutzt eine selbstüberwachte Strategie, um durch Darstellungsstreuung topologieinvariante Knoten-Embeddings zu lernen, während gleichzeitig ausgeglichene mittlere quadratische Fehler-(BMSE-)und ausgeglichene Softmax-Kreuzentropie-(BSCE-)Verlustfunktionen eingeführt werden, um Labelverteilungsunterschiede zwischen Schaltkreisen zu mildern. Experimente auf TSMC 28nm AMS-Designs zeigen Verbesserungen von R² um 33,64%–44,20% bei Kantenregressionsaufgaben und F1-Score-Verbesserungen um das 0,9×–2,1×-fache bei Knotenklassifizierungsaufgaben.

Forschungshintergrund und Motivation

Problemdefinition

Moderne AMS-Schaltkreise integrieren analoge Module (wie Verstärker, Oszillatoren) und digitale Subsysteme (wie Controller, SRAM-Arrays) und erfordern während des Designprozesses umfangreiche manuelle Iterationen. Mit schrumpfenden Prozesstechnologieknoten werden parasitäre Effekte zunehmend kritisch, insbesondere parasitäre Kapazitäten, die die Schaltkreisleistung erheblich beeinflussen und zu erhöhter Ausbreitungsverzögerung, gestiegenem Stromverbrauch und beeinträchtigter Signalintegrität führen.

Forschungsrelevanz

Herkömmliche Designabläufe sind auf Post-Layout-Simulation angewiesen, um parasitäre Effekte zu verifizieren – ein zeitaufwändiges und kostspieliges Verfahren. Die Vorhersage parasitärer Parameter in der Pre-Layout-Phase kann die Anzahl der Designiterationen erheblich reduzieren und die Designeffizienz verbessern. Graphische neuronale Netze (GNNs) bieten eine effektive Lösung, indem sie Schaltkreise als Graphstrukturen modellieren, um parasitäre Parameter vorherzusagen.

Einschränkungen bestehender Methoden

  1. Datensparsamkeit: Hochwertige AMS-Schaltkreisdaten (einschließlich SPICE-Netze und Layout-Parasiten) sind typischerweise proprietär und teuer zu generieren
  2. Schaltkreisvielfalt: AMS-Schaltkreise erstrecken sich über analoge, digitale und Mischsignalbereiche mit unterschiedlichen Designprinzipien und Leistungsanforderungen
  3. Label-Unausgeglichenheit: Parasitäre Kapazitätsverteilungen zeigen Long-Tail-Charakteristiken mit stark unterrepräsentierten großen Kapazitätswerten
  4. Schlechte Übertragbarkeit: Bestehende Methoden, die auf spezifischen Schaltkreistypen trainiert werden, lassen sich schwer auf andere Schaltkreistopologien verallgemeinern

Kernbeiträge

  1. CircuitGCL-Framework: Adaptiert den Darstellungsstreumechanismus (RSM) in Graph-Kontrastives Lernen, um übertragbare Darstellungen zu generieren, die direkt auf ungesehene AMS-Designs anwendbar sind, ohne aufgabenspezifisches Fine-Tuning
  2. Label-Rebalancierungstechniken: Adressiert Unausgeglichenheit in Schaltkreisdatensätzen durch ausgeglichene MSE- und BSCE-Verlustfunktionen und verbessert die domänenübergreifende Übertragungsfähigkeit
  3. Einheitlicher theoretischer Rahmen: Bietet auf Verteilungsausrichtungsprinzipien basierende einheitliche theoretische Grundlagen für unausgeglichene Regression und Klassifizierung
  4. Breite Anwendbarkeit: Die Methode lässt sich direkt auf Widerstands-/Induktivitätsvorhersage, Übersprechungsanalyse, IR-Spannungsabfallschätzung und prozessübergreifende Übertragung erweitern

Methodische Details

Aufgabendefinition

Eingabe: Schaltplan-Netzliste eines AMS-Schaltkreises, modelliert als heterogener Graph G = (V, E), wobei Knoten V Netze, Transistoren und Pins darstellen und Kanten E Verbindungsbeziehungen kodieren Ausgabe:

  • Kantenregressionsaufgabe: Vorhersage gekoppelter Kapazitätswerte
  • Knotenklassifizierungsaufgabe: Klassifizierung von Erdungskapazitäten in diskrete Bereiche (klein/mittel/groß)

Modellarchitektur

1. Graph-Transformationsmodul

Konvertiert heterogene AMS-Graphen in homogene Graphen durch Knotentyp-Attribute X ∈ {0,1,2}^{N×1}, die Netz-, Geräte- und Pin-Knoten unterscheiden. Die erweiterte Merkmalsmatrix X_C ∈ R^{N×d_C} erfasst detaillierte Designparameter und Verbindungsstatistiken.

2. Darstellungsstreumechanismus (RSM)

Definition: Im D-dimensionalen Einbettungsraum R^D erzwingt RSM zwei Einschränkungen:

  • Zentrum-Abstands-Einschränkung: Maximale Trennung von Knoten-Embeddings vom Streuzentrum c
  • Gleichmäßigkeits-Einschränkung: Gleichmäßige Verteilung von Knoten-Embeddings im Unterraum S_k

Implementierung:

h̃_i = h_i / max(||h_i||_2, ε)  # L2-Normalisierung
L_scattering = -1/N ∑||h̃_i - c||²_2  # Streuverlust
c = 1/N ∑h̃_i  # Streuzentrum

3. Online-Encoder

Der Ziel-Encoder generiert Streudarstellungen H_target = f_φ(A,X), der Online-Encoder erzeugt Zwischenrepräsentationen H_online, wobei durch Prädiktor q_θ(·) Vorhersagedarstellungen z_online erhalten werden. Der Ausrichtungsverlust ist:

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

Ziel-Encoder-Parameter werden durch exponentiellen gleitenden Durchschnitt aktualisiert: φ ← τφ + (1-τ)θ

Technische Innovationen

1. Vorteile gegenüber DSPD

CircuitGPS nutzt Dual-Anker-Kürzeste-Pfad-Distanz (DSPD) als Positionskodierung, aber die Rechen- und Speicherkosten wachsen quadratisch mit der Graphgröße. Das GCL-Vortraining von CircuitGCL hat hohe Parallelisierbarkeit und gute Modellskalierbarkeit, deutlich überlegen auf großen Schaltkreisen.

2. Theoretische Grundlagen der Label-Rebalancierung

Basierend auf dem Bayes-Theorem ist die Beziehung zwischen Trainingsverteilung p_train(y|x) und ausgeglichener Verteilung p_bal(y|x):

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

Experimentelle Einrichtung

Datensätze

Sechs AMS-Schaltkreisdesigns implementiert in TSMC 28nm CMOS-Technologie:

  • Trainingssatz: SSRAM (87K Knoten, 134K Kanten)
  • Testsätze: DIGITAL CLK GEN, TIMING CTRL, ARRAY 128 32, ULTRA8T, SANDWICH-RAM
  • Größtes Design enthält 4,3M Knoten und 13,3M Kanten

Bewertungsmetriken

  • Regressionsaufgaben: MAE, MSE, R²
  • Klassifizierungsaufgaben: Accuracy, Precision, Recall, F1-Score

Vergleichsmethoden

  1. ParaGraph: MPNN-basiertes Ensemble-Modell
  2. DLPL-Cap: Multi-Expert-GNN-Regressor
  3. CircuitGPS: Few-Shot-Learning-Methode mit Positionskodierung

Implementierungsdetails

  • Encoder: 4-schichtiges ClusterGCN, 256 versteckte Dimensionen, Tanh-Aktivierung, 0,3 Dropout
  • Downstream-GNN: 5-schichtiges GraphSAGE, 144 versteckte Dimensionen, PReLU-Aktivierung
  • σ_noise = 0,001, τ = 0,99

Experimentelle Ergebnisse

Hauptergebnisse

Kantenregressionsaufgabe (Gekoppelte Kapazitätsschätzung)

TestsatzBeste MethodeR²-Verbesserung
TIMING CTRLCircuitGCL(GAI)41,08%
ARRAY 128 32CircuitGCL(GAI)44,20%
ULTRA8TCircuitGCL(GAI)37,44%
SANDWICH-RAMCircuitGCL(GAI)33,64%

Knotenklassifizierungsaufgabe (Erdungskapazitätsklassifizierung)

TestsatzBeste MethodeF1-Verbesserung
DIGITAL CLK GENCircuitGCL(BSCE)0,9×
ARRAY 128 32CircuitGCL(BSCE)2,1×
ULTRA8TCircuitGCL(BSCE)1,2×
SANDWICH-RAMCircuitGCL(BSCE)1,5×

Ablationsstudien

RSM-Effektanalyse

RSM bringt Leistungsverbesserungen auf allen Datensätzen:

  • Maximale R²-Verbesserung 26,9% (ARRAY 128 32)
  • Maximale F1-Verbesserung 20,0% (ULTRA8T)
  • Minimale Verbesserung 4,1% (F1) und 6,56% (R²)

Effekt der Label-Rebalancierung

BSCE-Anwendung auf alle Baselines bringt signifikante Genauigkeitsverbesserungen, besonders ausgeprägt bei großen Designs. Ausgeglichenes MSE verbessert Modellleistung in datensparamen Regionen erheblich.

Verwandte Arbeiten

Parasitäre Parametervorhersage

  • ParaGraph: Konvertiert Schaltplan in Graphen, nutzt MPNN zur Vorhersage von Netzkapazitäten und Layout-Parametern
  • DLPL-Cap: Kombiniert GNN-Router und fünf Expert-Regressoren zur Behandlung unausgeglichener Datenverteilung in SRAM-Schaltkreisen
  • CircuitGPS: Nutzt Few-Shot-Learning und kostengünstige Positionskodierung für parasitäre Parametervorhersage

Graph-Kontrastives Lernen

Mainstream-GCL-Frameworks führen implizit Darstellungsstreuung durch, die für ihren Erfolg entscheidend ist. Diese Arbeit nutzt SGRL als GCL-Grundlage und bettet Knotendarstellungen auf der Hypersphäre ein.

Unausgeglichenes Lernen

  • Klassifizierung: Daten-Resampling, Verlust-Regewichtung, Logit-Anpassung und weitere Methoden
  • Regression: Relativ untererforscht, bestehende Methoden hauptsächlich SMOTE-Adaption und Verlust-Regewichtung

Schlussfolgerungen und Diskussion

Hauptschlussfolgerungen

  1. CircuitGCL löst erfolgreich Datensparsamkeit und Label-Unausgeglichenheit in AMS-Schaltkreisdesign durch selbstüberwachtes Repräsentationslernen und verteilungsbewusste Verlustfunktionen
  2. Von RSM generierte topologieinvariante Embeddings zeigen ausgezeichnete domänenübergreifende Übertragungsfähigkeit
  3. Ausgeglichene Verlustfunktionen vereinheitlichen theoretisch unausgeglichene Regressions- und Klassifizierungsprobleme
  4. Erreichen SOTA-Leistung auf mehreren TSMC 28nm Designs

Einschränkungen

  1. Validierung nur auf TSMC 28nm-Technologie; Übertragungsfähigkeit über Prozesstechnologieknoten hinweg bedarf weiterer Validierung
  2. Derzeit fokussiert auf Kapazitätsvorhersage; Anwendbarkeit auf Widerstands-/Induktivitätsvorhersage bedarf weiterer Verifikation
  3. Rechnerische Effizienz bei großen Schaltkreisen hat noch Optimierungspotenzial

Zukünftige Richtungen

  1. Erweiterung auf breitere AMS-Schaltkreistypen
  2. Anpassung an parasitäre Widerstandsschätzung
  3. Integration in RC-bewusste Platzierungs- und Routing-Werkzeuge
  4. Transferlernforschung über Prozesstechnologieknoten hinweg

Tiefgreifende Bewertung

Stärken

  1. Problemrelevanz: Adressiert praktische Schmerzpunkte im EDA-Bereich mit wichtigem Ingenieurwert
  2. Methodische Innovativität: Erste Anwendung von Graph-Kontrastivem Lernen auf Schaltkreis-Parasitärparametervorhersage, RSM-Adaption elegant
  3. Theoretischer Beitrag: Bietet einheitlichen theoretischen Rahmen für unausgeglichene Regression und Klassifizierung
  4. Umfangreiche Experimente: Validierung auf mehreren echten Schaltkreisdesigns mit überzeugenden Ergebnissen
  5. Reproduzierbarkeit: Bereitstellung von Open-Source-Code für einfache Reproduktion und Anwendung

Mängel

  1. Datensatz-Einschränkungen: Verwendung nur einzelner Prozesstechnologieknoten; Generalisierungsfähigkeit bedarf Validierung
  2. Rechnerischer Overhead: Obwohl effizienter als DSPD, erfordert GCL-Vortraining zusätzliche Rechenressourcen
  3. Feature-Engineering: Extraktion und Darstellung von Schaltkreismerkmalen bleibt auf manuellem Design angewiesen
  4. Theoretische Analyse: Mangelnde tiefgreifende theoretische Analyse, warum RSM besonders für Schaltkreisgraphen geeignet ist

Auswirkungen

  1. Akademischer Wert: Eröffnet Anwendung von Graph-Kontrastivem Lernen im EDA-Bereich, könnte weitere verwandte Forschung inspirieren
  2. Ingenieurwert: Direkt anwendbar auf kommerzielle EDA-Werkzeuge, verbessert Designeffizienz
  3. Methodische Universalität: Framework erweiterbar auf andere Parasitärparametervorhersagen und Schaltkreisanalyseaufgaben

Anwendungsszenarien

  1. Pre-Layout-Phase: Schnelle Bewertung parasitärer Effekte, Reduktion von Designiterationen
  2. Designraum-Exploration: Schnelle Filterung von Kandidatenlösungen in großen Designräumen
  3. Designübergreifende Übertragung: Anwendung von auf einem Schaltkreis trainiertem Modell auf neue Designs
  4. EDA-Werkzeug-Integration: Als Kernalgorithmusmodul in kommerziellen EDA-Werkzeugen

Gesamtbewertung: Dies ist ein hochqualitatives interdisziplinäres Forschungspapier, das erfolgreich Spitzentechnologien des maschinellen Lernens auf praktische Probleme im EDA-Bereich anwendet. Die Methode ist innovativ, Experimente umfassend, Ergebnisse signifikant und bietet wichtigen theoretischen und praktischen Wert.