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

تقدير الطفيليات القابلة للنقل عبر تعلم التباين الرسومي وإعادة توازن التسميات في دوائر AMS

المعلومات الأساسية

  • معرّف الورقة: 2507.06535
  • العنوان: تقدير الطفيليات القابلة للنقل عبر تعلم التباين الرسومي وإعادة توازن التسميات في دوائر AMS
  • المؤلفون: Shan Shen, Shenglu Hua, Jiajun Zou, Jiawei Liu, Jianwang Zhai, Chuan Shi, Wenjian Yu
  • التصنيف: cs.LG cs.SY eess.SY
  • تاريخ النشر: 10 أكتوبر 2025 (تقديم arXiv)
  • رابط الورقة: https://arxiv.org/abs/2507.06535

الملخص

تقدم هذه الورقة إطار عمل CircuitGCL لمعالجة مشكلة تقدير المعاملات الطفيلية في دوائر الإشارات المختلطة التناظرية (AMS). يعالج الإطار تحديات ندرة البيانات في تصميم الدوائس، وعدم توازن توزيع التسميات، والتنوع في تطبيقات الدوائر من خلال تعلم التباين الرسومي وتقنيات إعادة توازن التسميات. يستخدم CircuitGCL استراتيجية التعلم الذاتي غير الموجه، حيث يتعلم تضمينات العقد الثابتة طوبولوجياً من خلال تشتت التمثيل على الكرة الفائقة، مع إدخال دوال خسارة متوازنة MSE (BMSE) و softmax متقاطعة متوازنة (BSCE) لتخفيف الفروقات في توزيع التسميات بين الدوائر. تُظهر التجارب على تصاميم AMS في TSMC 28nm تحسناً بنسبة 33.64%~44.20% في R² لمهام انحدار الحواف، وتحسناً بنسبة 0.9×~2.1× في درجة F1 لمهام تصنيف العقد.

الخلفية البحثية والدافع

تعريف المشكلة

تدمج دوائر AMS الحديثة وحدات تناظرية (مثل المضخمات والمذبذبات) وأنظمة فرعية رقمية (مثل المتحكمات ومصفوفات SRAM)، وتتطلب عدداً كبيراً من التكرارات اليدوية أثناء عملية التصميم. مع تقليل عقد العملية باستمرار، تصبح التأثيرات الطفيلية ذات أهمية متزايدة، خاصة السعات الطفيلية التي تؤثر بشكل كبير على أداء الدائرة، مما يؤدي إلى زيادة تأخير الانتشار وارتفاع استهلاك الطاقة وتدهور سلامة الإشارة.

أهمية البحث

تعتمد عمليات التصميم التقليدية على محاكاة ما بعد التخطيط للتحقق من التأثيرات الطفيلية، وهي طريقة تستغرق وقتاً طويلاً وتكلفتها عالية. يمكن لتنبؤ المعاملات الطفيلية في مرحلة ما قبل التخطيط أن يقلل بشكل كبير من عدد تكرارات التصميم ويحسن كفاءة التصميم. توفر شبكات الأعصاب الرسومية (GNN) حلاً فعالاً لهذه المشكلة، حيث تمكن من نمذجة الدائرة كهيكل رسومي للتنبؤ بالمعاملات الطفيلية.

قيود الطرق الموجودة

  1. ندرة البيانات: عادة ما تكون بيانات دوائر AMS عالية الجودة (بما في ذلك شبكات SPICE ومعاملات الطفيليات في التخطيط) ملكية خاصة وتكلفة إنتاجها عالية جداً
  2. تنوع الدوائر: تمتد دوائر AMS عبر المجالات التناظرية والرقمية والمختلطة، مع مبادئ تصميم ومتطلبات أداء مختلفة
  3. عدم توازن التسميات: يُظهر توزيع السعات الطفيلية خصائص ذيل طويل، مع نقص شديد في عينات القيم الكبيرة للسعة
  4. ضعف القابلية للنقل: يصعب على الطرق الموجودة المدربة على نوع دائرة معين أن تعمم على طوبولوجيات دوائر أخرى

