An End-to-End Differentiable Forward Model for High-Energy Diffraction Microscopy

تخيل أنك تحاول اكتشاف الشكل الدقيق، والموقع، والإجهاد الداخلي لملايين البلورات الصغيرة غير المرئية داخل قطعة من المعدن، فقط من خلال النظر إلى نمط الضوء الذي تشتته عندما تصطدم بها حزمة أشعة سينية فائقة السطوع. هذه هي مهمة مجهر حيود الطاقة العالية (HEDM). فكر في الأمر كأنه لعبة "أين والدو؟" (Where's Waldo?) كونية، ولكن بدلاً من شخصية كرتونية، أنت تطارد الموقع الدقيق والتوجه الخاص بكل حبة فردية في عينة معدنية، باستخدام الظلال والبقع التي تلقيها على شاشة كاشف ضخمة. لعقود من الزمن، استخدم العلماء أجهزة كمبيوتر قوية لمحاكاة كيفية تشتيت هذه البلورات للضوء كما ينبغي لها أن تفعل، ثم قارنوا تلك المحاكاة بالبيانات الحقيقية لتخمين الإجابة. لكن كانت هناك عقبة: الكود البرمجي المستخدم لإنشاء هذه المحاكاة كان يشبه "الصندوق الأسود". يمكنك إدخال أرقام والحصول على صورة، ولكن إذا أردت تعديل الأرقام لجعل الصورة تتطابق بشكل أفضل، لم يكن بإمكان الكمبيوتر إخبارك بأي اتجاه يجب عليك تدوير المقابض. كان عليه أن يخمن بعمى، مجرباً إعداداً تلو الآخر، مما جعل العملية بطيئة وغالباً ما تنتهي في طريق مسدود.

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

الاكتشاف الجوهري للورقة البحثية

تقدم الورقة إطار عمل برمجياً جديداً يسمى midas-diffract يعمل كمترجم عالمي لمجهر حيود الطاقة العالية. أنشأ المؤلفون برنامجاً حاسوبياً واحداً موحداً مكتوباً بلغة PyTorch (وهي أداة شهيرة للذكاء الاصطناعي) يمكنه محاكاة ثلاث طرق مختلفة لرؤية البلورات: المجال البعيد (حيث يكون الكاشف بعيداً)، والمجال القريب (حيث يكون قريباً)، والتركيز النقطي (حيث تكون الحزمة صغيرة جداً).

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

ما أثبتوه وما رفضوه

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

لقد أثبتوا ذلك من خلال مقارنة نموذج PyTorch الجديد مع محاكيات المرجع "MIDAS" الراسخة. كانت النتائج دقيقة للغاية:

• المجال البعيد: طابقوا 162 من أصل 162 بقعة بدقة مستوى البكسل.
• المجال القريب: طابقوا 2,304 من أصل 2,304 بكسل تماماً، حتى عندما كان الكاشف مائلاً.
• التركيز النقطي: طابقوا 1,088 من أصل 1,096 بقعة، حيث حدثت حالات الإخفاق القليلة فقط في الحالات الرياضية المعقدة للغاية بالقرب من مركز الحزمة.

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

الاختبارات الواقعية والحدود

لإظهار مدى فعاليته في العالم الحقيقي، أجرى الفريق ثلاثة اختبارات رئيسية:

1. اختبار "الاضطراب والاستعادة": أخذوا مجموعة بيانات حقيقية لـ 214 حبة من عينة تيتانيوم، وقم بتغيير مواقعها الأولية بمقدار 1.5 درجة، وتركوا برنامجهم الجديد يصلحها. لقد استعادوا 100% من الحبات، ووجدوا التوجه والموقع الصحيح بدقة تبلغ حوالي 6 نانومتر (وهذا أصغر من الفيروس!).
2. اختبار "التحسين المشترك": أظهروا أن البرنامج يمكنه إصلاح هندسة الكاميرا (مثل ميلها ومسافتها) في نفس الوقت الذي يصلح فيه مواقع البلورات. وفي اختبار اصطناعي يحتوي على أربع لوحات كاشف، قاموا بتصحيح محور دوران الكاميرا بدقة تصل إلى 26 ميكروراديان، وهو مستوى من الدقة كان من المستحيل تحقيقه في خطوة واحدة سابقاً.
3. اختبار "حوض التقارب": تحققوا من مدى بعد التخمين الأولي عن الحقيقة قبل أن يفقد البرنامج مساره. وجدوا أنه طالما كان التخمين الأولي ضمن حدود 10 إلى 12 درجة من الحقيقة، فإن البرنامج سينجح في إيجاد الإجابة الصحيحة. وهذا خبر رائع لأن الأدوات القياسية عادة ما توصلك إلى حدود 5 درجات، مما يعني أن هذه الأداة الجديدة جاهزة للاستخدام الفوري.

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

لماذا يهم هذا الأمر؟

هذه الورقة لا تقدم مجرد آلة حاسبة أسرع؛ بل تفتح الباب لفئة جديدة تماماً من التجارب. نظرًا لأن النموذج قابل للتفاضل، يمكن للعلماء الآن:

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

باخت مختص، بنى المؤلفون جسراً بين فيزياء الماضي الصارمة والمثبتة، وبين أدوات التحسين المرنة والقوية للمستقبل. لم يقوموا فقط بتحسين السرعة؛ بل غيروا قواعد اللعبة، مما سمح للعلماء بحل مشكلات كانت معقدة للغاية بحيث لا يمكن معالجتها سابقاً.