ProgFormer: Hierarchical Voxel Diffusion Transformer for Longitudinal Brain MRI Prediction
يُعدّ "بروغفورمر" (ProgFormer) نموذج محول انتشار هرمي في فضاء الفوكسل، يعالج تحدي التنبؤ بالتصوير بالرنين المغناطيسي للدماغ عبر دراسات طولية من خلال توظيف بنية ثنائية المسار للحفاظ بشكل مشترك على الاتساق الهيكلي العالمي والتقاط تفاصيل تطور المرض الدقيقة عبر مطابقة التدفق الشرطي، متفوقاً بذلك على الأساليب الرائدة الحالية عبر معايير متعددة.
المؤلفون الأصليون:Dexuan Ding, Yuankai Qi, Luping Zhou, Jian Yang, Quan Z. Sheng, Ming-Hsuan Yang
تخيل أنك مسافر عبر الزمن تمتلك كاميرا خاصة جداً. يمكنك التقاط صورة لدماغ شخص ما اليوم، لكنك تريد معرفة كيف سيبدو بعد خمس أو عشر سنوات من الآن. لا يقتصر هذا على مجرد التخمين؛ بل يتعلق بالتنبؤ بكيفية تغير الدماغ مع تقدم العمر أو مع سيطرة أمراض مثل ألزهايمر ببطء. يسمي العلماء هذا "التنبؤ بالرنين المغناطيسي الطولي". الجزء الصعب هو أن معظم الدماغ يبقى كما هو تماماً من سنة إلى أخرى؛ إنه يشبه منزلاً متيناً لا يتغير كثيراً. لكن المرض يشبه تسريباً صغيراً وبطيئاً في السقف أو شقاً في جدار محدد. إنه يحدث في بقع صغيرة ومحددة للغاية. إذا حاولت التنبؤ بالمستقبل من خلال النظر إلى المنزل بأك-له مرة واحدة، فقد تفوتك تلك الشقوق الصغيرة لأن بقية المنزل تبدو مثالية وغير متغيرة.
لحل هذه المشكلة، حاول الباحثون استخدام حيلتين رئيسيتين. إحداهما هي تقليص صورة الدماغ إلى ملخص صغير وضبابي ("الفضاء الكامن" أو latent space)، ثم إجراء التنبؤ، وبعد ذلك تكبيرها مرة أخرى. المشكلة هي أنه عندما تقلص الصورة ثم تكبّرها، فإنك غالباً ما تفقد التفاصيل الدقيقة، مثل ملمس الجدار أو الشكل الدقيق للشق. الحيلة الأخرى هي النظر إلى كل "بكسل" (أو "فوكسل" في الأبعاد الثلاثية) بشكل مباشر. ولكن إذا حاولت التنبؤ بالدماغ بأك-له وبالشقوق الصغيرة في آن واحد باستخدام أداة بسيطة واحدة، فإن الأداة ستصاب بالارتباك بسبب الأجزاء الضخمة والمستقرة من الدماغ وستتجاهل التغييرات الصغيرة والمهمة.
هنا يأتي دور ProgFormer، وهو نموذج ذكاء اصطناعي جديد مصمم ليكون المصور النهائي المسافر عبر الزمن للدماغ. بدلاً من تقليص الصورة أو استخدام أداة واحدة تناسب الجميع، يستخدم ProgFormer استراتيجية "خطوتين" ذكية للتنبؤ بالمستقبل لصور الرنين المغناطيسي للدماغ. فكر في الأمر كفنان يرسم لوحة بورتريه لشخص يتقدم في العمر. أولاً، يقوم الفنان برسم الوجه والرأس بالكامل لضبط الصورة الكبيرة بشكل صحيح — وهذا هو المسار الخشن (Coarse Pathway). ينظر هذا المسار إلى قطع من الدماغ (تسمى "رقع" أو patches) لفهم الشكل العام وكيف تغير الدماغ بأك-له بمرور الوقت. إنه يشبه رسم الخطوط العريضة للمنزل.
بمجرد رسم الصورة الكبيرة، ينتقل الفنان إلى العدسة المكبرة من أجل المسار الدقيق (Fine Pathway). هذا الجزء يكبر على المناطق المحددة التي قد تحدث فيها "التسريبات" أو الشقوق، مثل الحصين (مركز الذاكرة) أو البطينات (المساحات المملوءة بالسوائل). إنه يستخدم الرسم التخطيطي من الخطوة الأولى كدليل، متسائلاً: "حسناً، أنا أعرف الشكل العام لهذه الغرفة؛ الآن، كيف يجب أن يتحرك هذا الجدار هنا بالضبط؟". ومن خلال الجمع بين هاتين الخطوتين، يستطيع ProgFormer الحفاظ على استقرار الدماغ وواقعيته، مع القدرة في الوقت نفسه على رصد تلك العلامات الدقيقة والحساسة لتطور المرض.
اختبر الباحثون هذا النموذج على ثلاث مجموعات كبيرة من بيانات المرضى الحقيقية (ADNI، وAIBL، وOASIS). ووجدوا أن ProgFormer أفضل في التنبؤ بصور الدماغ المستقبلية مقارنة بالطرق السابقة. فهو ينشئ صوراً أكثر وضوحاً، والأهم من ذلك، أنه يضبط التفاصيل الصغيرة بدقة. وعندما نظروا في مدى نجاح النموذج في التنبؤ بحجم مناطق معينة في الدماغ، وجدوا أنه يرتكب أخطاء أقل من النماذج الرائدة الأخرى. كما أظهرت الدراسة أن النظر في التاريخ التصويري للمريض (وليس فقط آخر مسح تم إجراؤه) يساعد النموذج على تقديم تنبؤات أفضل.
باخت-صار، لا يقوم ProgFormer بمجرد التخمين للمستقبل؛ بل يبنيه طبقة تلو أخرى. إنه يحترم حقيقة أن الدماغ مستقر في معظمه، لكنه شديد الوعي بالتغييرات الصغيرة والحرجة التي تشير إلى تطور المرض. ومن خلال فصل عمل "الصورة الكبيرة" عن عمل "التفاصيل الدقيقة"، فإنه ينجح في التنبؤ بمستقبل مسح الدماغ بمستوى من الدقة يشير إلى أن هذا النهج المكون من خطوتين هو طريقة جديدة وقوية لفهم كيف يتقدم أعمار أدمغتنا.
ملخص تقني: ProgFormer
بيان المشكلة
يعد التنبؤ بالصور المستقبلية للرنين المغناطای الهيكلي (sMRI) للدماغ مهمة بالغة الأهمية لتحليل وتوقع الأمراض التنكسية العصبية مثل مرض ألزهايمر. ومع ذلك، يفرض هذا المهمة تحديًا فريدًا: فالتغيرات الطولية غالبًا ما تكون دقيقة ومحصورة في مناطق تشريحية محددة (مثل الحصين، والبطينات الجانبية)، بينما يظل معظم بنية دماغ الشخص مستقرة بمرور الوقت.
تندرج الطرق الحالية عمومًا تحت فئتين، كل منهما تعاني من قيود:
الطرق القائمة على الفضاء الكامن (Latent-space-based methods): تقوم بتشفير الصور إلى تمثيلات منخفضة الأبعاد قبل التنبؤ. وبينما تتسم هذه الطرق بالكفاءة الحسابية، إلا أنها تعتمد على مشفر تلقائي (autoencoder) يتم تعلمه بشكل منفصل، مما قد يؤدي إلى فقدان المعلومات أثناء الضغط وإعادة البناء، وهو ما قد يضعف دقة التفاصيل الهيكلية الدقيقة.
طرق فضاء الفوكسل المباشرة (Direct voxel-space methods): تتنبأ بالمخرجات مباشرة في فضاء الفوكسل لتجنب إعادة بناء الفضاء الكامن. ومع ذلك، فهي عادة ما تستخدم مسار تنبؤ موحد لنمذجة كل من بنية الدماغ العالمية المستقرة والتغيرات المحلية الدقيقة المرتبطة بالتطور. ونتيجة لذلك، فإن المحتوى المستقر المهيمن غالبًا ما يطغى على التغيرات المحلية الدقيقة، مما يجعل من الصعب رصد تطور المرض بدقة متناهية.
المنهجية: ProgFormer
يقترح المؤلفون ProgFormer، وهو نموذج "ترانسفورمر" انتشار (Diffusion Transformer) هرمي يعمل في فضاء الفوكسل، صُمم لمعالجة هذه التحديات دون الاعتماد على مشفر تلقائي للصور يتم تدريبه بشكل منفصل. يعمل النموذج ضمن إطار عمل مطابقة التدفق الشرطي (Conditional Flow Matching)، حيث يتعلم حقل السرعة لنقل الضوضاء الغاوسية إلى توزيع الرنين المغناطيسي المستقبلي المستهدف.
البنية الجوهرية
يهيكل ProgFormer عملية التنبؤ إلى مسارين متميزين ولكن مترابطين:
المسار الخشن (الاستدلال على مستوى الرقعة - Patch-Level Reasoning):
المدخلات: يعالج المسوحات المرصودة وحالة التدفق الحالية كرموز رقع ثلاثية الأبعاد (non-overlapping 3D patch tokens) بأبعاد (p×p×p).
الآلية: يستخدم الانتباه المتعدد الرؤوس المكاني (Spatial MHSA) داخل كل مسح، والانتباه المتعدد الرؤوس السببي الزمني (Causal Temporal MHSA) عبر المسوحات المرصودة. وهذا يسمح للنموذج بالتقاط بنية الدماغ العالمية والسياق الطولي.
المخرجات: يولد سرعة حجمية أولية (νcoarse) وتمثيلات خشنة (C) تشفر بنية الدماغ الكاملة والسياق الزمني.
المسار الدقيق (التحسين على مستوى الفوكسل - Voxel-Level Refinement):
المدخلات: يعمل على رموز الفوكسل الفردية داخل كل رقعة ثلاثية الأبعاد.
الآلية: يطبق انتباه متعدد الرؤوس للفوكسل (Voxel MHSA) داخل الرقع، والأهم من ذلك، تطبيق انتباه متقاطع متعدد الرؤوس من الخشن إلى الدقيق (Coarse-to-Fine MHCA). في هذا الانتباه المتقاطع، تعمل رموز الفوكسل كاستعلامات (queries)، بينما تعمل التمثيلات الخشنة (C) من الرقعة المقابلة كـ مفاتيح (keys) وقيم (values).
الوظيفة: تستخدم هذه الآلية مخرجات المسار الخشن كـ "تأسيس مكاني-زماني" (spatio-temporal grounding) لتوجيه تحسين التفاصيل الهيكلية المحلية وتغيرات الكثافة.
المخرجات: يولد سرعة دقيقة (νfine).
التكامل والاستدلال
تقدير السرعة: حقل السرعة المتوقع النهائي هو مجموع السرعتين الخشنة والدقيقة: νθ=νcoarse+νfine.
التوليد: أثناء الاستدلال، يبدأ النموذج من ضوضاء غاوسية ويقوم بدمج حقل السرعة المقدر عبر سلسلة من خطوات أويلر (Euler steps) لتوليد صورة الرنين المغناطيسي المستقبلية المتوقعة.
الأنواع (Variants): يقيم المؤلفون تركيبين:
التسلسلي (ProgFormer-S): تكتمل جميع الكتل الخشنة قبل بدء أي من الكتل الدقيقة.
المتداخل (ProgFormer-I): يتم تداخل الكتل الخشنة والدقيقة في مراحل، مما يسمح للكتل الدقيقة بتلقي ميزات خشنة محسنة تدريجيًا.
التكييف والتدريب
التكييف (Conditioning): يتم تكييف النموذج مع وقت التدفق، والعمر المستهدف، والمتغيرات المصاحبة (التشخيص، الجنس) عبر AdaLN-Zero (التطبيع الطبقي التكيفي مع التهيئة الصفرية)، والذي يعدل معاملات القياس والإزاحة والبوابة لوحدات الانتباه وMLP.
دالة الخسارة (Loss Function): تُستخدم دالة خسارة مطابقة تدفق موزونة. حيث يقوم خريطة تسمية تشريحية (مشتقة من SynthSeg) بتعيين أوزان أعلى للمناطق ذات الصلة بالتطور (مثل الحصين، والبطينات) وأوزان أقل للأنسجة المستقرة/الخلفية، مما يضمن منح النموذج الأولوية للتغيرات المرتبطة بالمرض.
المساهمات الرئيسية
بنية هرمية شاملة (End-to-End): يعد ProgFormer أول بنية موحدة تقوم معًا بعمليات الاستدلال المكاني-الزماني على مستوى الرقعة العالمية ونمذجة الفوكسل الدقيقة داخل الرقعة للتنبؤ الطولي بالرنين المغناطيسي دون الحاجة لمشفر تلقائي منفصل.
التأسيس من الخشن إلى الدقيق (Coarse-to-Fine Grounding): يصمم النموذج لفصل نمذجة الاستقرار العالمي (المسار الخشن) عن التطور المحلي (المسار الدقيق) بشكل صريح، باستخدام الأول لتأسيس الثاني، مما يمنع التغيرات الدقيقة من أن تطغى عليها الهياكل المستقرة.
أداء يتفوق على الحالة الراهنة (State-of-the-Art): أظهرت التجارب المكثفة على ثلاثة معايير (ADNI, AIBL, OASIS) أداءً فائقًا في كل من إعدادات التنبؤ الثنائي (pairwise) وتنبؤ المسار (trajectory prediction) مقارنة بالطرق الحالية القائمة على الفضاء الكامن أو الفضاء الحجمي المباشر.
النتائج التجريبية
تم تقييم النموذج على مجموعات بيانات ADNI وAIBL وOASIS باستخدام مقاييس PSNR وSSIM وR-MAE على مستوى المنطقة.
التنبؤ الثنائي (Pairwise Prediction): حقق نموذج ProgFormer-S أعلى PSNR عبر المجموعات الثلاث (على سبيل المثال، 26.39 في ADNI، متفوقًا على أقرب منافس بمقدار 0.20). وحقق نموذج ProgFormer-I أفضل دقة إقليمية (أقل R-MAE) وأعلى SSIM على مجموعة بيانات ADNI.
تنبؤ المسار (Trajectory Prediction): أدى استخدام التاريخ المرصود (مسوحات متعددة) إلى تحسين الأداء بشكل كبير مقارنة بالنماذج ذات المدخل الواحد. حقق ProgFormer-S باستخدام التاريخ أعلى PSNR عبر جميع المجموعات.
دراسات الاستئصال (Ablation Studies):
أدى إزالة المسار الخشن إلى انهيار التنبؤ (انخفض PSFT إلى ~7)، مما يؤكد ضرورته لتأسيس البنية العالمية.
أدت إزالة المسار الدقيق أو الانتباه المتقاطع إلى تدهور كبير في كل من جودة الصورة والدقة الإقليمية، مما يسلط الضوء على أهمية تحسين مستوى الفوكسل الموجه بالسياق الخشن.
أظهر الوزن الإقليمي أنه حاسم؛ حيث أدت إزالته إلى انخفاضات كبيرة في الأداء، مما يثبت أن التركيز على المناطق ذات الصلة بالتطور يحسن الدقة الإجمالية.
التحليل البصري: تظهر النتائج النوعية أن ProgFormer يحافظ على تكوينات البطينات وخطوط القشرة بشكل أكثر اتساقًا من النماذج المرجعية، حيث تتركز الأخطاء في المناطق ذات الصلة بالتطور بدلاً من التراكم عالميًا.
الأهمية والادعاءات
يدعي البحث أن ProgFormer يثبت فعالية النمذجة الهرمية في فضاء الفوكسل للتنبؤ الحجمي الطولي. ومن خلال فصل نمذجة البنية العالمية المستقرة عن التغيرات المحلية الدقيقة، تحقق الطريقة تنبؤًا دقيقًا تشريحيًا وحساسًا للتطور. يضع المؤلفون عملهم كخطوة مهمة للأمام بعيدًا عن طرق الفضاء الكامن (التي تعاني من فقدان إعادة البناء) والطرق المباشرة الموحدة (التي تجد صعوبة في رصد التغيرات الدقيقة مقابل الخلفيات المستقرة). ويؤكد العمل أن النمذجة المباشرة القائمة على "الترانسفورمر"، عندما تُبنى هيكليًا مع التأسيس المكاني-الزماني، تعد نهجًا قابلًا للتطبيق ومتفوقًا لتوليد صور الرنين المغناطيسي المستقبلية للدماغ.