المساهمات الأساسية

  1. اقتراح إطار عمل CircuitGCL: يقوم بتكييف آلية تشتت التمثيل (RSM) مع تعلم التباين الرسومي لإنتاج تمثيلات قابلة للنقل يمكن تطبيقها مباشرة على تصاميم AMS الجديدة دون الحاجة إلى ضبط دقيق خاص بالمهمة
  2. تقنيات إعادة توازن التسميات: تحل مشكلة عدم التوازن في مجموعات بيانات الدوائر من خلال دوال خسارة MSE متوازنة و BSCE متوازنة، مما يعزز قدرة النموذج على النقل عبر المجالات
  3. إطار عمل نظري موحد: يوفر أساساً نظرياً موحداً لمشاكل الانحدار والتصنيف غير المتوازنة بناءً على مبادئ محاذاة التوزيع
  4. قابلية التطبيق الواسعة: يمكن توسيع الطريقة مباشرة لتشمل مهام التنبؤ بالمقاومة/الحث وتحليل التداخل والتقدير الكهربائي IR والنقل عبر العمليات

شرح الطريقة

تعريف المهمة

الإدخال: شبكة المخطط الكهربائي لدائرة AMS، نمذجة كرسم بياني غير متجانس G = (V, E)، حيث تمثل العقد V الشبكات والأجهزة والدبابيس، والحواف E تشفر علاقات الاتصال الإخراج:

  • مهمة انحدار الحواف: التنبؤ بقيم السعة المقترنة
  • مهمة تصنيف العقد: تصنيف السعات الأرضية إلى نطاقات منفصلة (صغيرة/متوسطة/كبيرة)

معمارية النموذج

1. وحدة تحويل الرسم البياني

تحويل رسم بياني AMS غير المتجانس إلى رسم بياني متجانس، من خلال تمييز عقد الشبكة والأجهزة والدبابيس باستخدام خصائص نوع العقدة X ∈ {0,1,2}^{N×1}. مصفوفة الميزات المحسّنة X_C ∈ R^{N×d_C} تلتقط معاملات التصميم التفصيلية ومعلومات الإحصائيات الاتصال.

2. آلية تشتت التمثيل (RSM)

التعريف: في فضاء التضمين ذي البعد D R^D، تفرض RSM قيدين:

  • قيد الابتعاد عن المركز: فصل أقصى لتضمينات العقد عن مركز التشتت c
  • قيد التوحيد: توزيع موحد لتضمينات العقد على الفضاء الجزئي S_k

التطبيق:

h̃_i = h_i / max(||h_i||_2, ε)  # تطبيع L2
L_scattering = -1/N ∑||h̃_i - c||²_2  # خسارة التشتت
c = 1/N ∑h̃_i  # مركز التشتت

3. المشفر عبر الإنترنت

يُنتج المشفر الهدف تمثيلات التشتت H_target = f_φ(A,X)، بينما ينتج المشفر عبر الإنترنت تمثيلات وسيطة H_online، مع الحصول على تمثيلات التنبؤ z_online من خلال المتنبئ q_θ(·). خسارة المحاذاة هي:

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

يتم تحديث معاملات المشفر الهدف من خلال المتوسط المتحرك الأسي: φ ← τφ + (1-τ)θ

نقاط الابتكار التقني

1. المزايا مقابل DSPD

يستخدم CircuitGPS مسافة أقصر مسار ثنائي الارتساء (DSPD) كترميز موضعي، لكن التكلفة الحسابية والتخزينية تنمو بشكل تربيعي مع حجم الرسم البياني. يتمتع التدريب المسبق GCL في CircuitGCL بتوازي عالي وقابلية توسع نموذج جيدة، متفوقة بشكل كبير على DSPD في الدوائر الكبيرة.

2. الأساس النظري لإعادة توازن التسميات

بناءً على نظرية بايز، العلاقة بين توزيع التدريب p_train(y|x) والتوزيع المتوازن p_bal(y|x) هي:

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

إعداد التجارب

مجموعة البيانات

