High-Dimensional Bayesian Calibration of Expensive Nuclear Models with Differentiable Emulation

ملخص تقني: المعايرة البايزية عالية الأبعاد للنماذج النووية المكلفة باستخدام المحاكاة القابلة للتفاضل

بيان المشكلة
كانت المعايرة البايزية الكاملة للنماذج النووية المكلفة غير عملية تاريخياً بسبب عائقين متراكمين. أولاً، تتطلب تقييمات دالة الإمكان (likelihood) حلولاً ميكانيكية كمومية مكلفة (تستغرق غالباً دقائق لكل تقييم). ثياً، العمليات المعتمدة على المعلمات داخل هذه الحلّالين عادة ما تكون مُنفذة في أكواد قديمة (مثل فورتران - Fortran) لا توفر أي معلومات عن التدرج (gradient). وبدون تدرجات دالة الإمكان الدقيقة، تضطر العينات إلى الاعتماد على طرق خالية من التدرج (مثل عينات المجموعات ذات التباين الأفقي)، والتي يتدهور أداؤها بشكل حاد في التوزيعات البعدية عالية الأبعاد والمرتبطة ببعضها. هذا يجبر الطرق الخالية من التدرج على طلب حوالي $10^5$ تقييم لاستكشاف فضاء المعلمات، مما يجعل الاستدلال البايزي الكامل لعمليات معقدة مثل القنوات المقترنة أو أنظمة الأجسام المتعددة أمراً مستحيلاً من الناحية الحسابية.

المنهجية: إطار عمل DREAM
يقدم البحث إطار عمل DREAM (المحاكي ذو القاعدة المختزلة القابل للتفاضل مع تدرجات تلقائية للاستدلال بسلاسل ماركوف)، وهو استراتيجية مصممة لتمكين الاستدلال البايزي القائم على التدرج للعمليات القديمة غير الأفينية (non-affine). يعمل الإطار عبر مرحلتين متمايزتين:

1. المرحلة غير المتصلة (Pre-computation):

   • يتم التعامل مع العملية المعتمدة على المعلمات كصندوق أسود. يتم أخذ عينات منها على شبكة كثيفة من المعلمات باستخدام الكود القديم.
   • يتم ضغط مجموعة "اللقطات" (snapshots) الناتجة (وهي تمثيلات مصفوفية للعملية) باستخدام تحليل القيم المفردة (SVD).
   • من الضروري ملاحظة أن الضغط يتم تطبيقُه على العملية نفسها، وليس فقط على الملاحظة البديلة. بالنسبة لتطبيق محدد (القنوات المقترنة المتقطعة المستمرة - CDCC)، يتم تفكيك الجهد البصري المختزل إلى جزء مستقل عن المعلمات ومجموع من عوامل الشكل (form factors) مضروبة في معاملات العمق الخطية. يتم التقاط الاعتماد الهندسي غير الخطي في مصفوفات عوامل الشكل، والتي تُضغط إلى قاعدة منخفضة الرتبة (أنماط POD) ومعاملات توسع قياسية.

2. المرحلة المتصلة (Inference):

   • لا يتم استدعاء الحل القديم أبداً أثناء الاستدلال. بدلاً من ذلك، يقوم محرك قابل للتفاضل (مُنفذ باستخدام JAX) بإعادة بناء العملية لأي متجه معلمات مقترح $\Omega$.
   • إعادة البناء: يتم استكمال المعاملات القياسية من الـ SVD (باستخدام الاستكمال الثنائي الخطي) بناءً على معلمات الهندسة المقترحة لإعادة بناء العملية المختزلة $V_{red}(\Omega)$.
   • الحل: يتم حل نظام غاليركن (Galerkin) المختزل باستخدام الجبر الخطي المباشر.
   • التفاضل: يتم تركيب خط الإنتاج بأكمله — من الاستكمال إلى تجميع المصفوفة، والحل الخطي، وتقييم دالة الإمكان — كعمليات قابلة للتفاضل تلقائياً. يتم تطبيق التفاضل التلقائي للنمط العكسي (Reverse-mode AD) عبر الحل الخطي باستخدام التفاضل الضمني. يسمح هذا بحساب تدرج دالة الإمكان الدقيق $\nabla_\Omega \mathcal{L}$ بالنسبة لجميع المعلمات بتكلفة حل أمامي إضافي واحد فقط، بغض النظر عن بُعد المعلمة $N$.
   • أخذ العينات: تمكن التدرجات الدقيقة من استخدام مونت كارلو الهاميلتوني (HMC) وعينة عدم العودة (NUTS)، والتي تتوسع بكفاءة مع الأبعاد، على عكس الخلطات الخالية من التدرج.

المساهمات الرئيسية

• التفاضلية على مستوى العملية: البناء هو على مستوى العملية، ويتطلب فقط أن تكون العملية المعتمدة على المعلمات متغيرة بسلاسة كافية لتسمت بفك تشفير SVD منخفض الرتبة. لا يتطلب ذلك تعديل برنامج الفيزياء الأساسي أو اشتقاق مشتقات تحليلية للكود القديم.
• التدرجات الدقيقة عبر التفاضل الضمني: يوضح البحث كيفية الحصول على تدرجات دالة الإمكان الدقيقة لحل خطي ضمن إطار عمل قابل للتفاضل، مما يتجنب العبء الذاكري للانتشار العكسي الساذج عبر الحلول التكرارية.
• سير عمل متكامل: يجمع DREAM بين محاكاة القاعدة المختزلة (من أجل السرعة) والتفاضل التلقائي (من أجل التدرجات)، مما يخلق سير عمل حيث يمكن قياس دقة المحاكي مباشرة مقابل الكود القديم، وحيث تكون التدرجات دقيقة بالنسبة للمحاكي.

النتائج: تشتت مرن لـ $d + ^{58}\text{Ni}$
تم اختبار الإطار على تحليل CDCC لتشتت الديوتيريوم ($d$) + $^{58}\text{Ni}$ المرن عند طاقة 21.6 MeV، والذي يتضمن 18 معلمة للجهد البصري.

• الأداء: حققت عينات NUTS التقارب على وحدة معالجة رسومات (GPU) واحدة في حوالي 27 دقيقة مع 24,000 عينة بعد مرحلة الإحماء، مع صفر من حالات الانتقال المتباعدة. في المقابل، فشل عينة خالية من التدرج بدأت من نفس النقطة في التقارب بشكل صحيح، حيث أنتجت متوسطات هامشية منحازة (تصل إلى 3.4$\sigma$) بينما فشلت في تفعيل إنذارات التقارب.
• الدقة: وُجد أن متوسط خطأ المحاكي أصغر بأكثر من مرتبة عشرية واحدة من عدم التطابق في النموذج ($\beta \approx 0.076$). وبناءً على ذلك، فإن التوزيع البعدي محكوم بفيزياء نموذج التفاعل بدلاً من التقريب البديل.
• النتائج الفيزيائية: حددت التوزيعات البعدية المعايرة متوسط الامتصاص السطحي للديوتيريوم $\bar{W}_d = 11.7 \pm 0.8$ MeV، وهو أعلى بنحو 40% من قياسات كونينج-ديلاروش (Koning-Delaroche). كما دعمت وجود عدم تماثل كبير بين البروتون والنيوترون في الامتصاص السطحي، رغم أن المؤلفين يشيرون إلى أن هذا استخراج يعتمد على النموذج ويتطلب بيانات متعددة الطاقات للتأكيد. كان عدم تطابق النموذج المستنتج حوالي 8%، وهو أقل بكثير من عدم اليقين التجريبي.

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

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