Machine-learned, finite temperature Fermi-operator expansions suitable for GPUs and AI-hardware
تقدم هذه الورقة طريقة توسيع مؤثر فيرمي تكرارية عند درجات حرارة محدودة، معززة بالتعلم الآلي بناءً على مخطط SP2، والتي تقوم برسم خرائط حسابات البنية الإلكترونية إلى بنيات شبكات عصبية عميقة، مما يتيح تسريعاً بمقدار رتبة عشرية على وحدات معالجة الرسومات وأجهزة الذكاء الاصطناعي من خلال استخدام عمليات ضرب المصفوفات بضرب المصفوفات وتحجيم تقاربي لتجنب التقطيع الصريح وإعادة تدريب النموذج.
البحث الأصلي مُهدى إلى الملك العام بموجب CC0 1.0 (http://creativecommons.org/publicdomain/zero/1.0/). هذا شرح مولَّده بالذكاء الاصطناعي للبحث أدناه. لم يكتبه المؤلفون ولم يصادقوا عليه. وللتحقق من الدقة التقنية، يرجى الرجوع إلى البحث الأصلي. اقرأ إخلاء المسؤولية الكامل
الصورة الكبيرة: طريقة أسرع لمحاكاة الذرات
تخيل أنك تحاول التنبؤ بكيفية تحرك وتفاعل حشد من الناس (الإلكترونات) في غرفة (مادة ما). في عالم الفيزياء الكمية، هذا أمر صعب للغاية. للحصول على الإجابة الدقيقة، يتعين عليك عادةً حل لغز ضخم ومعقد يسمى "القطرية" (Diagonalization).
فكر في القطرية كأنك تحاول ترتيب مليون كتاب عن طريق قراءة كل صفحة في كل كتاب للعثور على الترتيب الصحيح. إنها دقيقة، لكنها تستغرق وقتاً طويلاً، خاصة كلما كبر حجم الغرفة.
لقد بنى مؤلفو هذه الورقة البحثية طريقاً مختصراً. فبدلاً من قراءة كل صفحة، أنشأوا "آلة تخمين ذكية" تتعلم كيفية ترتيب الكتب بشكل فوري تقريباً. ويطلقون عليها اسم توسيع مؤثر فيرمي المعتمد على التعلم الآلي (Machine-Learned Fermi-operator expansion).
المشكلة: الحشود الساخنة مقابل الباردة
في الماضي، كانت هذه الطرق المختصرة تعمل بشكل جيد فقط عندما يكون "الحشد" بارداً جداً (درجة حرارة صفر). في الحشد البارد، يقف الجميع ساكنين في خط منتظم يمكن التنبؤ به. الرياضيات هنا بسيطة: إما أنك داخل الخط أو لست داخله.
ومع ذلك، في العالم الحقيقي، غالباً ما تكون الأشياء "ساخنة". عندما تصبح الإلكترونات ساخنة، تصبح مضطربة. بعض الأشخاص الذين كانوا واقفين في الصف قد يخرجون منه، والبعض الآخر الذي كان ينتظر قد يدخل فيه. هذا يخلق حدوداً "ضبابية" حيث يكون الناس جزئياً في الداخل وجزئياً في الخارج.
فشلت الطرق المختصرة السابقة هنا لأنها كانت جامدة للغاية؛ لم تكن قادرة على التعامل مع "الضبابية" الناتجة عن الحشد الساخن.
الحل: تعليم الشبكة العصبية كيفية "الضغط"
أدرك المؤلفون أن الرياضيات المستخدمة لترتيب الحشد البارد تشبه تماماً هيكل الشبكة العصبية العميقة (نوع الذكاء الاصطناعي المستخدم للتعرف على الوجوه أو كتابة القصائد).
- الطريقة القديمة (SP2): تخيل آلة تأخذ رقماً وتقوم إما بتربيعه () أو تقوم بعملية طرح محددة (). تكرر هذه العملية مراراً وتكراراً، "تضغط" الأرقام حتى تصبح إما 0 أو 1. هذا يعمل بشكل رائع للحشود الباردة.
- الطريقة الجديدة (MLSP2): أخذ المؤلفون هذه الآلة ومنحوها "دماغاً". بدلاً من استخدام قواعد ثابتة، قاموا بتدريب الآلة باستخدام التعلم الآلي. لقد علموا الآلة كيفية ضبط "مقابضها" الداخلية (المعاملات) بحيث يمكنها التعامل مع الحشد الساخن "الضبابي" بشكل مثالي.
فكر في الأمر كالتالي:
- الآلة القديمة: ختم جامد يطبع فقط "نعم" أو "لا".
- الآلة الجديدة: طابعة ثلاثية الأبعاد مرنة تتعلم بالضبط كيفية تشكيل الـ "نعم" والـ "لا" لإنشاء منحنى سلس ومثالي بينهما، اعتماداً على مدى سخونة الحشد.
الخدعة السحرية: نموذج واحد يناسب درجات حرارة متعددة
عادةً، إذا قمت بتغيير درجة حرارة المحاكاة الخاصة بك، يتعين عليك إعادة تدريب نموذج الذكاء الاصطناعي الخاص بك من الصفر. وهذا يستغرق وقتاً طويلاً جداً.
اكتشف المؤلفون خدعة ذكية تسمى إعادة القياس الأفيني (Affine Rescaling).
تخيل أن لديك خريطة لمدينة ما. إذا كنت تريد التكبير أو التصغير، فأنت لا تحتاج إلى إعادة رسم المدينة بأكملها؛ بل تحتاج فقط إلى شد أو تقليص الخريطة.
وجد المؤلفون أنه يمكنهم تدريب نموذج الذكاء الاصطناعي الخاص بهم مرة واحدة فقط لمستوى "تكبير" معين (درجة حرارة وجهد كيميائي محدد). بعد ذلك، لأي درجة حرارة أخرى ضمن نطاق معين، يقومون ببساطة بـ "شد" أو "تقليص" البيانات المدخلة (مصفوفة الهاميلتوني) قبل تغذيتها للنموذج. لا يحتاج النموذج إلى إعادة تعلم أي شيء؛ فهو يرى البيانات بمقياس مختلف قليلاً ويعطي الإجابة الصحيحة.
هذا يعني أن بإمكانهم تشغيل عمليات محاكاة حيث تتغير درجة الحرارة باستمرار (مثل التفاعلات الكيميائية) دون الحاجة إلى التوقف لإعادة تدريب الذكاء الاصطناعي.
الأجهزة: استخدام رقائق الذكاء الاصطناعي للعلوم
توضح الورقة البحثية أن هذه الطريقة مبنية خصيصاً للرقائق الحاسوبية الحديثة، وخاصة وحدات معالجة الرسومات (GPUs) وأنوية تنسور (Tensor Cores) (وهي رقائق مصممة للذكاء الاصطناعي).
- التشبيه: القطرية التقليدية تشبه النجار الماهر الذي ينحت كل قطعة أثاث يدوياً. إنها دقيقة ولكنها بطيئة.
- الطريقة الجديدة: هذه تشبه استخدام طابعة ثلاثية الأبعاد عالية السرعة. فهي تستخدم البنية الخاصة برقائق الذكاء الاصطناعي لأداء عمليات ضرب المصفوفات الضخمة بسرعة هائلة.
اختبر المؤلفون هذا على وحدة معالجة رسومات Nvidia RTX 6000 Ada. ووجدوا أن طريقتهم كانت أسرع بـ 9 إلى 16 مرة من الطرق المعيارية المحسنة للغاية التي يستخدمها العلماء اليوم، مع الحفاظ على دقة عالية.
ملخص النتائج
- السرعة: حققوا تسارعاً هائلاً (يصل إلى 16 ضعفاً) في حساب كيفية سلوك الإلكترونات في المواد، خاصة على أجهزة الذكاء الاصطناعي الحديثة.
- الدقة: يمكنهم نمذجة الإلكترونات "الساخنة" (الاحتلال الجزئي) بدقة متناهية، وهو أمر لم تستطع الطرق المختصرة السابقة القيام به بشكل جيد.
- الكفاءة: من خلال تدريب النموذج مرة واحدة واستخدام خدع رياضية لإعادة قياس المدخلات، يتجنبون الحاجة إلى إعادة تدريب النموذج في كل مرة تتغير فيها درجة الحرارة في المحاكاة.
- لا وجود لـ "القطرية السحرية": لقد تجنبوا تماماً الرياضيات الثقيلة والبطيئة للقطرية، معتمدين بدلاً من ذلك على خطوات الضرب المتكررة والسريعة التي تعشقها رقائق الذكاء الاصطناعي.
باختصار، حول المؤلفون عملية رياضية بطيئة وجامدة إلى أداة مرنة وسريعة مدعومة بالذكاء الاصطناعي، تعمل بكفاءة عالية على رقائق الكمبيوتر الحديثة، مما يسمح للعلماء بمحاكاة المواد المعقدة بشكل أسرع بكثير من ذي قبل.
غارق في أبحاث مجالك؟
تصلك نشرة يومية بأحدث الأبحاث المطابقة لكلماتك البحثية المفتاحية — مع ملخصات تقنية، بلغتك.