Scalable parallel 3-D TEM inversion via rational approximation of the matrix exponential

ملخص تقني: عكس ثلاثي الأبعاد متوازي وقابل للتوسع للنماذج الكهرومغناطيسية العابرة (TEM) عبر التقريب العقلاني لأس المصفوفة

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

المنهجية
يقترح المؤلفون إطار عمل جديد للعكس يعمل على فك الارتباط بين التكلفة الحوسبية وعدد أوقات الرصد من خلال إعادة صياغة المشكلة الأمامية باستخدام تأثير "أس المصفوفة" (Matrix Exponential).

1. التقريب العقلاني القريب من الأمثل (RBA) لأس المصفوفة:
   بدلاً من الخطوات الزمنية، يتم التعبير عن الحل العابر $u(t)$ كعملية تأثير لأس المصفوفة $\exp(-t M^{-1}K)$ على متجه أولي. يستخدم المؤلفون تقريباً عقلانياً قريباً من الأمثل (RBA) لهذا الأس، تم بناؤه باستخدام خوارزمية RKFIT. يمثل هذا التقريب الحل كمجموعة خطية من الأنظمة الخطية المزاحة:
   $$u(t_j) \approx 2 \text{Re} \left( \sum_{i=1}^m \alpha_i^{(j)} (K - \xi_i M)^{-1} f \right)$$
   حيث أن $\{ \xi_i \}$ هي أقطاب مشتركة عبر جميع النقاط الزمنية، و $\{ \alpha_i^{(j)} \}$ هي معاملات تم حسابها مسبقاً. ومن الأهمية بمكان أن مصفوفات الأنظمة المزاحة $(K - \xi_i M)$ مستقلة عن الزمن.

2. الحساب الأمامي وحساب الحساسية المتوازي:
   تسمح إعادة الصياغة هذه بحل الأنظمة الخطية المزاحة $m$ بشكل مستقل ومتوازٍ. يعتمد الاعتماد الزمني كلياً على البواقي (المعاملات $\alpha$)، مما يعني أن حساب الاستجابات الأمامية والحساسيات مستقل فعلياً عن عدد أوقات الرصد المرغوبة $K$.

• المحلل الأمامي (Forward Solver): يستخدم محللات مباشرة متوازية (MUMPS) لتحليل وحل الأنظمة المزاحة.
• الحساسيات (الجاكوبي): يتم حساب مصفوفة جاكوبي ضمنياً باستخدام نفس الأنظمة المزاحة. يتم تطبيق مشتقة مصفوفة الكتلة على متجهات الحل المحسوبة مسبقاً $g_i = (K - \xi_i M)^{-1}f$. وهذا يتجنب عمليات التكرار الأمامية/الخلفية المتسلسلة المعتادة في طرق الخطوات الزمنية.

3. إطار عمل العكس:

• التحسين (Optimization): يتم حل مشكلة العكس باستخدام إطار عمل "غاوس-نيوتن". تُحل مشكلة المربعات الصغرى الخطية عبر خوارزمية LSQR، التي تتعامل مع مصفوفة جاكوبي كمشغل ضمني (بدون مصفوفة صريحة).
• التنظيم (Regularization): تُصاغ قيود النعومة باستخدام عناصر رافيارت-توماس (RT0) للتعامل مع بارامترات النموذج ذات القيم الثابتة جزئياً على شبكات غير منتظمة، مما يضمن اتساق تمثيل التدرج.
• التنفيذ: تمت كتابة الكود بلغة جوليا (Julia)، باستخدام إطار العمل Gridap.jl لتجميع العناصر المحدودة وواجهة MPI مخصصة لـ MUMPS لإدارة التحليلات المتوازية على بنيات الذاكرة المشتركة.

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

• تحول في التصميم الخوارزمي: توضح الورقة تحولاً من تكييف طرق التدرج الزمني المتسلسل مع الأجهزة المتوازية، إلى تصميم صياغات رياضية (RBA) تظهر التزامن بطبيعتها.
• فك ارتباط القنوات الزمنية: يتم تحديد التكلفة الحوسبية بعدد الأقطاب $m$ (عادة ما يكون صغيراً، مثلاً 16–21) بدلاً من عدد القنوات الزمنية $K$ (التي غالباً ما تكون بالمئات). يسمح هذا بأخذ عينات زمنية عالية الدقة دون زيادة عبء العمل في الجبر الخطي.
• سير عمل عكس قابل للتوسع: من خلال دمج التوازي على مستوى النموذج الرياضي، يحقق المؤلفون عملية عكس قابلة للتوسع حيث يمكن توزيع التكلفة المهيمنة (حل الأنظمة المزاحة) عبر رتب MPI.
• إصدار مفتوح المصدر: تم إصدار التنفيذ كبرمجيات مفتوحة المصدر لدعم البحث القابل لإعادة الإنتاج في المجتمع الجيوفيزيائي.

النتائج
تم التحقق من صحة الطريقة باستخدام نموذج اصطناعي ثلاثي الأبعاد يحتوي على حوالي 700,000 درجة من الحرية، ويتميز بكتل موصلة ومقاومة في نصف فضاء.

• دقة إعادة البناء: نجحت عملية العكس في استعادة مواقع وأشكال عامة الشذوذات. تم تحديد الكتل الموصلة بدقة جيدة، بينما أظهرت الكتل المقاومة تبايناً أقل، وهو ما يتفق مع القيود المعروفة لحساسية TEM وتنظيم النعومة.
• الأداء:
  • التوسع (Scaling): أظهرت تجارب التوسع القوي أداءً مثالياً تقريباً حتى 8 رتب MPI لكل من التحليل والتعويض الخلفي. انخفضت الكفاءة قليلاً عند 16 رتبة بسبب تأثيرات ذاكرة الوصول غير المتماثل (NUMA) في البنية المشتركة وتنازع عرض نطاق الذاكرة.
  • الذاكرة: كان استهلاك ذروة الذاكرة حوالي 620 جيجابايت، وهو المطلوب لتخزين تحليلات 21 نظاماً مزاحاً في آن واحد.
  • المقارنة: مقارنة بطريقة أويلر الضمنية (التي تتطلب 31 تحليلاً حقيقياً لنفس النافذة الزمنية)، يتطلب نهج RBA حوالي 21 تحليلاً مركباً. وبينما تكون الحسابات المركبة أكثر تكلفة، إلا أن نهج RBA يلغي المسار الحرج للتدرج الزمني المتسلسل، مما يوفر إمكانات فائقة لتقليل وقت الجدار (wall-clock time) في البنيات الموزعة.

الأهمية والادعاءات
يدعي المؤلفون أن هذا العمل يوفر بديلاً قابلاً للتوسع لعكس TEM ثلاثي الأبعاد القياسي، وتحديداً لمعالجة عقبة التكرار الزمني. الميزة الأساسية ليست بالضرورة في تقليل العمل الحسابي التسلسلي، بل في تقليل المسار الحرج المتوازي واستقلال التكلفة عن عدد القنوات الزمنية.

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