Rapid Earthquake-to-Tsunami Waveform Generation via Large-Scale Multi-GPU FFT Convolution Applied to the Cascadia Subduction Zone
تقدم هذه الورقة خط معالجة قابلاً للتوسع ومتعدد وحدات معالجة الرسومات، يستفيد من عملية الالتفاف المسرعة بواسطة تحويل فوريه السريع لدوال "جرين" المحسوبة مسبقاً لتوليد موجات الزلازل وال been تسونامي لمنطقة اندساس كاسكاديا في غضض ميلي ثانية، مما يتيح التقييم السريع لمجموعات التمزق الكبيرة لأنظمة الإنذار المبكر.
المؤلفون الأصليون:Bowen Shi, Sreeram Venkat, Stefan Henneking, Omar Ghattas
عندما يضرب زلزال هائل تحت المحيط، فإن الخطر لا ينتظر توقف الاهتزاز. ففي غضون دقائق، يمكن لقاع البحر المزاح أن يولد جداراً من المياه يندفع نحو السواحل البعيدة، مما يترك وقتاً ضئيلاً للتحذير. ولإنقاذ الأرواح، يحتاج العلماء إلى التنبؤ بدقة بمدى ارتفاع تلك الأمواج وأين ستضرب، ولكن القيام بذلك يتطلب محاكاة الفيزياء المعقدة للأرض والمحيط لآلاف السيناريوهات المحتملة للزلازل. تقليدياً، كانت عملية تشغيل عمليات المحاكاة عالية الدقة هذه لكل احتمال على حدة بطيئة ومكلفة للغاية لدرجة تجعل من المستحيل تقريباً توليد مجموعات البيانات الكبيرة اللازمة للتحذيرات المبكرة الموثوقة. لقد كان التحدي يكمن في إيجاد طريقة لإنتاج توقعات هذه الأمواج الحرجة بسرعة كافية لتكون مفيدة في حالات الطوارئ، دون التضحية بدقة النماذج.
لقد طور باحثون في جامعة تكساس في أوستن طريقة جديدة تحل هذه المشكلة من خلال التعرف على نمط أساسي في كيفية سلوك الزلازل وأمواج التسونامي. فقد اكتشفوا أن القوانين الفيزيائية التي تحكم هذه الأحداث خطية وثابتة زمنياً، مما يعني أن النظام يستجيب للاضطراب بطريقة متسقة ومتوقعة بغض النظر عن وقت حدوث هذا الاضطراب. وبدلاً من حل المعادلات الفيزيائية الصعبة من الصفر لكل سيناريو زلزال جديد، قام الفريق بحساب "بصمات" استجابة الأرض والمحيط مسبقاً. فقد قاموا بحساب كيفية تفاعل الأرض والمياه مع صدمة واحدة معيارية، مما أدى إلى إنشاء مكتبة من هذه الاستجابات. وبمجرد تجهيز هذه البصمات، يمكن للفريق توليد أشكال الموجات لأي زلزال معقد ببساه عملية دمج هذه الاستجابات المحسوبة مسبقاً، وهي عملية مكافئة رياضياً لعملية الالتفاف (convolution). يحول هذا النهج مهمة تتطلب عادةً حل معادلات ضخمة بشكل متكرر إلى عملية أسرع بكثير تعتمد على مطابقة الأنماط وإضافتها.
ولجعل هذه الطريقة تعمل في منطقة كوارث حقيقية، طبقها الباحثون على منطقة "كاسكاديا" (Cascadia Subduction Zone)، وهي خط صدع ضخم قبالة سواحل شمال غرب المحيط الهادئ قادر على إنتاج زلازل مدمرة. لقد بنوا نموذجاً رقمياً لهذه المنطقة يتضمن 963 قسماً متميزاً لخط الصدع، وأكثر من مليوني نقطة في قاع البحر، و64 موقعاً حيث يمكن للمستشعرات قياس الأمواج. كما يتتبع النموذج حركة المياه على مدار 256 ثانية من الوقت المحاكى. ونظراً لأن البيانات المطلوبة لتخزين الاستجابات المحسوبة مسبقاً هائلة — حيث تصل إلى ما يقرب من 9.5 تيرابايت من الذاكرة — لم يتمكن الفريق من تشغيل ذلك على جهاز كمبيوتر واحد. بدلاً من ذلك، قاموا بتوزيع العمل عبر 64 من أقوى معالجات الرسوميات في العالم، مرتبة في عنقود متخصص عالي السرعة. وقد سمح هذا الإعداد بالاحتفاظ بالبيانات الوسيطة الضخمة في شرائح ذاكرة أجهزة الكمبيوتر، ونقل المعلومات مباشرة بين خطوات المعالجة دون التباطؤ لحفظها في القرص الصلب.
النتائج التي حقق هذا النهج مذهلة من حيث السرعة والنطاق. فبمجرد تحميل الاستجابات المحسوبة مسبقاً في النظام، يمكن للمسار توليد شكل موجة التسونامي الكامل لتمزق زلزالي واحد في غضون 24 ميلي ثانية فقط. وهذا أسرع بأكثر من 20 مرة من أفضل الطرق الحالية التي تستخدم البرامج القياسية. ولوضع هذه السرعة في منظور، يمكن للنظام تقييم مجموعة من 10,000 سيناريو زلزالي مختلف في حوالي أربع دقائق، وهي مهمة كانت ستستغرق من الطريقة القديمة قرابة 87 دقيقة. وقد اختبر الباحثون ذلك من خلال محاكاة ثلاثة أحداث زلزالية متميزة تتراوح أقدارها بين 7.95 و9.09 درجة. وفي كل حدث، نجح النظام في رسم خريطة الانزلاق على الصدع وصولاً إلى حركة قاع البحر الناتجة، ومن ثم إلى أمواج التسونامي الواصلة إلى نقاط المراقبة. وتضمن المخرج خرائط تفصيلية لكيفية تحرك قاع البحر وارتفاعات الأمواج المحددة المسجلة في مواقع مختلفة، وكل ذلك تم توليده بشكل فوري تقرياً.
يُظهر هذا العمل أنه من الممكن إنشاء "توأم رقمي" لمنطقة معرضة للتسونامي يكون دقيقاً للغاية وسريعاً بما يكفي للاستخدام في الوقت الفعلي. ومن خلال الاستفادة من الطبيعة المتوقعة لفيزياء الأمواج وقوة الحوسبة الفائقة الحديثة، أثبت الفريق أنه يمكن تقييم مجموعات كبيرة من السيناريوهات الزلزالية في دقائق بدلاً من ساعات. وتفتح هذه القدرة الباب أمام أنظمة إنذار مبكر أكثر تطوراً يمكنها مراعاة الواقع المعقد والفوضوي لكيفية حدما وقوع الزلازل، بدلاً من الاعتماد على تقديرات مبسطة. إن القدرة على توليد أشكال الموجات هذه بسرعة تعني أن المتنبئين سيتمكنون قريباً من الوصول إلى مكتبة واسعة من النتائج المحتملة، مما يسمح لهم بتقديم تحذيرات أكثر دقة وفي الوقت المناسب للمجتمعات الساحلية عندما تبدأ الأرض بالاهتزاز.
ملخص تقني: التوليد السريع لموجات الزلازل والمد والجزر عبر عملية التفاف (Convolution) باستخدام تحويل فوريه السريع (FFT) على نطاق واسع عبر وحدات معالجة رسومية متعددة (Multi-GPU)
بيان المشكلة تعتمد أنظمة الإنذار المبكر القائمة على البيانات للزلازل والمد والجزر على مجموعات ضخمة من سيناريوهات التصدع وما ينتج عنها من أشكال موجية لإجراء الاستدلال البايزي والتنبؤ. ومع ذلك، فإن توليد هذه المجموعات من البيانات عبر عمليات محاكاة عددية عالية الدقة لديناميكيات الزلازل والمد والجزر هو أمر مكلف حاسوبياً. تتطلب النهج التقليدية حل المعادلات التفاضلية الجزئية (PDEs) لعشرات الآلاف من حالات التحقق الزلزالي، وهي عملية بطيئة جداً لتقييم المجموعات الضخمة بسرعة. وتحديداً بالنسبة لمنطقة تصدع كاسكاديا (CSZ)، تواجه سير العمل الحالية صعوبة في توليد مجموعات الأشكال الموجية واسعة النطاق ضمن أطر زمنية عملية دون التضحية بالدقة الفيزيائية.
المنهجية يقترح المؤلفون مسار عمل يستغل بنية النظام الخطي الثابت زمنياً (LTI) المتأصلة في كل من معادلات المرونة الديناميكية (انتشار الموجات الزلزالية) ومعادلات الجاذبية الصوتية (انتشار المد والجزر). بدلاً من حل المعادلات التفاضلية الجزئية لكل حالة تحقق، تقوم الطريقة بحساب استجابات النبضة مسبقاً وتختزل عملية الربط من المصدر إلى الشكل الموجي إلى عمليتي التفاف متتاليتين.
الحساب المسبق لاستجابات النبضة:
دوال غرين المرنة (Elastic Green's Functions): يقوم المعامل الأول بربط دالات معدل انزلاق الصدع بالانزياح الرأسي لقاع البحر. باستخدام طريقة التردد والعدد الموجي (FK)، قام المؤلفون بحساب دوال غرين مسبقاً لـ 963 صدعاً فرعياً مثلث الشكل. تم بناء هذه الدوال "أوفلاين" وتخزينها في نطاق التردد.
استجابات الجاذبية الصوتية المرافقة (Acoustic-Gravity Adjoint Responses): يقوم المعامل الثاني بربط حركة قاع البحر بضغط المياه في عرض البحر وأشكال موجات المد والجزر. وبدلاً من إجراء حلول أمامية لكل نقطة شبكية في قاع البحر (والتي تبلغ Nm≈2.4 مليون)، استخدم المؤلفون صيغة مرافقة (Adjoint formulation). يتيح لهم ذلك حساب الاستجابة عبر إجراء Nq (64) من الحلول المرافقة، بواقع حل واحد لكل موقع رصد، مما يعكس العلاقة بين المدخلات والمخرجات لتقليل التكلفة الحسابية.
صياغة بلوك توبلتز (Block Toeplitz Formulation): تؤدي الديناميكيات المتقطعة زمنياً إلى عمليتي ضرب مصفوفة بمتجه (matvecs) من نوع "بلوك توبلتز" منخفضة المثلث ومتكررة (lower-triangular Toeplitz). تقوم الخريطة الكاملة بتحويل دالات معدل الانزلاق (s) إلى حركة قاع البحر (m) ثم إلى أشكال موجات الرصد (q): q=FqDtFGs حيث تمثل FG و Fq معاملات توبلتز التي تمثل الديناميكيات المرنة وديناميكيات الجاذبية الصوتية على التوالي، بينما تمثل Dt مؤثر الفرق الزمني.
مسار العمل الموزع والمعجل بواسطة FFT: لتقييم عمليات الالتفاف الضخمة هذه بكفاءة، طبق المؤلفون خوارزمية موزعة باستخدام FFTMatvec.
تجزئة النطاق (Domain Decomposition): يتم تقسيم شبكة قاع البحر الكبيرة عبر عدة وحدات معالجة رسومية (GPUs). يتم تقسيم المعامل المرن (FG) حسب صفوف المخرجات (نقاط قاع البحر)، بينما يتم تقسيم المعامل المرفق (Fq) حسب أعمدة المدخلات (نقاط قاع البحر المطابقة).
المعالجة داخل الجهاز (In-Device Processing): يتم إبقاء حقل قاع البحر المتوسط مقيماً داخل وحدات المعالجة الرسومية بين خطوتي الالتفاف، لتجنب اختناق الإدخال/الإخراج الناتج عن كتابة وقراءة الحقل المكاني-الزماني الكامل.
التسريع في نطاق التردد: يتم تقييم عمليات ضرب المصفوفات (matvecs) المحلية من نوع توبلتز باستخدام تحويلات فوريه السريعة (FFTs). يتم تحويل نوى الاستجابة إلى نطاق التردد مرة واحدة أثناء الإعداد وتظل في ذاكرة وحدة المعالجة الرسومية. يتم إجراء الالتفاف كعملية ضرب عنصر بعنصر في نطاق التردد، يتبعها تحويل فوريه عكسي (Inverse FFT).
المساهمات الرئيسية
توليد المصدر إلى الشكل الموجي عبر تعدد وحدات المعالجة الرسومية وتسريع FFT: يقدم البحث مسار عمل يصيغ عملية الربط من الزلزال إلى المد والجزر كعمليتي ضرب مصفوفة (matvecs) من نوع "بلوك توبلتز" متتاليتين يتم تقييمهما عبر تحويلات FFT موزعة. تلغي هذه الطريقة الحاجة لحل المعادلات الموجية لكل حالة تحقق وتحافظ على الحقول المتوسطة داخل الجهاز.
القابلية للتوسع إلى نطاق CSZ الكامل: أثبت المؤلفون فعالية المسار على تكوين كامل لمنطقة تصدع كاسكاديا يتضمن 963 صدعاً فرعياً، و2,416,530 نقطة شبكية لقاع البحر، و64 موقع رصد، و256 خطوة زمنية.
إدارة الذاكرة: نجح التنفيذ في إدارة متطلبات الذاكرة الهائلة لنوى نطاق التردد (حوالي 9.45 تيرابايت من إجمالي ذاكرة وحدات المعالجة الرسومية) من خلال توزيع البيانات عبر 64 وحدة معالجة رسومية ضمن نطاق NVL72 واحد.
النتائج تم تقييم مسار العمل على نظام TACC Vista باستخدام وحدات معالجة الرسوميات NVIDIA GH200 وGB200.
الأداء: على 64 وحدة معالجة رسومية من نوع GB200، يقوم المسار بتوليد الأشكال الموجية لصدع واحد في 24 ميلي ثانية بمجرد استقرار معاملات الاستجابة في الذاكرة.
السرعة: يمثل هذا تسارعاً قدره 21.8 ضعفاً مقارنة بالتنفيذ المرجعي المعتمد على PyTorch والذي يستخدم الالتفاف المباشر في النطاق الزمني.
التوسع (Scaling):
التوسع القوي (Strong Scaling): أدى الانتقال من 8 إلى 64 وحدة معالجة رسومية إلى تحقيق تسارع قدره 6.5× (GH200) و6.8× (GB200)، بكفاءة موازية بلغت 81.2% و85.0% على التوالي.
التوسع الضعيف (Weak Scaling): حافظ النظام على كفاءة عالية (95.1% على GH200 و95.8% على GB200) مع توسع حجم المشكلة من 8 إلى 64 وحدة معالجة رسومية.
توليد المجموعات (Ensemble Generation): عند المعدل المقاس، يتطلب تطبيق المعاملات على مجموعة مكونة من 104 من حالات التصدع حوالي 4 دقائق، مقارنة بـ 87 دقيقة للتنفيذ المرجعي باستخدام PyTorch.
الأهمية يزعم البحث أن مسار العمل هذا يوفر طريقاً عملياً لتوليد مجموعات أشكال موجية واسعة النطاق بناءً على نماذج عالية الدقة. ومن خلال اختزال خريطة المصدر إلى الشكل الموجي إلى عمليات التفاف محسوبة مسبقاً والاستفادة من تسريع FFT الموزع، تمكن هذه الطريقة من تقييم مجموعات التصدع الضخمة في غضض دقائق. هذه القدرة حاسمة للنماذج القائمة على البيانات للمد والجزر، خاصة في مسائل الاستدلال البايزي حيث يتطلب الأمر توليد أشكال موجية اصطناعية سريعة لتقييد بارامترات المصدر والتنبؤ بارتفاعات المد والجزر. يثبت هذا العمل أنه من الممكن تجاوز المعادلات التفاضلية الجزئية المكلفة للحل لكل حالة تحقق مع الحفاظ على الدقة الفيزيائية المطلوبة لأنظمة الإنذار المبكر.