استخدام 6 تصاميم دوائر AMS المطبقة في تقنية TSMC 28nm CMOS:

  • مجموعة التدريب: SSRAM (87K عقدة، 134K حافة)
  • مجموعة الاختبار: DIGITAL CLK GEN, TIMING CTRL, ARRAY 128 32, ULTRA8T, SANDWICH-RAM
  • يحتوي أكبر تصميم على 4.3M عقدة و13.3M حافة

مقاييس التقييم

  • مهام الانحدار: MAE, MSE, R²
  • مهام التصنيف: Accuracy, Precision, Recall, F1-score

طرق المقارنة

  1. ParaGraph: نموذج متكامل قائم على MPNN
  2. DLPL-Cap: مسجلات GNN متعددة الخبراء
  3. CircuitGPS: طريقة التعلم القليل العينات باستخدام الترميز الموضعي

تفاصيل التطبيق

  • المشفر: 4 طبقات ClusterGCN، 256 بُعد مخفي، تفعيل Tanh، 0.3 dropout
  • GNN النهائي: 5 طبقات GraphSAGE، 144 بُعد مخفي، تفعيل PReLU
  • σ_noise = 0.001، τ = 0.99

نتائج التجارب

النتائج الرئيسية

مهمة انحدار الحواف (تقدير السعة المقترنة)

مجموعة الاختبارأفضل طريقةتحسن R²
TIMING CTRLCircuitGCL(GAI)41.08%
ARRAY 128 32CircuitGCL(GAI)44.20%
ULTRA8TCircuitGCL(GAI)37.44%
SANDWICH-RAMCircuitGCL(GAI)33.64%

مهمة تصنيف العقد (تصنيف السعة الأرضية)

مجموعة الاختبارأفضل طريقةتحسن F1
DIGITAL CLK GENCircuitGCL(BSCE)0.9×
ARRAY 128 32CircuitGCL(BSCE)2.1×
ULTRA8TCircuitGCL(BSCE)1.2×
SANDWICH-RAMCircuitGCL(BSCE)1.5×

تجارب الاستئصال

تحليل تأثير RSM

تحقق RSM تحسناً في الأداء على جميع مجموعات البيانات:

  • أقصى تحسن في R² بنسبة 26.9% (ARRAY 128 32)
  • أقصى تحسن في F1 بنسبة 20.0% (ULTRA8T)
  • الحد الأدنى للتحسن يصل إلى 4.1% (F1) و6.56% (R²)

تأثير إعادة توازن التسميات

يؤدي تطبيق BSCE على جميع الطرق الأساسية إلى تحسن كبير في الدقة، خاصة على التصاميم الكبيرة. يحسّن MSE المتوازن بشكل كبير أداء النموذج في مناطق البيانات النادرة.

الأعمال ذات الصلة

التنبؤ بالمعاملات الطفيلية

  • ParaGraph: تحويل المخطط الكهربائي للدائرة إلى رسم بياني، استخدام MPNN للتنبؤ بسعات الشبكة ومعاملات التخطيط
  • DLPL-Cap: دمج موجه GNN وخمسة مسجلات خبراء للتعامل مع توزيع البيانات غير المتوازن في دوائر SRAM
  • CircuitGPS: استخدام التعلم القليل العينات والترميز الموضعي منخفض التكلفة للتنبؤ بالمعاملات الطفيلية

تعلم التباين الرسومي

تنفذ أطر عمل GCL السائدة ضمنياً تشتت التمثيل، والذي يلعب دوراً حاسماً في نجاحها. تستخدم هذه الورقة SGRL كأساس لـ GCL، مع تضمين تمثيلات العقد على الكرة الفائقة.

التعلم غير المتوازن

  • التصنيف: إعادة أخذ العينات، إعادة ترجيح الخسارة، تعديل logit وغيرها
  • الانحدار: مستكشف نسبياً، الطرق الموجودة تركز بشكل أساسي على تكييف SMOTE وإعادة ترجيح الخسارة

الخلاصة والنقاش

