A Unified Benchmark of Deep Learning Models for Multi-task 3D Brain Tumor Segmentation from Magnetic Resonance Imaging
تقدم هذه الورقة معياراً موحداً يقيم بإنصاف خمس بنيات للتعلم العميق هي الأحدث حالياً (بما في ذلك الشبكات العصبية الالتفافية CNNs، والمحولات Transformers، ونماذج فضاء الحالة SSMs) لتقسيم أورام الدماغ ثلاثية الأبعاد متعددة المهام على مجموعات بيانات BraTS 2023 و2024 تحت ظروف تجريبية متطابقة لتحليل المقايضات بين دقة التقسيم والكفاءة الحسابية.
المؤلفون الأصليون:Diego J. Torrejón, Luna Y. Hernández, Javier Sánchez
تخيل أنك محقق يحاول حل لغز داخل دماغ مريض. الأدلة مخبأة في صور الرنين المغناطيسي (MRI)، وهي تشبه صور الأشعة السينية ثلاثية الأبعاد فائقة التفصيل التي تظهر الأنسجة الرخوة بتدرجات من اللون الرمادي. لعقود من الزمن، كان على الأطباء رسم الخطوط الخارجية للأورم يدوياً على هذه الصور، وهي مهمة بطيئة، ومجهدة، وعرضة للخطأ البشري — تماماً مثل محاولة رسم خريطة مثالية لخط ساحلي أثناء ركوب أفعوانية متعرجة. ولتسريع العملية، بنى العلماء "محققين رقميين" يُطلق عليهم نماذج الذكاء الاصطناعي. تستخدم هذه النماذج التعلم العميق، وهو نوع من "أدمغة الكمبيوتر" التي تتعلم من خلال النظر في آلاف الأمثلة، لتحديد وتحديد حدود الأورام تلقائياً. لكن العقدة تكمل في أن هناك أنواعاً مختلفة من هؤلاء المحققين الرقميين، وكل منها يتعلم بطرق مختلفة قليلاً. فبعضها يشبه الفنانين من الطراز القديم الذين ينظرون إلى ضربات الفرشاة الصغيرة (الشبكات العصبية التلافيفية - CNNs)، بينما البعض الآخر يشبه المراقبين الفائقين الذين يمكنهم رؤية الصورة الكاملة دفعة واحدة (المحولات - Transformers)، وهناك مجموعة جديدة ومثيرة تعمل كمسافرين أكفاء يتذكرون المسار الذي سلكوه دون أن يضلوا الطريق (نماذج فضاء الحالة - State Space Models). السؤال الكبير هو: أي محقق هو الأفضل حقاً في حل القضية؟
تضع هذه الورقة البحثية "نزالاً عادلاً" ضخماً لإيجاد الإجابة. فقد قرر المؤلفون، وهم فريق من الباحثين من جامعة لاس بالماس دي غران كناريا، التوقف عن مقارنة "التفاح بالبرتقال". فبدلاً من النظر في دراسات استخدمت طرق تدريب أو بيانات مختلفة، قاموا ببناء "معيار موحد". فكر في الأمر كصالة ألعاب رياضية معيارية حيث يُجبر خمسة رياضيين مختلفين — ثلاثة من الأبطال الراسخين واثنان من المنافسين الجدد — على ركض نفس مسار العقبات تماماً وتحت نفس القواعد. لقد اختبروا هذه النماذج في سيناريوهين مختلفين جداً لأورام الدماغ: أحدهما يتعلق بالأورام السحائية (meningiomas) (وهي أورام عادة ما تكون ذات حواف واضحة ومحددة) والآخر يتعلق بالأورام الدبقية (gliomas) التي خضعت بالفعل للعلاج بالجراحة أو الإشعاع (حيث يكون التشريح فوضوياً، ومليئاً بالندبات والمربك).
كانت نتائج هذا السباق الرقمي كاشفة للغاية. تشير الورقة إلى أن الجيل الجديد من النماذج، المعروفة بنماذج فضاء الحالة (وتحديداً SegMamba و SegMambaV2)، تفوقت باستمرار على الأبطال القدامى الراسخين. كانت هذه النماذج الجديدة أفضل في العثور على الأورم ورسم حدودها بدقة أكبر، حتى في الحالات الفوضوية لما بعد العلاج. وبينما كانت نماذج "المحولات" القديمة (مثل Swin UNETR) جيدة جداً في رصد أجزاء معينة من الورم، إلا أنها واجهت صعوبات أحياناً في رؤية الصورة العامة أو استغرقت وقتاً طويلاً في "التفكير". أما نماذج "الشبكات التلافيفية" الكلاسيكية (CNN) فقد كانت سريعة وفعالة، لكنها لم تكن دقيقة تماماً مثل المنافسين الجدد.
ومن الأهمية بمكان أن الورقة تجادل ضد فكرة أن عليك الاختيار بين السرعة والدقة. فقد تمكنت نماذج فضاء الحالة الجديدة من أن تكون دقيقة للغاية وسريعة بشكل معقول في آن واحد، مما قدم توازناً أفضل من الخيارات المثلى السابقة. ومع ذلك، يوضح المؤلفون بحذر أنه على الرغم من أن هذه النماذج هي الفائزة في هذا السباق المحدد والمسيطر عليه، إلا أن المقارنة تعتمد على مجموعة واحدة من القواعد ومجموعتي بيانات محددتين. هم لا يدّعون أن هذه النماذج هي الحل النهائي والمثالي لكل موقف طبي في كل مكان، بل يمثلون خطوة كبيرة للأمام في كيفية تحديد ورسم خرائط أورام الدماغ بشكل تلقائي وموثوق. وتخلص الدراسة إلى أنه بالنسبة للغز ثلاثي الأبعاد المعقد لتقسيم أورام الدماغ، فإن بنية "فضاء الحالة" الجديدة هي التي تحمل التاج حالياً، مما يوفر طريقة أكثر قوة لمساعدة الأطباء على رؤية ما لا يُرى.
ملخص تقني: معيار موحد لنماذج التعلم العميق لتقسيم أورام الدماغ متعددة المهام من التصوير بالرنين المغناطيسي
بيان المشكلة يعد التقسيم التلقائي لأورام الدماغ من صور الرنين المغناطيسي (MRI) أمراً بالغ الأهمية للتشخيص، وتخطيط العلاج، والمراقبة. ورغم التقدم الكبير الذي حققته بنيات التعلم العميق، إلا أن المقارنة الموضوعية لهذه الأساليب تظل أمراً صعباً؛ حيث غالباً ما تستخدم الدراسات المنشورة ظروفاً تجريبية متباينة، تشمل مجموعات بيانات مختلفة، ومسارات معالجة مسبقة، وبروتوكولات تدريب، ومقاييس تقييم متنوعة. علاوة على ذلك، تركز العديد من المقارنات فقط على دقة التقسيم مع إهمال الكفاءة الحسابية، وهو عامل حيوي للنشر السريري في العالم الحقيقي. هذا النقص في التوحيد يجعل من الصعب تحديد ما إذا كانت فروق الأداء ناتجة عن القدرات الجوهرية للبنية المعمارية أم عن المتغيرات التجريبية.
المنهجية لمعالجة هذه القيود، طور المؤلفون معياراً تجريبياً موحداً وقابلاً لإعادة الإنتاج يقيم خمس بنيات تمثيلية للتقسيم ثلاثي الأبعاد تحت ظروف متطابقة. تقارن الدراسة بين ثلاث عائلات معمارية:
الشبكات العصبية الالتفافية (CNNs): نموذج 3D U-Net ونموذج SegResNet.
النماذج القائمة على المحولات (Transformers): نموذج Swin UNETR.
نماذج فضاء الحالة (SSMs): نموذجي SegMamba وSegMambaV2 (المشتقين من إطار عمل Mamba).
مجموعات البيانات والبروتوكول التجريبي أُجري التقييم على سيناريوهين سريريين متميزين من تحديات تقسيم أورام الدماغ (BraTS):
BraTS 2023: تقسيم الورم السحائي داخل الجمجمة، والذي يتميز بحدود واضحة ولكن مع تباين في الحجم والموقع.
BraTS 2024: تقسيم الورم الدبقي بعد العلاج، وهو سيناريو أكثر تعقيداً يتضمن تجاويف الاستئصال، والوذمة، والأنسجة النخرية.
تم تدريب جميع النماذج باستخدام مسار موحد تم تنفيذه عبر مكتبة MONAI. وشملت الضوابط المنهجية الرئيسية ما يلي:
معالجة البيانات: معالجة مسبقة متطابقة (إعادة تسجيل صلب، إعادة أخذ العينات إلى دقة متساوية الأبعاد قدرها 1 مم³، وإزالة الجمجمة)، وتعزيز البيانات (القص العشوائي، الانعكاس، وتغيير شدة الإضاءة)، وتقسيمات التحقق المتقاطع خماسي الطيات المستمدة من البيانات الموسومة للتدريب.
تكوين التدريب: تم تدريب جميع النماذج من البداية باستخدام مُحسن AdamW ودالة فقدان "Dice Cross-Entropy" المشتركة لمعالجة عدم توازن الفئات. كما تم توحيد المعلمات الفائقة (معدلات التعلم، أحجام الدفعات)، مع إجراء تعديلات طفيفة لضمان استقرار نماذج SSM (مثل قص التدرج).
مقاييس التقييم: تم تقييم الأداء باستخدام التداخل الحجمي (معامل تشابه Dice، وتقاطع الاتحاد)، ودقة الحدود (مسافة Hausdorff عند المئوية 95)، والكفاءة الحسابية (وقت الاستدلال وعدد المعلمات).
النتائج الرئيسية أسفرت التجارب عن النتائج التالية عبر كلا مجموعتي البيانات:
الأداء العام: حققت نماذج فضاء الحالة (SSMs)، وتحديداً SegMamba وSegMambaV2، باستمرار أعلى دقة تقسيم عالمية وأكثر تحديد دقيق للحدود (أدنى قيمة لـ HD95) عبر مهام الورم السحائي والورم الدبقي بعد العلاج.
تحليل المناطق الفرعية: أظهرت نماذج SSM قوة فائقة في تقسيم المناطق الفرعية النادرة والصغيرة (مثل قلب الورم غير المعزز). ومع ذلك، أظهر نموذج Swin UNETR القائم على المحولات أداءً تنافسياً، لا سيما في تحديد هياكل تشريحية معينة مثل الورم المعزز في سيناريوهات ما بعد العلاج.
مقايضات الكفاءة:
قدم نموذج 3D U-Net أسرع استدلال (0.2 ثانية) وأقل عدد من المعلمات (2 مليون) ولكنه أعطى أدنى دقة.
وفر SegResNet توازناً قوياً بين الدقة والكفاءة، مما يجعله مناسباً للبيئات ذات الموارد المحدودة.
تطلب Swin UNETR موارد حسابية أعلى بكثير (62 مليون معلمة) ووقت استدلال أطول (1.4 ثانية)، مع دقة متغيرة اعتماداً على مجموعة البيانات.
حقق SegMambaV2 أفضل مقايضة بين الدقة والكفاءة للاحتياجات عالية الأداء، حيث تفوق على النموذج الأساسي SegMamba في السرعة (حوالي 1.55–1.70 ثانية مقابل 1.85–1.95 ثانية) رغم امتلاكه عدداً أكبر من المعلمات (~138 مليون مقابل ~67 مليون).
المقارنة مع أحدث ما توصل إليه العلم: عند مقارنتها بأفضل المشاركات في التحديات (التي استخدمت أساليب التجميع والتدريب المكثف)، حققت البنيات أحادية النموذج في هذا المعيار درجات Dice تنافسية للغاية، خاصة نماذج SSM. ومع ذلك، يشير المؤلفون إلى أن المقارنة المباشرة لمقاييس الخطأ المكاني (HD95) معقدة بسبب الاختلافات في منهجيات التقييم (على مستوى الآفة مقابل مستوى الفوكسل).
الأهمية والمساهمات يدعي البحث أربع مساهمات رئيسية:
إطار عمل موحد: يقترح إطار عمل تجريبي قابل لإعادة الإنتاج بالكامل يقيم بنيات تقسيم أورام الدماغ ثلاثية الأبعاد الحديثة تحت ظروف تدريب وتقييم متطابقة، مما يلغي المتغيرات التجريبية المربكة.
مقارنة شاملة: يقدم مقارنة منهجية بين بنيات CNN وTransformer وSSM عبر مجموعتي بيانات BraTS المتميزتين سريرياً والمتحديتين.
تحليل الدقة والكفاءة: يقدم رؤى عملية حول المقايضات بين الأداء التنبئي والتكلفة الحسابية، مما يسلط الضوء على مدى ملاءمة النماذج المختلفة لسيناريوهات النشر السريري المحددة.
السلوك المعماري: يحلل كيفية سلوك عائلات النماذج المختلفة عبر مورفولوجيا الأورام المتنوعة والسياقات السريرية المختلفة، مع تحديد نقاط قوة نماذج SSM في نمذجة السياق العالمي ودقة الحدود.
يخلص المؤلفون إلى أنه بينما توفر نماذج SSM حالياً الأداء الأكثر قوة لمهام التقسيم ثلاثي الأبعاد المعقدة، فإن نماذج CNN تظل قابلة للتطبيق في التطبيقات التي تتطلب كفاءة عالية، بينما تحتفظ المحولات (Transformers) بقيمتها لتحديد هياكل تشريحية معينة. وتؤكد الدراسة أن الاختيارات المعمارية يجب أن تُوزن مقابل كل من المتطلبات السريرية والقيود الحسابية.