PrefixAgent: An LLM-Powered Design Framework for Efficient Prefix Adder Optimization
إن PrefixAgent هو إطار عمل مدعوم بنماذج اللغات الكبيرة (LLM) يعمل على تحسين تصميم جامع البادئة (prefix adder) من خلال تفكيك المهمة إلى تخليق الهيكل الأساسي وتحسين البنية بتوجيه من التغذية الراجعة لأدوات التصميم الآلي للإلكترونيات (EDA)، مع الاستفادة من الرسوم البيانية-e (e-graphs) لتوليد بيانات تدريب دقيقة للغاية تعمل على تحسين كفاءة المساحة وقابلية التوسع بشكل كبير عبر مختلف عرض البتات.
المؤلفون الأصليون:Dongsheng Zuo, Jiadong Zhu, Yang Luo, Yuzhe Ma
في قلب القلب السيليكوني لكل حاسوب حديث، من الهاتف الذكي في الجيب إلى الحاسوب الفائق الذي يدرب الذكاء الاصطناعي، تكمن مهمة أساسية: جمع الأرقام. وبينما يبدو هذا الأمر بسيطاً بالنسبة للإنسان، إلا أن الدوائر الإلكترونية التي تقوم بعملية الجمع هذه يجب أن تفعل ذلك بسرعة الضوء، موازنةً بين السرعة والمساحة الفيزيائية التي تشغلها. وتعد إحدى أكثر الطرق كفاءة لبناء هذه الدوائر هي ما يسمى بـ "الجامع البادئ" (prefix adder). تخيل فريقاً من العمال ينقلون رسالة على طول خط؛ يقوم الجامع البادئ بتنظيم هذا الفريق بحيث يعرف الجميع النتيجة النهائية بشكل فوري تقريباً، بدلاً من الانتظار حتى تنتقل الرسالة واحداً تلو الآخر. ومع ذلك، كلما كبرت الأرقام — مما يتطلب المزيد من البتات لتمثيلها — ينمو عدد الطرق الممكنة لترتيب هؤلاء العمال بشكل انفجاري. إن البحث عن الترتيب المثالي لعدد كبير يشبه محاولة العثور على حبة رمل واحدة محددة على شاطئ يستمر في النمو؛ فهو واسع جداً بحيث لا يمكن البحث فيه عبر كل الاحتمالات، ولا يمكن للمهندسين البشر تصميم أفضل نسخة يدوياً لكل شريحة جديدة.
لسنوات، حاول الباحثون استخدام الحواسيب لحل هذا اللغز، معتمدين غالباً على أساليب التجربة والخطأ التي تتسم بالبطء وتواجه صعوبة في التكيف مع المواقف الجديدة. ويقدم نهج جديد، فصلته مؤخراً أعمال بحثية أجراها باحثون من جامعة هونغ كونغ للعلوم والتكنولوجيا، نوعاً مختلفاً من المساعدين: نموذج لغوي كبير، وهو نفس نوع التكنولوجيا الكامنة وراء الذكاء الاصطناعي التوليدي المتقدم. ولكن بدلاً من مطالبته بكتابة الأكواد أو الدردشة، علم الباحثون هذا النموذج ليعمل كمعماري ماهر لهذه الدوائر الإلكترونية. لقد أنشأوا نظاماً يسمى "PrefixAgent"، وهو لا يحاول بناء الدائرة المعقدة بأكملها في قفزة واحدة ضخمة، بل يقسم المهمة إلى خطوتين يمكن إدارتهما. أولاً، يصمم النظام "الهيكل العظمي"، وهو الهيكل الإنشائي الرئيسي للجامع الذي يحدد سرعته وشكله العام. وبمجرد أن يصبح هذا الهيكل صلباً، ينتقل النظام إلى مرحلة ثانية، وهي إجراء تعديلات صغيرة ودقيقة على الاتصالات المحلية لضمان دقة التوقيت. يسمح هذا التقسيم في العمل للذكاء الاصطناعي بالتركيز على القرارات رفيعة المستوى بينما تتولى أدوات متخصصة التعامل مع التفاصيل المعقدة لهندسة الدائرة.
ولتعليم هذا الذكاء الاصطناعي كيفية اتخاذ قرارات جيدة، واجه الباحثون عقبة كبيرة: لم يكن هناك مكتبة موجودة مسبقاً من الأمثلة المثالية لتوضيح ما يجب فعله. ولحل هذه المشكلة، استخدموا تقنية رياضية تسمى "إشباع التساوي" (equality saturation)، والتي تعمل مثل آلة يمكنها توليد كل التباينات الصالحة لهيكل الدائرة في آن واحد. ومن هذه المجموعة الضخمة من الاحتمالات، استخرجوا أفضلها وسجلوا الخطوات الدقيقة المتخذة للوصول إليها. ثم استخدموا هذه الخطوات المسجلة لتدريب الذكاء الاصطناعي، حيث لم يكتفوا بإظهار الإجابة النهائية له فحسب، بل أظهروا أيضاً عملية الاستنتاج وراء كل تغيير. سمح هذا التدريب للذكاء الاصطناعي بتعلم استراتيجية عامة للتحسين بدلاً من مجرد حفظ تصميمات محددة. وعند اختباره، أثبت النظام فعالية ملحوظة؛ ففي كل سيناريو جربوه تقريباً، بما في ذلك التصميمات للأرقام ذات الـ 64 بت، استخدمت الدوائر التي صممها الذكاء الاصطناعي مساحة فيزيائية أقل من تلك التي أنشأتها الطرق التقليدية أو مناهج التعلم الآلي الأخرى. وأصبح هذا التفوق أكثر وضوحاً مع كبر حجم الأرقام، حيث أنتج الذكاء الاصطناعي تصميمات أصغر بكثير من تلك التي تنتجها الأدوات التجارية المستخدمة من قبل مصنعي الرقائق اليوم.
يسلط نجاح هذه الطريقة الضوء أيضاً على تحول في كيفية حل المشكلات الهندسية المعقدة في المستقبل. فمن خلال الجمع بين قدرة الاستنتاج لنموذج لغوي كبير ودقة أدوات التصميم المتخصصة، أنشأ الباحثون نظاماً يمكنه التكيف مع قيود جديدة دون الحاجة إلى إعادة تدريبه من الصفر لكل تصميم شريحة جديد. وأظهرت النتائج أن الذكاء الاصطناعي يمكنه التعامل مع الظروف غير المتجانسة، حيث تتلقى بعض أجزاء الدائرة الإشارات في أوقات مختلفة، وهو تحدٍ واقعي شائع غالباً ما يحبط الأدوات المؤتمتة. وفي الاختبارات التي شملت تدفق تصنيع تجاري، تفوقت الدوائر المصممة بواسطة الذكاء الاصطناعي باستمرار على الحلول القياسية في الصناعة، مما قدم مساراً واضحاً نحو أجهزة إلكترونية أكثر كفاءة وقوة. يشير هذا العمل إلى أن مستقبل تصميم الرقائق قد لا يعتمد فقط على الحدس البشري أو البحث بالقوة الغاشمة، بل على وكلاء أذكياء يمكنهم الاستنتاج عبر المشكلات الهيكلية المعقدة، والتعلم من المنطق ذاته للدوائر التي طُلب منهم بناؤها.
ملخص تقني: PrefixAgent
بيان المشكلة تعد الجامعات البادئة (Prefix adders) من الدوائر الحسابية الأساسية في الأنظمة الرقمية، ومع ذلك فإن مساحة التصميم الخاصة بها تنمو بشكل أسي مع زيادة عرض البت (bit-width)، مما يجعل الاستكشاف الشامل غير عملي. تعتمد منهجيات التصميم التقليدية على هياكل منتظمة (مثل Kogge–Stone أو Brent–Kung) أو التحسين اليدوي، والتي غالبًا ما تفشل في تحقيق التوازن بين المساحة، والتأخير، والطاقة تحت قيود زمنية متنوعة وأوقات وصول مدخلات غير منتظمة. وبينما تم اقتراح مناهج تعلم الآلة (مثل التعلم التعزيزي RL، أو المشفرات التلقائية المتغيرة VAE، أو البحث عبر شجرة مونت كارلو MCTS) لأتمتة هذه العملية، إلا أنها تعاني من ضعف التعميم، ومشاções القابلية للتوسع بسبب متطلبات إعادة التدريب المكثفة للسيناريوهات الجديدة، وعدم الكفاءة في استكشاف مساحات العمل الكبيرة. علاوة على ذلك، تواجه تطبيقات النماذج اللغوية الكبيرة (LLM) في تصميم الدوائر عائقين حرجين: (1) عدم القدرة على التفكير بكفاءة في الرسوم البيانية المعقدة التي تحتوي على أكثر من 20 عقدة، و(2) نقص البيانات عالية الجودة والقابلة للتفسير لضبط النماذج (fine-tuning) لمهام تحسين محددة.
المنهجية يقترح البحث PrefixAgent، وهو إطار عمل لتصميم الدوائر مدعوم بالنماذج اللغوية الكبيرة (LLM)، يعالج هذه التحديات من خلال نهج التفكيك ثنائي المراحل والاستدلال المتكامل مع الأدوات (TIR).
التفكيك ثنائي المراحل:
المرحلة الأولى (تحسين الهيكل الأساسي - Backbone Optimization): بدلاً من توليد رسم بيفي بادئ كامل مباشرة، يقوم إطار العمل بتفكيك المشكلة إلى "تخليق الهيكل الأساسي" و"تحسين الهيكل المحلي". يُعرَّف الهيكل الأساسي بأنه الرسم البيفي الفرعي الذي يحسب الحمل (carry) للبت الأكثر أهمية (MSB). يقلل هذا من مساحة التصميم من حاصل ضرب أعداد كاتالان (Catalan numbers) إلى عدد كاتالان واحد، مما يبسط التعقيد بشكل كبير. يقوم النموذج اللغوي الكبير بتحسين هيكل الهيكل الأساسي التسلسلي الأولي بشكل تكراري من خلال إجراء عمليات "إعادة التجميع" (regroup) (إعادة تنظيم الأشجار الفرعية) لتحسين نمط التجميع.
المرحلة الثانية (تحسين الهيكل المحلي - Local Structure Refinement): بمجرد إنشاء هيكل أساسي مُحسَّن، يتم بناء جامع بادئ كامل عن طريق إضافة عقد مساعدة. بعد ذلك، يقوم النموذج اللغوي الكبير بإجراء تحسينات هيكلية محلية (مثل تحسين المستوى، وتقليل التفرع/fanout، واستنساخ العقد) مسترشدًا بالتغذية الراجعة من أدوات التنفيذ الفيزيائي لحل انتهاكات التوقيت.
الاستدلال المتكامل مع الأدوات (TIR): يستخدم PrefixAgent استدعاء الدوال (function calling) للسماف للنموذج اللغوي الاستدلالي الكبير (LRM) بتفويض التعديلات الهيكلية الدقيقة إلى أدوات خارجية. يركز النموذج اللغوي الكبير على اتخاذ القرار رفيع المستوى (تحديد ماذا يجب تحسينه وأين)، بينما تقوم الأدوات الخارجية بتنفيذ التعديلات وتقديم التغذية الراجعة (مثل التعبيرات S-expressions الملحقة بالتوقيت، والمقاييس الفيزيائية). هذا يسد الفجوة بين الاستدلال عالي المستوى والدقة الخوارزمية.
توليد البيانات القائم على الرسم البيفي E-Graph: لمعالجة ندرة البيانات عالية الجودة، استخدم المؤلفون الرسوم البيفية e-graphs والتشبع بالتساوي (equality saturation). لقد حددوا لغة وسيطة خاصة بالمجال (BackboneLang) لتمثيل هياكل الهيكل الأساسي كتعبيرات S-expressions. ومن خلال تطبيق قواعد إعادة الكتابة الترابطية داخل e-graph، قاموا باستكشاف مساحة الحلول بشكل منهجي. يستخرج إطار العمل مرشحي الهيكل الأساسي المُحسَّن باستخدام دالة تكلفة مدركة للتوقيت، ويستخدم مفسر e-graph لتوليد مسارات إعادة كتابة قابلة للتفسير. تعمل هذه المسارات كحقائق أرضية (ground truth) لتخليق بيانات استدلال سلسلة الأفك "Chain-of-Thought (CoT)"، والتي تُستخدم لضبط النموذج اللغوي الكبير (LRM).
المساهمات الرئيسية
إطار عمل PrefixAgent: أول نهج يعتمد على النماذج اللغوية الكبيرة قادر على توسيع نطاق تحسين الجامع البادئ لتصاميم 64 بت، متفوقًا على الأدوات التجارية.
منظور تصميم مبتكر: استراتيجية تفكيك تفصل بين توليد الهيكل الأساسي والتحسين المحلي، مما يتيح الاستدلال الفعال وتخليق جامع خالٍ من العيوب.
نموذج توليد البيانات: طريقة منهجية تستخدم e-graphs لتوليد مسارات تحسين واسعة النطاق وعالية الجودة وقابلة للتفسير لضبط النماذج اللغوية الكبيرة، مما يتغلب على ندرة البيانات في تصميم الدوائر.
الاستدلال المتكامل مع الأدوات: بنية يعمل فيها النموذج اللغوي الكبير كمتحكم يستدعي أدوات خارجية للتنفيذ، مما يضمن الصلاحية الهيكلية والجدوى الفيزيائية.
النتائج التجريبية أُجريت التجارب باستخدام نموذج QwQ-32B الذي تم ضبطه بدقة باستخدام البيانات المولدة، وتم تقييمه تحت ملفات تعريف وصول مدخلات منتظمة وغير منتظمة باستخدام مكتبة NanGate45 وتدفق OpenROAD.
الأداء: يقوم PrefixAgent بتخليق جامعات بادئة بمساحات أصغر من الطرق المرجعية (بما في ذلك DP، وMCTS، وRL، وVAE، وPrefixGPT) في جميع التكوينات تقريبًا. وتتسع ميزة المساحة بشكل ملحوظ عند عرض البت الأكبر (على سبيل المثال، تقليل بنسبة 11.3% عند 64 بت مقارنة بأفضل طريقة مرجعية).
التعميم: أظهر إطار العمل قدرة قوية على التعميم، حيث حقق نتائج متفوقة على عروض البت غير المرئية (مثل 54 بت) وملفات تعريف الوصول غير المنتظمة التي لم تكن موجودة في مجموعة التدريب.
الكفاءة: يحقق PrefixAgent تسريعًا يزيد عن 10 أضعاف مقارنة بالنماذج المرجعية القائمة على البحث (MCTS، وRL، وVAE) لأنه يتجنب إعادة التدريب لكل سيناريو جديد ويقوم بعمليات تحسين مستهدفة موجهة بواسطة LRM.
التدفق التجاري: عند تقييمه تحت تدفق تصميم فيزيائي تجاري (تقنية 32nm)، حافظت الجامعات المولدة بواسطة PrefixAgent على مساحات أصغر مقارنة بأدوات التخليق التجارية.
دراسات الاستئصال (Ablation Studies): تؤكد النتائج ضرورة تحسين المرحلة الثانية، وتفوق تمثيل EPR (Enhanced Prefix Representation) على ترميزات الشبكة/قائمة التجاور من أجل استدلال النموذج اللغوي الكبير، والدور الحاسم لاستدلال سلسلة الأفك (CoT) في تحقيق نتائج عالية الجودة.
الأهمية يزعم البحث أن PrefixAgent يمثل خطوة كبيرة للأمام في أتمتة تصميم الدوائر الحسابية. ومن خلال تفكيك مهمة التصميم المعقدة إلى مشكلات فرعية يمكن إدارتها والاستفادة من e-graphs لتوليد بيانات عالية الجودة، فإنه يتغلب على قيود القابلية للتوسع والتعميم التي واجهت طرق تعلم الآلة السابقة. يوضح إطار العمل أن النماذج اللغوية الكبيرة، عندما يتم توجيهها بشكل صحيح عبر التكامل مع الأدوات وضبطها بدقة بناءً على مسارات استدلال قابلة للتفسير، يمكنها بفعالية التنقل في مساحات التصميم المعقدة للتفوق على كل من الاستدلالات التقليدية وأدوات EDA التجارية، لا سيما للتصاميم واسعة النطاق. ويقترح المؤلفون إمكانية توسيع نموذج توليد البيانات واستراتيجية التفكيك هذه لتشمل مهام تحسين مسارات البيانات الأخرى، مثل تحسين المضروبات (multiplier optimization).