الاستنتاجات الرئيسية

  1. يحل CircuitGCL بنجاح مشاكل ندرة البيانات وعدم توازن التسميات في تصميم دوائر AMS من خلال التعلم الذاتي غير الموجه للتمثيلات ودوال الخسارة الحساسة للتوزيع
  2. تتمتع التضمينات الثابتة طوبولوجياً التي تنتجها RSM بقدرة ممتازة على النقل عبر المجالات
  3. توحد دوال الخسارة المتوازنة نظرياً مشاكل الانحدار والتصنيف غير المتوازنة
  4. تحقق أداء متقدمة على عدة تصاميم TSMC 28nm

القيود

  1. تم التحقق فقط على عملية TSMC 28nm، وتبقى قدرة النقل عبر عقد العملية قيد الاختبار
  2. يركز حالياً على التنبؤ بالسعة، وتبقى قابلية التطبيق على التنبؤ بالمقاومة/الحث قيد التحقق
  3. لا تزال كفاءة الحساب للدوائر الكبيرة بحاجة إلى تحسين

الاتجاهات المستقبلية

  1. التوسع إلى أنواع دوائر AMS أوسع
  2. التكيف مع تقدير المقاومة الطفيلية
  3. التكامل مع أدوات التخطيط والتوجيه الحساسة لـ RC
  4. البحث في التعلم النقل عبر عقد العملية

التقييم المتعمق

المزايا

  1. أهمية المشكلة: معالجة نقطة ألم فعلية في مجال EDA، بقيمة هندسية مهمة
  2. ابتكار الطريقة: أول تطبيق لتعلم التباين الرسومي على التنبؤ بالمعاملات الطفيلية في الدوائر، مع تكييف RSM ذكي
  3. المساهمة النظرية: توفير إطار نظري موحد لمشاكل الانحدار والتصنيف غير المتوازنة
  4. التجارب الشاملة: التحقق على عدة تصاميم دوائر حقيقية، مع نتائج مقنعة
  5. قابلية إعادة الإنتاج: توفير كود مفتوح المصدر لتسهيل إعادة الإنتاج والتطبيق

أوجه القصور

  1. قيود مجموعة البيانات: استخدام بيانات من عقدة عملية واحدة فقط، مع قدرة تعميم قيد الاختبار
  2. التكلفة الحسابية: على الرغم من الكفاءة مقابل DSPD، لا يزال التدريب المسبق GCL يتطلب موارد حسابية إضافية
  3. هندسة الميزات: لا يزال استخراج وتمثيل ميزات الدائرة يعتمد على التصميم اليدوي
  4. التحليل النظري: نقص التحليل النظري العميق حول سبب ملاءمة RSM بشكل خاص للرسوم البيانية للدوائر

التأثير

  1. القيمة الأكاديمية: فتح تطبيق تعلم التباين الرسومي في مجال EDA، قد يلهم المزيد من الأبحاث ذات الصلة
  2. القيمة الهندسية: قابل للتطبيق مباشرة على أدوات EDA التجارية، لتحسين كفاءة التصميم
  3. عمومية الطريقة: يمكن توسيع الإطار ليشمل أنواعاً أخرى من التنبؤ بالمعاملات الطفيلية ومهام تحليل الدوائر

السيناريوهات المطبقة

  1. مرحلة ما قبل التخطيط: التقييم السريع للتأثيرات الطفيلية، تقليل تكرارات التصميم
  2. استكشاف فضاء التصميم: الفحص السريع للمرشحين في فضاء تصميم واسع
  3. النقل عبر التصاميم: تطبيق نموذج مدرب على دائرة واحدة على تصاميم دوائر جديدة
  4. تكامل أدوات EDA: كوحدة خوارزمية أساسية في أدوات EDA التجارية

التقييم الإجمالي: هذه ورقة بحثية عالية الجودة متعددة التخصصات، تطبق بنجاح تقنيات التعلم الآلي المتقدمة على مشاكل عملية في مجال EDA. الطريقة مبتكرة، التجارب شاملة، النتائج كبيرة، وتتمتع بقيمة نظرية وعملية مهمة.