Full event interpretation with machine-learning-based particle-flow reconstruction in the CMS detector
تقدم هذه الورقة خوارزمية تدفق الجسيمات القائمة على تعلم الآلة (MLPF) لكاشف CMS، والتي تستبدل إعادة البناء التقليدية القائمة على الاستدلال بنموذج موحد مسرع بواسطة وحدة معالجة الرسومات، محققةً أداءً فيزيائياً يضاهي الطرق القياسية مع تحسين دقة طاقة النفاثات بشكل كبير وتقليل وقت الاستدلال.
في قلب أوروبا، تحت الحدود الفاصلة بين فرنسا وسويسرا، تكمن آلة بُنيت للإجابة على أكثر الأسئلة جوهرية حول الكون. يقوم مصادم الهادرونات الكبير بصدم البروتونات مع بعضها البعض بسرعة تقارب سرعة الضوء، مما يخلق رذاذًا فوضويًا من الجسيمات الجديدة التي تتلاشى في جزء من الثانية. ولفهم ما حدث، يجب على العلماء إعادة بناء الحدث بأكمله من الإشارات الضئيلة المتبقية في كاشف ضخم يسمى "سي إم إس" (CMS). تشبه هذه المهمة محاولة تحديد كل شخص في ملعب مزدحم عبر رؤية آثار أقدامهم على العشب والظلال التي خلفوها فقط، وكل ذلك بينما يتحرك آلاف الأشخاص الآخرون في نفس الوقت. تعتمد الطريقة الحالية للقيام بذلك على مجموعة من القواعد الصارمة والمكتوبة يدويًا والتي صقلها العلماء على مدى عقود. تخبر هذه القواعد الحاسوب كيفية ربط مسار تركه جسيم مشحون بانفجار طاقة في مقياس السعرات الحرارية، مما يبني فعليًا قائمة بكل جسيم نتج عن التصادم. ومع ذلك، مع زيادة قوة المصادم وزيادة كثافة التصادمات، بدأت هذه الأنظمة القائمة على القواعد تعاني من الحجم الهائل والتعقيد الشديد للبيانات.
قام باحث في تجربة "سي إم إس" الآن بتقديم نهج مختلف، نهج يستبدل هذه القواعد الجامدة بنظام يتعلم من الخبرة. فقد طور خوارزمية جديدة تعتمد على التعلم الآلي، وهو شكل من أشكال الذكاء الاصطناعي الذي يتحسن من خلال دراسة الأمثلة بدلاً من اتباع دليل تعليمات ثابت. وبدلاً من إخبار الحاسوب بالضبط كيفية ربط مسار بترسب طاقة، عرض عليه ملايين التصادمات المحاكات وترك له المجال لاكتشاف الأنماط بنفسه. هذا النظام الجديد، الذي يسم old بـ "خوارزمية تدفق الجسيمات القائمة على التعلم الآلي"، يعامل حدث التصادم بأكمله كأحجية واحدة مترابطة. فهو ينظر إلى جميع المسارات وترسبات الطاقة في آن واحد، ويتعلم التعرف على الإشارات التي تنتمي لكل جسيم من خلال إيجاد علاقات دقيقة قد تغفل عنها القواعد التي وضعها البشر. لقد درب الباحث هذا النظام على عمليات محاكاة حاسوبية مفصلة لتصادمات البروتونات، ثم اختبره على بيانات حقيقية جمعها الكاشف.
تظهر النتائج أن هذا النهج القائم على التعلم يعمل بشكل جيد للغاية. ففي البيئات المحاكات التي تحاكي الظروف الحالية للمصادم، أعادت الخوارزمية الجديدة بناء طاقة نفاثات الجسيمات بدقة أفضل بكثير من الطريقة القياسية. وبالنسبة للجسيمات ذات الطاقة المتوسطة، كان التحسن في الدقة يتراوح بين عشرة وعشرين بالمائة، مما يعني أن العلماء يمكنهم الآن قياس خصائص هذه الجسيمات بوضوح أكبر. وبعيدًا عن الدقة، يقدم النظام الجديد دفعة هائلة في السرعة؛ فبينما تستغرق الطريقة التقليدية حوالي مئة وعشرة مللي ثانية لمعالجة حدث واحد على معالج حاسوبي قياسي، تنجز الخوارزمية الجديدة المهمة نفسها في عشرين مللي ثانية فقط عند تشغيلها على معالج رسومات حديث. وتعد هذه السرعة أمرًا بالغ الأهمية لأن المصادم من المقرر أن يصبح أكثر كثافة في السنوات القادمة، حيث سينتج تصادمات أكثر بكثير في الثانية الواحدة مما ينتجه اليوم. ستواجه الطرق الحالية القائمة على القواعد صعوبة في مواكبة هذا الحجم المتزايد، لكن نظام التعلم الآلي يتوسع بكفاءة، ويتعامل مع التعقيد المتزايد دون تباطؤ.
وعندما طبق الباحث هذه الخوارمة الجديدة على بيانات حقيقية من المصادم، كانت النتائج واعدة بنفس القدر. فقد طابقت توزيعات طاقات الجسيمات وأشكال الأحداث المعاد بناؤها ما شوهد باستخدام الطريقة القياسية، مما يؤكد أن نموذج التعلم الآلي ليس مجرد خدعة محاكاة بل هو أداة تعمل على بيانات فيزيائية فعلية. نجح النظام في تحديد الجسيمات المتعادلة، وهي جسيمات يصعب تتبعها بشكل ملحوظ لأنها لا تترك أثر شحنة، بكفاءة أعلى من ذي قبل، مع الحفاظ على نفس المعدل المنخفض من الإنذارات الكاذبة. إن هذه القدرة على رؤية "غير المرئي" بوضوح أكبر، جنباً إلى جنب مع القدرة على معالجة البيانات بشكل أسرع بكثير، تشير إلى أن هذا النهج يمكن أن يصبح المعيار للتجارب المستقبلية. ويؤكد الباحث أنه بينما يعتمد النجاح الحالي على بيانات "التشغيل 3"، فإن الهدف النهائي هو نشر هذه التكنولوجيا في ترقية "السطوع العالي" للمصادم، حيث سيكون عدد التصادمات المتزامنة أعلى بكثير. ومن خلال إثبات أن الآلة يمكنها تعلم إعادة بناء العالم دون الذري بفعالية أكبر من مجموعة من القواعد التي كتبها البشر، يفتح هذا العمل الباب أمام عصر جديد من الاكتشاف، حيث يتم مواجهة تعقيد الكون بنظام قادر على التعلم منه مباشرة.
ملخص تقني: تفسير الحدث الكامل باستخدام إعادة بناء تدفق الجسيمات القائم على التعلم الآلي في كاشف CMS
بيان المشكلة تعد خوارزمية تدفق الجسيمات (Particle-Flow - PF) حجر الزاوية في عملية إعادة البناء لـ تجربة CMS في مصادم الهادرونات الكبير (LHC) التابع لـ CERN، حيث تهدف إلى توفير وصف عالمي للجسيمات على مستوى التصادم من خلال دمج المعلومات من المتتبع (tracker)، والمك kalorimeters (المسعرات)، وأنظمة الميونات. تعتمد التنفيذات الحالية لـ PF المستخدمة في الإنتاج الفعلي على استدلالات (heuristics) مدفوعة بالفيزياء وطرق تكرارية قائمة على القواعد (مثل الربط القائم على القرب) لربط ضربات الكاشف بالجسيمات النهائية. تواجه هذه النهج التقليدية تحديات كبيرة:
القابلية للتوسع الحسابي: التعقيد التوافقي لربط الضربات في البيئات ذات الإشغال العالي يتوسع بشكل سيئ (غالبًا O(N3) أو يتطلب انحيازات استقرائية قوية لتقليلها إلى O(NlogN))، مما يحد من الكفاءة مع زيادة تعقيد الأحداث.
استغلال الأجهزة: خوارزميات PF الحالية ليست محسنة بشكل أصيل لوحدات معالجة الرسومات (GPUs) الحديثة، والتي تعد ضرورية للتعامل مع معدلات البيانات المتزايدة وظروف التراكم (pileup) المتوقعة في مرحلة مصادم الهادرونات الكبير عالي السطوع (HL-LHC).
المرونة: تجد الخوارزميات التقليدية صعوبة في دمج المعلومات الخاصة بالتحليل أو انحيازات الاختيار مباشرة في سلسلة إعادة البناء دون الحاجة إلى إعادة ضبط واسعة النطاق.
مع توقع إنتاج HL-LHC لمتوسط 200 تفاعل تراكم (pileup) لكل حدث، هناك حاجة ماسة لتقنيات إعادة بناء توفر كلاً من تحسين الأداء الفيزيائي والقابلية للتوسع الحسابي.
المنهجية تقدم الورقة تنفيذ خوارزمية إعادة بناء تدفق الجسيمات القائمة على التعلم الآلي (MLPF) ضمن إطار عمل برمجيات CMS. يستبدل هذا النهج خطوات إعادة بناء متعددة ونمطية بنموذج موحد واحد يتم تدريبه من البداية إلى النهاية (end-to-end) على بيانات محاكاة.
البنية: تستخدم خوارزمية MLPF بنية قائمة على المحولات (Transformer) توظف آليات الانتباه الذاتي (self-attention). وعلى عكس مقاربات الشبكات العصبية الرسومية (GNN) التي تم اختبارها سابقًا في CMS، يستخدم هذا النموذج طبقات Transformer يتم تسريعها بواسطة نواة FLASHATTENTION للتخفيف من تكالين الذاكرة والحوسبة التربيعية المرتبطة بآليات الانتباه القياسية.
المدخلات والمخرجات: يأخذ النموذج متجهات سمات المسارات (tracks) وتجمعات الكالوريمتر (clusters) كمدخلات. ويتنبأ في آن واحد بقائمة من مرشحي الجسيمات النهائية، مُخرجًا تحديد هوية الجسيم (PID) والزخم الرباعي (pT,η,ϕ,E).
هدف التدريب: تم اعتماد نهج تعلم تحت الإشراف. تم تحديد هدف التدريب كعينة فرعية من الجسيمات المستقرة من المولد (PYTHIA) التي تترك إشارات قابلة للكشف في الكاشف (إما مباشرة أو عبر نواتج اضمحلالها). تم بناء هذا الهدف بعناً ليكون ذا معنى فيزيائي، مع استبعاد الجسيمات خارج نطاق قبول الكاشف مع مراعاة عدم كفاءة إعادة البناء (مثل معاملة الهادرونات المشحونة غير المعاد بناؤها كأنها هادرونات متعادلة في الهدف).
دالة الخسارة: يتم تدريب النموذج باستخدام دالة خسارة لكل جسيم تتكون من ثلاثة عناصر:
التصنيف الثنائي (Lcls−binary): يحدد ما إذا كان العنصر المدخل (مسار/تجمع) هو المصدر الأساسي للجسيم المستهدف.
تصنيف الهوية متعدد الفئات (Lcls−PID): يصنف نوع الجسيم (فوتون، إلكترون، ميون، هادرون مشحون، هادرون متعادل) باستخدام "خسارة التركيز" (focal loss) للتعامل مع عدم توازن الفئات.
الانحدار (Lreg): يتنبأ بمكونات الزخم الرباعي. يتم حساب الخسارة باستخدام متغيرات محولة (نسب لوغاريتمية للزخم/الطاقة المستهدفة إلى المدخل الأساسي) لتحسين الاستقرار العددي، مع ترجيح يعتمد على pT للتركيز على الجسيمات عالية الطاقة.
التكامل: يتم تصدير النموذج كملف ONNX ثابت ودمجه في إطار عمل إعادة بناء CMS الخارجي (offline). ويستخدم واجهة ONXRUNTIME لأداء الاستدلال على وحدات معالجة الرسومات (GPUs)، مستبدلاً وحدة PF القياسية القائمة على المعالج المركزي (CPU) مع ترك وحدات التتبع (tracking) وما بعد التتبع (jet clustering) دون تغيير.
المساهمات الرئيسية
أول عملية تحقق من البيانات لإعادة البناء بالتعلم الآلي: يقدم هذا العمل أول تطبيق لمسار إعادة بناء كامل للحدث قائم على التعلم الآلي للبيانات التي جمعها تجربة مصادم هادروني (CMS Run 3, 2023–2024).
نموذج موحد: تدمج خوارزمية MLPF خطوات إعادة بناء متعددة في نموذج واحد قائم على المحولات، مما يلغي الحاجة إلى الاستدلالات المصممة يدويًا لربط المسارات والتجمعات.
تسريع GPU: من خلال الاستفادة من FLASHATTENTION وبنيات المحولات، تحقق الخوارزمية توسعًا فعالًا على وحدات معالجة الرسومات، وهي قدرة كان من الصعب تحقيقها سابقًا لإعادة بناء PF للحدث الكامل بسبب الطبيعة التكرارية للخوارزميات القياسية.
القدرة على التعميم: تظهر الخوارذج قدرة على التعميم عبر ظروف كاشف مختلفة (Run 3 2023 مقابل 2024) وطاقات تصادم مختلفة (14 TeV للتدريب مقابل 13.6 TeV للاختبار)، مما يشير إلى المتانة ضد التباينات خارج التوزيع.
النتائج تم تقييم أداء MLPF في المحاكاة (باستخدام عينات QCD multijet و top quark-antiquark) وعلى عينة تجريبية صغيرة من بيانات Run 3.
الأداء الفيزيائي:
دقة طاقة النفاثة (Jet Energy Resolution): في أحداث ttˉ المحاكات تحت ظروف Run 3، تحقق MLPF تحسنًا بنسبة 10–20% في دقة طاقة النفاثة للنفثات التي تتراوح قيم زخمها المستعرض بين 30–100 GeV مقارنة بـ PF القياسي. يُلاحظ هذا التحسن باستمرار عبر مختلف العمليات الفيزيائية.
كفاءة الجسيمات: تظهر MLPF كفاءة محسنة للهادرونات المتعادلة مع الحفاظ على نفس معدل الخطأ في تحديد الهوية مقارنة بـ PF القياسي. بالنسبة للهادرونات المشحونة والفوتونات، تكون الكفاءة أعلى قليلاً مع زيادة طفيفة في معدل الخطأ في تحديد الهوية عند قيم pT المنخفضة.
الزخم المستعرض المفقود (pTmiss): توزيعات pTmiss التي تعيد بناءها MLPF متسقة عمومًا مع PF القياسي، رغم أن MLPF تميل إلى إعادة بناء طيف أكثر صلابة (harder spectrum) في الذيول.
التحقق من البيانات: في بيانات dijet من Run 3، تتوافق توزيعات pT للنفثات المعاد بناؤها، وعدم التماثل في dijet، و pTmiss من MLPF مع تلك الناتجة عن PF القياسي، مما يؤكد قدرة النموذج على إنتاج مخرجات فيزيائية واقعية في البيانات الحقيقية.
الأداء الحسابي:
وقت الاستدلال: على وحدة Nvidia L4 GPU، يبلغ وسيط وقت الاستدلال لـ MLPF حوالي 20 مللي ثانية لكل حدث، مقارنة بحوالي 110 مللي ثانية لـ PF القياسي الذي يعمل على المعالجات المركزية (CPUs).
الإنتاجية (Throughput): دعمت وحدة L4 GPU واحدة 64 مسار استدلال متوازي، مما أظهر إمكانات إنتاجية كبيرة.
التوسع: تظهر MLPF توسعًا محسّنًا في وقت التشغيل مع زيادة تعقيد الحدث (pileup) مقارنة بـ PF القياسي، ويُعزى ذلك إلى التعامل الفعال مع المدخلات متغيرة الحجم بواسطة بنية المحولات و FLASHATTENTION.
الأهمية تزعم الورقة أن خوارزمية MLPF نجحت في إثبات أن التعلم الآلي يمكن أن يحل محل إعادة بناء PF التقليدية القائمة على الاستدلال مع الحفاظ على أو تحسين الأداء الفيزيائي وتعزيز الكفاءة الحسابية بشكل كبير. تكمن الأهمية الأساسية في:
إثبات المفهوم لـ HL-LHC: يضع هذا العمل مسارًا قابلاً للتطبيق لنشر إعادة البناء القائمة على التعلم الآلي في بيئة التراكم العالي لـ HL-LHC، حيث ستكون الموارد الحسابية عاملًا محددًا.
مسار البرمجة التفاضلية: من خلال استبدال الاستدلالات الجامدة بنموذج قابل للتعلم، يمهد هذا النهج الطريق لمسارات تحسين الكاشف المستقبلية باستخدام البرمجة التفاضلية، حيث يمكن تحسين خوارزميات إعادة البناء بشكل مشترك مع معلمات الكاشف.
المرونة: يسم ت إطار عمل التعلم الآلي إمكانية دمج معلومات خاصة بالتحليل مباشرة في إجراء إعادة البناء من خلال الضبط الدقيق (fine-tuning)، وهي مرونة يصعب تحقيقها مع خوارزميات PF القياسية.
يشير المؤلف إلى أنه بينما يقتصر التنفيذ الحالي على ظروف Run 3، فإن المنهجية مصممة للتوسع إلى عصر HL-LHC، بانتظار إعادة التدريب على محاكاة الكاشف المحدثة ومزيد من التحسين لنواة الاستدلال لمختلف بنيات الأجهزة.