MadVfold: accelerating NLO event generation and reducing negative weights with SIMD vectorization and GPUs

في عالم فيزياء الجسيمات عالي المخاطر، يقوم العلماء في مصادم الهادرون الكبير بصدم البروتونات مع بعضها البعض لإعادة تهيئة ظروف الكون المبكر. ولتفسير الحطام الناتج، يعتمدون على محاكاة حاسوبية تتنبأ بما يجب أن يحدث أثناء هذه التصادمات. ومع ذلك، فإن حساب هذه التنبؤات بدقة متناهية هو مهمة هائلة؛ إذ إن الطرق الأكثر دقة، والمعروفة باسم حسابات "الرتبة التالية للرتبة القياسية" (next-to-leading-order)، بطيئة للغاية ومكلفة حاسوبياً. ويتمثل أحد العوائق الرئيسية في هذه المحاكاة في ظهور "الأوزان السالبة"، وهي مفارقة رياضية حيث تحمل بعض الأحداث المحاكات قيمة سالبة. ولإلغاء الضجيج الذي تخلقه هذه القيم السالبة وتحقيق صورة واضحة، يتعين على الباحثين توليد دفعات أكبر بكثير من الأحداث المعتادة، مما يستهلك كميات هائلة من القدرة الحسابية والوقت.

ولحل هذه المشكلة، طور الفيزيائيون تقنية تسمى "الطي" (folding). تخيل أنك تحاول تقدير قيمة تكامل معقد عن طريق أخذ عينة لنقطة واحدة؛ قد تكون النتيجة مليئة بالضجيج وغير دقيقة. أما "الطي" فهو يشبه حساب نفس الدالة المعقدة لعدة متغيرات مختلفة من المتغيرات الكينماتيكية لحدث واحد ثم حساب متوسط النتائج. يوفر هذا النهج تقديراً أكثر استقراراً للتكامل، مما يساعد في تنعيم الضجيج الإحصائي ويقلل من الحاجة إلى الأوزان السالبة. ورغم فعالية هذه الطريقة، إلا أنها بطيئة جداً لأنها تتطلب من الحاسوب إجراء نفس الحساب المعقد مرات عديدة لحدث واحد. ويهدف نهج جديد وصفه أندريا فالاسي في سيرن (CERN) إلى تسريع هذه العملية من خلال استخدام أجهزة الكمبيوتر الحديثة لأداء هذه الحسابات المتكررة بشكل متزامن، بدلاً من أدائها واحداً تلو الآخر.

يقدم عمل فالاسي طريقة تسمى "الطي المتجه" (vectorized folding)، والتي تستفيد من قدرة المعالجة المتوازية لوحدات معالجة الرسومات الحديثة والمعالجات المركزية المتقدمة. فبدلاً من أن يطلب من الكمبيوتر حساب الفيزياء لمتغير واحد من متغيرات الحدث، ثم الذي يليه، وهكذا، يقوم البرنامج الجديد بتجميع عشرات المتغيرات معاً، ثم يرسل هذه الدفعة بأكملها إلى معالج الكمبيوتر، والذي يقوم بحسابها جميعاً في وقت واحد. ويمثل هذا تحولاً كبيراً عن الطريقة التقليدية التي تعمل بها هذه المحاكاة، والتي تعالج الأحداث بشكل تسلسلي. ومن خلال إعادة تنظيم البرمجيات للاستفادة من هذه القدرة على التوازي، ابتكر الباحث أداة جديدة تسمى MADVFOLD، صُممت للعمل مع حزمة محاكاة Madgraph5_aMC@NLO واسعة الانتشار.

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

كما استكشف البحث ما إذا كان يمكن تطبيق هذا النهج المتوازي على توليد الأحداث دون استخدام تقنية "الطي" على الإطلاق. فحتى في سيناريو "عدم الطي" هذا، حيث يكون الهدف ببساطة هو تسريع المحاكاة القياسية، حقق البرنامج الجديد زيادة في السرعة بمقدار ثلاثة أضعاف. وهذا يشير إلى أن التغييرات الهيكلية التي أُجريت لدعم عملية "الطي" لها فوائد أوسع، مما يجعل خط أنابيب المحاكاة بأكمله أكثر كفاءة. وقد تم تطوير العمل باستخدام عملية مبتكرة اعتمد فيها الباحث بشكل كبير على نماذج اللغات الكبيرة للمساعدة في كتابة واختبار الكود البرمجي، وهي طريقة سمحت للمشروع بالانتقال من مجرد مفهوم إلى برنامج يعمل في حوالي ستة أساباف فقط.

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