CiUNet: A Hybrid Swin-CNN UNet for Medical Image Segmentation
تقترح هذه الورقة البحثية نموذج CiUNet، وهو بنية هجينة خفيفة الوزن من نوع Swin-CNN U-Net تدمج مشفر CNN متوازياً، ودمجاً غير متماثل للميزات، واتصالات تخطي عبر الطبقات لتحقيق أقصى درجات الدقة والكفاءة والقابلية للتفسير في تقسيم الصور الطبية.
في هدوء المستشفيات الحديثة، غالباً ما يُطلب من الحاسوب أداء مهمة تتطلب يد جراح ثابتة وعين محقق ثاقبة: يجب عليه النظر في مسح طبي وتتبع الخطوط العريضة الدقيقة لعضو بشري. هذه العملية، المعروفة باسم "تجزئة الصور" (image segmentation)، هي المعادل الرقمي للتلوين داخل الخطوط، لكن الخطوط هنا غالباً ما تكون باهتة، والأشكال معقدة، والرهانات عالية جداً. لسنوات طويلة، كانت الأداة القياسية لهذه المهمة هي نوع من الذكاء الاصطناعي يسمى "الشبكة العصبية التلافيفية" (CNN). فكر في هذا النظام كعامل بارع في ملاحظة التفاصيل الدقيقة، مثل ملمس الجلد أو حافة العظم، ولكنه يعاني في فهم الصورة الكبيرة، مثل كيفية اتصال الكبد ببقية الجسم. وفي الآونة الأخيرة، ظهر نوع مختلف من الذكاء الاصطناعي، يعتمد على تقنية تسمى "المحول" (Transformer). هذا العامل الجديد هو سيد في رؤية المشهد بأكمله وفهم الروابط بعيدة المدى، ومع ذلك فإنه غالباً ما يفتقد التفاصيل الصغيرة والحاسمة التي تحدد الحدود الدقيقة للعضو. لقد كان التحدي الذي يواجه العلماء هو بناء نظام يمتلك أفضل سمات كليهما: القدرة على رؤية الغابة والقدرة على عد الأشجار.
لقد عالج فريق من الباحثين في شركة "Ciphowork GmbH" هذه المعضلة من خلال بنية جديدة أطلقوا عليها اسم "CiUNet". يقترح عملهم، الذي تم اختباره على مجموعة بيانات من الأشعة المقطعية للبطن تحتوي على ثلاثين صورة أشعة مقطعية، حلاً هجيناً يدمج هذين النهجين المتميزين في نظام واحد انسيابي. وبدلاً من إجبار نوع واحد من الذكاء الاصطناعي على القيام بكل شيء، قاموا ببناء إطار عمل ثنائي المحركات. أحد المحركين هو "المحول"، الذي يمسح الصورة لفهم التخطيط العام وسياق الأعضاء. وبالتوازي معه، يعمل محرك ثانٍ، وهو "الشبكة العصبية التلافيفية"، والتي تركز حصرياً على التقاط الأنسجة عالية الدقة والحواف الحادة التي قد يفتقدها المحرك الأول. هذه التدفقات من المعلومات لا تُلقى معاً فحسب؛ بل يتم نسجها بعناية. فقد صمم الباحثون آلية محددة لأخذ التفاصيل الدقيقة من المراحل المبكرة للشبكة التلافيفية وتغذيتها مباشرة في الطبقات العميقة لـ "مفكك" (decoder) المحول. وهذا يضمن أنه عندما يعيد الكمبيوتر بناء الصورة النهائية للأعضاء، فإنه لا يفقد دقة الحدود.
تم قياس نتائج هذا النهج مقابل معيار صارم باستخدام مجموعة بيانات تُعرف باسم "Synapse"، والتي تتضمن صوراً لثمانية أعضاء مختلفة في البطن مثل الأبهر، والمرارة، والكلى. ووجد الفريق أن نموذجهم الهجين حقق مستوى عالياً من الدقة، بمتوسط "درجة دايس" (Dice score) بلغ 83.72 بالمائة. والأهم من ذلك، أن النظام تفوق في تحديد حواف الأعضاء، وهو عامل حاسم للتخطيط الجراحي. وفي الاختبارات التي تقيس مدى ابتعاد الحدود المتوقعة عن الحدود الحقيقية، أظهر النموذج الجديد تحسناً ملحوظاً عن الطرق السابقة التي اعتمدت فقط على تصميم "المحول". وقد أدى أداءً جيداً بشكل خاص في الأعضاء التي يصعب تحديدها غالباً، مثل الكلى والمرارة، مما يشير إلى أن إضافة محرك التفاصيل المحلية قد نجح في تعويض نقاط ضعف محرك السياق العالمي.
لضمان تعلم النظام بفعالية، استخدم الباحثون استراتيجية تعليمية تحاكي كيفية تعلم البشر لمهارة معقدة. فبدلاً من إلقاء مجموعة البيانات بأكملها على الكمبيوتر دفعة واحدة، بدأوا بأمثلة أبسط، مع التركيز في البداية على الشرائح التي تحتوي على الأعضاء المستهدفة، وتحديداً البنكرياس والمرارة. ومع إتقان النظام لهذه الأمثلة، أصبحت بيانات التدريب أكثر تعقيداً تدريجياً، حيث أدخلوا المزيد من الضوضاء الخلفية ومجموعة متنوعة أوسع من الهياكل التشريحية. سمح هذا التقدم خطوة بخطوة للنموذج ببناء أساس متين قبل التعامل مع الحالات الأكثر صعوبة. علاوة على ذلك، أضاف الباحثون طبقة من الإشراف الداخلي، مما منح النظام تغذية راجعة في مراحل متوسطة من معالجته. عمل هذا كدليل، يساعد الأجزاء العميقة من الشبكة على فهم ما تراه ويضمن بقاء المخرجات النهائية متسقة مع الملاحظات التفصيلية المبكرة.
إن تداعيات هذا العمل تمتد إلى ما وراء مجرد الأرقام في المخططات. فقد سلط الباحثون الضوء على أن تصميمهم يوفر ميزة عملية للنشر في العالم الحقيقي، لا سيما فيما يتعلق بخصوصية المريض. ولأن النظام يمكنه فصل المعالجة الأولية للصورة عن العمل الحسابي الثقيل، يمكن للمستشفى تقنياً معالجة بيانات المرضى الحساسة محلياً على جهاز صغير وآمن وإرسال الميزات المجردة وغير القابلة لتحديد الهوية فقط إلى خادم سحابي لإجراء التحليل النهائي. هذا الفصل يسمح بسير عمل آمن وفعال يحافظ على دقة عالية دون كشف الصور الطبية الخام. وبينما لا يزال النموذج يواجه تحديات مع بعض الأعضاء المعقدة مثل المعدة، حيث لم يطابق تماماً أداء النماذج ثلاثية الأبعاد الأكثر تقدماً، إلا أنه يمثل خطوة كبيرة للأمام. فهو يثبت أنه من خلال الجمع بين نقاط القوة لفلسفتين مختلفتين من الذكاء الاصطناعي، من الممكن إنشاء أداة ليست دقيقة وفعالة فحسب، بل أيضاً قابلة للتفسير وآمنة بما يكفي للبيئة المتطلبة للطب السريري.
ملخص تقني لنموذج CiUNet: شبكة هجينة من نوع Swin-CNN UNet لتجزئة الصور الطبية
بيان المشكلة تتطلب تجزئة الصور الطبية توازناً بين الدقة العالية، والمتانة، والكفاءة الحسابية، لا سيالما عند النشر التجاري حيث يكون الحفاظ على الخصوصية أمراً بالغ الأهمية. وبينما ظلت بنية U-Net هي المعيار السائد لفترة طويلة، إلا أن الشبكات العصبية التلافيفية (CNNs) الصرفة تعاني في التقاط الاعتمادات العالمية بعيدة المدى بسبب المحلية الجوهرية للتلافيف. ومن ناحية أخرى، تتفوق البنيات القائمة على المحولات (Transformers) الصرفة، مثل Swin-UNet، في نمذجة الاعتمادات بعيدة المدى عبر آلية الانتباه الذاتي، لكنها تعاني من ثلاث مشكلات محددة: عدم كفاية الحفاظ على التفاصيل النسيجية المحلية بسبب غياب التحيز الاستقرائي التلافي، وإعادة البناء المكاني غير الدقيقة بسبب عدم كفاية انتشار الميزات عبر اتصالات التخطي (skip connections)، ومحدودية القابلية للتفسير إلى جانب عدم استقرار التدريب في بنيات فك التشفير (decoder) العميقة.
المنهجية يقترح المؤلفون CiUNet، وهي بنية هجينة خفيفة الوزن على شكل حرف U، صُممت للتخفيف من عيوب تصميمات المحولات الصرفة مع الحفاظ على كفاءة نموذج U-Net. وتتضمن المنهجية الجوهرية المكونات التالية:
إطار عمل ثنائي المشفر غير متماثل (Asymmetric Dual-Encoder Framework): بدلاً من استخدام مشفر Swin واحد، يستخدم CiUNet بنية متوازية.
فرع Swin: يستخدم مشفر Swin Transformer (المُهيأ بأوزان Swin-Tiny مسبقة التدريب) ليعمل كعمود فقري أساسي للاستدلال الدلالي العالمي.
فرع CNN: يدمج مشفر ResNet خفيف الوزن ومتوازٍ لالتقاط التفاصيل المكانية المحلية والنسيج عالي الدقة بشكل صريح.
استراتيجية التجميد (Freezing Strategy): للاستفادة من الأولويات البصرية مسبقة التدريب، يتم تجميد وحدات استخراج الميزات في الطبقات الأولية لكلا الفرعين، مما يحصر المعلمات القابلة للتدريب في طبقات خفض العينات اللاحقة.
الدمج الديناميكي: يقوم وحدة دمج مخصصة بدمج الميزات من كلا الفرعين. وهي تعمل ديناميكياً: إذا قدم كلا الفرعين ميزات، يتم دمجها ومعالجتها عبر شبكة خفيفة الوزن؛ وإذا كان فرع واحد فقط موجوداً، يتم تمريرها عبر رسم الخرائط المتطابقة (identity mapping). ومن الجدير بالذكر أن طبقة عنق الزجاجة (bottleneck) تُشتق حصرياً من فرع Swin، بينما يتم توفير الطبقة الأكثر ضحالة (Layer0) حصرياً من فرع CNN.
آلية XSkip (التخطي المتقاطع): لمعالجة الفجوة الدلالية واستعادة الدقة المكانية المفقودة، استبدل المؤلفون اتصالات التخطي القياسية بوحدة XSkip. تقوم هذه الآلية بمحاذاة ونشر ميزات CNN الضحلة في طبقات فك التشفير الأعمق باستخدام الانتباه المكاني. وهي تتضمن الاستكمال الثنائي (bilinear interpolation) وتلافيف 1×1 لضغط القنوات قبل دمجها مع ميزات فك التشفير العميقة، مما يضمن تحديداً أكثر حدة للحدود.
فك التشفير والاهتمام المعزز: تم بناء فك التشفير بشكل غير متماثل لاستيعاب تصميم المشفر المزدوج. وهو يتضمن عملية رفع عينات محسنة باستخدام الاستكمال الثنائي وآلية وزن ديناميكية خفيفة الوزن. علاوة على ذلك، تتبنى آلية الانتباه نظام LogSpacedCPB (الانحياز الموضعي المستمر ذو المسافات اللوغاريتية) من Swin Transformer V2، مستبدلةً فهرسة الإحداثيات المنفصلة بنهج انحدار مستمر لتحسين التعميم عبر المقاييس المختلفة للمدخلات.
الإشراف المساعد متعدد المستويات: لتحسين استقرار التدريب والقابلية لتفسير الميزات العميقة، تم إرفاق رؤوس إشراف مساعدة (Aux-Head) بمخرجات فك التشفير المتوسطة (تحديداً Layer1 و Layer2). تستخدم هذه الرؤوس كتل متبقية (residual blocks) خفيفة الوزن لتوليد تنبؤات التجزئة عند مستويات متوسطة، مما يوفر توجيهاً للميل (gradient guidance) متعدد المستويات. هذا التصميم متاح بفضل تعزيز قابلية تفسير الميزات العميقة نتيجة لفرع CNN المتوازي.
استراتيجية التدريب:
التعلم المنهجي (Curriculum Learning): يتم استخدام استراتيجية متعددة المراحل حيث تزداد تعقيدات مجموعة التدريب تدريجياً. تركز المراحل المبكرة على الشرائح التي تحتوي على الأعضاء البؤرية (البنكرياس، المرارة)، بينما تدمج المراحل اللاحقة الشرائح المهيمن عليها الخلفية وتوزيع مجموعة البيانات الكاملة.
دالة الخسارة (Loss Function): يجمع هدف التحسين بين ثلاثة قيود هندسية: التصنيف على مستوى البكسل (Focal Loss)، والتداخل على مستوى المنطقة (Weighted Dice Loss مع وزن حدودي ديناميكي)، ومحاذاة الحدود (HausdorffER Loss).
المساهمات الرئيسية تحدد الورقة ثلاث مساهمات رئيسية:
إطار عمل ثنائي المشفر غير متماثل: تقديم مشفر ResNet موازٍ ليكمل Swin Transformer، مما يلتقط صراحةً التفاصيل المكانية المحلية بينما يتعامل فرع Swin مع السياق العالمي.
آلية XSkip: تعزيز لفك التشفير يقوم بمحاذاة ونشر ميزات CNN الضحلة انتقائياً عبر الانتباه المكاني لاستعادة التفاصيل عالية الدقة وتحسين دقة الحدود.
الإشراف المساعد متعدد المستويات: دمج رؤوس مساعدة لتسريع التقارب وتعزيز قابلية تفسير مُحول فك التشفير العميق، بالاستفادة من الوضوح الدلالي الذي يوفره فرع CNN.
النتائج التجريبية تم تقييم النموذج على مجموعة بيانات Synapse متعددة الأعضاء لصور CT، وهي معيار يتكون من 33 صورة CT معززة بالتباين مع تسميات لثمانية أعضاء بطنية.
الأداء: حقق CiUNet متوسط معامل تشابه Dice (DSC) بنسبة 83.72% ومسافة Hausdorff 95% (HD95) قدرها 12.13 مم.
المقارنات: تفوق النموذج على Swin-UNet الأصلي (79.13% DSC)، وTransUNet (77.48%)، وMISSformer (81.96%). وظل مقارباً للنماذج المرجعية القوية مثل Swin-UNETR (83.48%) و HiFormer-L (80.69%)، رغم استخدامه عدداً أقل بكثير من المعلمات (31.68 مليون معلمة قابلة للتدريب). ومع ذلك، تشير الورقة إلى أن أداء CiUNet أقل من الطرق الحديثة (SOTA) مثل nnFormer (86.57% DSC) و UNETR++ (87.22%).
الأداء الخاص بالأعضاء: أظهر CiUNet أداءً فائقاً في المرارة (79.47%) وكلتا الكليتين (اليسرى: 91.39%، اليمنى: 91.27%). ولوحظ أن الأداء في البنكرياس (59.31%) والمعدة (76.00%) كان دون المستوى مقارنة بالنماذج ثلاثية الأبعاد الحديثة.
دراسات الاستئصال (Ablation Studies): أكدت التجارب أن ترقيات الإطار وحده حسنت DSC إلى 81.40%، كما عززت آلية XSkip دقة الحدود (انخفض HD95 إلى 11.26 مم)، وساهم الإشراف المساعد في تسريع التقارب للوصول إلى ذروة الأداء النهائية.
الأهمية والادعاءات تضع الورقة CiUNet كحل دقيق، وفعال، وقابل للتفسير للنشر السريري. ويتمثل ادعاء مهم بخصوص جدواه التجارية في نموذج SaaS الذي يحافظ على الخصوصية والذي تتيحه عملية فصل المشفر عن فك التشفير. يجادل المؤلفون بأن صور CT الحساسة للمرضى يمكن معالجتها محلياً عند طبقة المشفر خفيفة الوزن، مع نقل فقط متجهات الميزات الدلالية عالية المستوى إلى السحابة لإجراء حسابات فك التشفير الثقيلة، مما يسهل التعاون الآمن بين المستشفى والسحابة.
يقر المؤلفون بالقيود، بما في ذلك الانحياز في المقاييس نحو النتائج الإيجابية الكاذبة، وفجوة الأداء مع النماذج ثلاثية الأبعاد بالنسبة للأعضاء الصغيرة بسبب غياب السياق بين الشرائح في المعالجة ثنائية الأبعاد، والطبيعة غير المثالية لاستخدام خلفيات مدربة مسبقاً على الصور الطبيعية (Swin-Tiny و ResNet) بدلاً من التدريب المسبق المتخصص في المجال الطبي. وتخلص الورقة إلى أنه بينما يوفر CiUNet توازناً مواتياً بين الدقة والكفاءة، يجب أن تعالج الأعمال المستقبلية الوعي بين الشرائح واستراتيجيات التدريب المسبق المخصدة طبياً.
ملاحظة: نظراً للاعتبارات التجارية، ذكر المؤلفون أنه يتم إصدار الأوزان المدربة على مجموعة بيانات Synapse فقط مع مُتنبئ عرض (demo) خفيف الوزن، وليس التنفيذ الكامل.