OptiModNet: A UNet-Transformer Hybrid with Grouped-Query and Channel Attention for Optic Disc and Cup Segmentation
تقدم الورقة البحثية OptiModNet، وهي بنية هجينة خفيفة الوزن تجمع بين UNet وTransformer وتتضمن آليات استعلام مجمعة وانتباه القنوات إلى جانب فقد هرمي مجمع، والتي تحقق أداءً هو الأفضل في فئته في تقسيم قرص العصب البصري والقعرة على مجموعة بيانات REFUGE2 مع الحفاظ على كفاءة حوسبية عالية.
يُعد الجلوكوما (المياه الزرقاء) سبباً رئيسياً للعمى غير القابل للعلاج في جميع أنحاء العالم، وغالباً ما يتطور بصمت دون أعراض مبكرة. ولأن الضرر الناتج عنه دائم، فإن اكتشاف المرض في مراحله الأولى يعد أمراً بالغ الأهمية. يعتمد الأطباء على فحص الجزء الخلفي من العين، وتحديداً من خلال النظر إلى هيكلين دائريين: القرص البصري، حيث يدخل العصب إلى العين، واللقفة البصرية، وهي منخفض أصغر داخل القرص. ومن خلال قياس حجم اللقفة بالنسبة للقرص، يمكن للأطباء حساب نسبة تعمل كعلامة تحذير حيوية للمرض. ومع ذلك، فإن رسم هذه الحدود يدوياً أمر صعب، ويستغرق وقتاً طويلاً، وعرضة للخطأ البشري، خاصة عندما تكون الصور ضبابية أو التباين فيها منخفضاً. ولحل هذه المشكلة، لجأ العلماء إلى الذكاء الاصطناعي، حيث قاموا بتعليم الحواسيب كيفية التعرف على هذه الهياكل تلقائياً. وبينما تستطيع البرامج الحاسوبية الحالية القيام بذلك، إلا أن النماذج الأكثر دقة غالباً ما تكون ثقيلة ومعقدة للغاية بحيث تتطلب أجهزة قوية ومكلفة، مما يجعل استخدامها صعباً في العيادات اليومية أو برامج الفحص واسعة النطاق حيث السرعة والبساطة أمران ضروريان.
لقد طور فريق من الباحثين نهجاً جديداً يوازن بين الدقة العالية والخفة، حيث أنشأوا نظاماً أطلقوا عليه اسم "OptiModNet". كان هدفهم هو بناء نموذج يمكنه رؤية الصورة الكاملة لتشريح العين مع ملاحظة التفاصيل الدقيقة والصغيرة للحواف أيضاً، دون الحاجة إلى حاسوب خارق لتشغيله. وقد حققوا ذلك من خلال الجمع بين نوعين مختلفين من بنيات الذكاء الاصطناعي. النوع الأول، المعروف باسم "UNet"، ممتاز في رصد التفاصيل المحلية والحواف، تماماً مثل شخص ينظر عن كثب إلى جزء محدد من خريطة. أما النوع الآخر، القائم على تقنية "المحول" (Transformer)، فهو أفضل في فهم الصورة الكبيرة وكيفية ارتباط الأجزاء المختلفة ببعضها البعض عبر مسافات طويلة. وكانت المحاولات السابقة لدمج هذين النوعين غالباً ما تؤدي إلى نماذج لا تزال ثقيلة أو معقدة. وقد قام الباحثون في هذه الدراسة بتحسين عملية الدمج من خلال إدخال تحسينين محددين. أولاً، قاموا بتغيير كيفية معالجة النموذج للمعلومات في طبقة "التفكير" الخاصة به لتجميع الأسئلة معاً، مما قلل بشكل كبير من كمية العمليات الحسابية المطلوبة. ثان، أضافوا آلية في طبقة فك التشفير تساعد النموذج على التركيز على الميزات الأكثر أهمية مع تجاهل الميزات الأقل فائدة، مما يؤدي إلى شحذ الخطوط النهائية للقرص البصري واللقفة البصرية.
ولتدريب هذا النظام الجديد، استخدم الباحثون مجموعة كبيرة من صور الشبكية تُعرف باسم مجموعة بيانات "REFUGE2"، والتي تحتوي على آلاف مسحات العين مع حدود رسمها الخبراء. كما اختبروا نموذجهم على مجموعة بيانات ثانية تسمى "ORIGA" للتأكد من عمله بشكل جيد على أنواع مختلفة من الصور. كانت النتائج مذهلة؛ ففي مجموعة البيانات الأولى، حدد النموذج الجديد القرص البصري بدقة بلغت 99.45% واللقفة البصرية بدقة بلغت 93.23%، متفوقاً على أفضل الطرق السابقة بأكثر من 2.5%. وفي مجموعة البيانات الثانية، حقق أعلى دقة تم تسجيلها لهذا الاختبار تحديداً. ولعل الأهم من ذلك هو أن النموذج فعال للغاية؛ فهو يتطلب 1.93 مليون معلمة فقط ليعمل، وهو ما يقل بنسبة 96% تقريباً عن النماذج المنافسة الأكثر تعقيداً. ومن حيث القدرة الحسابية، فإنه يستخدم 3.73 جيجا فلوب (GFLOPs) فقط، وهو جزء ضئيل من الطاقة التي تتطلبها الأنظمة الأخرى ذات الأداء العالي. وهذا يعني أن النموذج يمكنه نظرياً العمل على المعدات الطبية القياسية بدلاً من الحاجة إلى خوادم متخصصة وعالية المواصفات.
كما اختبر الباحثون مدى مساهمة كل جزء من تصميمهم في النجاح النهائي. ووجدوا أن إضافة آلية التجميع وميزة طبقة تعزيز التركيز بشكل فردي قد حسنت النتائج، لكن الجمع بينهما أنتج النتيجة الأفضل. كما قدموا طريقة جديدة لتعليم النموذج أثناء التدريب، تقوم بفحص تقدم النموذج في عدة مراحل مختلفة من التعلم بدلاً من مجرد التحقق في النهاية. ساعدت هذه الطريقة النموذج على التعلم بشكل أكثر اتساقاً وإنتاج حدود أكثر سلاسة ودقة. وعندما قارنوا المخرجات المرئية لنموذجهم مع الأنظمة الأخرى، أنتج النموذج الجديد خطوطاً خارجية تتوافق بشكل أكبر بك mucho مع "الحقيقة الأرضية" التي رسمها الخبراء، حتى في الحالات الصعبة حيث تتقاطع الأوعية الدموية مع المنطقة أو تكون الإضاءة غير متساوية. وتخلص الدراسة إلى أنه من خلال المزج الدقيق بين هذه التقنيات المختلفة، من الممكن إنشاء أداة تتسم بالدقة العالية والخفة الكافية للاستخدام على نطاق واسع. يشير هذا التقدم إلى أن الفحص الآلي عالي الجودة للجلوكوما قد يصبح قريباً واقعاً ملموساً في البيئات محدودة الموارد، مما قد ينقذ بصر الكثير من الناس.
ملخص تقني: OptiModNet
بيان المشكلة
يعد التجزئة الدقيقة للقرص البصري (OD) والكأس البصري (OC) في صور قاع العين أمراً بالغ الأهمية للتشخيص المبكر والوقاية من الجلوكوما (المياه الزرقاء)، وذلك بشكل أساسي من خلال حساب نسبة الكأس إلى القرص (CDR). ومع ذلك، فإن تحقيق أداء عالٍ في التجزئة عبر مجموعات بيانات متنوعة مع الحفاظ على تكلفة حوسبة منخفضة يظل تحدياً كبيراً. تواجه النهج الحالية للتعلم العميق مقايضات محددة:
النماذج القائمة على UNet: تتفوق في التقاط الميزات المحلية ولكنها تعاني في نمذجة الاعتمادات طويلة المدى والسياق التشريحي العالمي.
محولات الرؤية (ViTs): تلتقط العلاقات العالمية بفعالية عبر آلية الانتباه الذاتي، ولكنها تسبب تكاليف حوسبة باهظة (تعقيد تربيعي) وتتطلب مجموعات بيانات موسومة ضخمة، مما يجعلها مستهلكة للموارد في تصوير قاع العين عالي الدقة.
البنى الهجينة (مثل TransUNet و UNetR): تحاول سد هذه الفجوة، لكنها غالباً ما ترث عدم الكفاءة الحوسبية للمحولات التقليدية.
النماذج القائمة على الانتشار (Diffusion-based models): تظهر نتائج تنافسية، لكنها تعاني من تكاليف تدريب واستنتاج عالية، مما يحد من ملاءمتها لسير العمل السريري في الوقت الفلي.
وبناءً على ذلك، هناك حاجة إلى بنية هجينة خفيفة الوزن وفعالة توازن بين نمذجة السياق العالمي ودقة الحدود المحلية دون عبء حوسبي باهظ.
المنهجية
يقترح المؤلفون OptiModNet، وهي بنية هجينة جديدة خفيفة الوزن مصم-مة لتجزئة القرص والكأس البصري ثنائي الأبعاد. يدمج النموذج مشفراً (Encoder) قائماً على المحولات مع فك تشفير (Decoder) تلافيفي، معزز بآليات انتباه محددة ودالة خسارة متعددة المراحل.
1. نظرة عامة على البنية
المشفر (قائم على المحولات): يتم تقسيم الصورة المدخلة إلى رقع غير متداخلة وتُدمج مع معلومات موضعية. يتكون المشفر من 12 كتلة محولات متراكمة.
انتباه الاستعلام المجموعاتي (GQA): لتقليل التعقيد الحوسبي مقارنة بآلية الانتباه متعدد الرؤوس القياسية (MSA)، يستخدم المشفر تقنية GQA. تقوم هذه الآلية بتقسيم الاستعلامات إلى مجموعات، حيث تشترك كل مجموعة في زوج واحد من المفاتيح والقيم (key-value pair). يعمل هذا على تبسيط آلية الانتباه، مما يقلل بشكل كبير من العبء الحوسبي مع الحفاظ على القدرة على نمذجة الاعتمادات طويلة المدى في المدخلات عالية الدقة.
فك التشفير (تلافيفي): يقوم فك التشفير بزيادة حجم خرائط الميزات تدريجياً باستخدام التلافيف المنقولة (transposed convolutions) ويدمج اتصالات التخطي متعددة الدقة من المشفر.
انتباه القنوات (CA): مكون مبتكر لفك التشفير، حيث يتم تطبيق انتباه القنوات لإعادة معايرة أهمية الميزات. يستخدم التجميع المتوسط العالمي متبوعاً بطبقات متصلة بالكامل لتوليد أوزان الانتباه، مما يبرز القنوات الميزية المعلوماتية ويخمد القنوات الأقل أهمية. يساعد هذا تحديداً في التحديد الدقيق لحدود القرص والبصري (OD/OC).
2. خسارة الهرم المتجمعة (Aggregated Pyramid Loss)
لتحسين كل من التداخل على مستوى المنطقة والتصنيف على مستوى البكسل، قدم المؤلفون خسارة الهرم المتجمعة.
التركيب: تجمع بين خسارة "دايس" (Soft Dice Loss) وخسارة الإنتروبيا المتقاطعة الثنائية (BCE Loss).
الإشراف متعدد المراحل: على عكس النهج القياسية التي تشرف فقط على المخرجات النهائية، تقوم دالة الخسارة هذه بحساب الخسارة المجمعة في مراحل فك التشفير المتوسطة.
التجميع الموزون: إجمالي الخسارة هو مجموع موزون لخسائر المخرجات المتوسطة والنهائية (∑λkLk). يعزز هذا التصميم تدفق التدرج عبر الشبكة ويضمن الاتساق الهيكلي في تجزئة الحدود الدقيقة.
المساهمات الرئيسية
تحدد الورقة أربع مساهمات رئيسية:
انتباه الاستعلام المجموعاتي في المشفر: دمج GQA لتقليل التعقيد الحسابي مع نمذجة الاعتمادات طويلة المدى بفعالية، مما يعالج عدم كفاءة المحولات القياسية.
انتباه القنوات في فك التشفير: تنفيذ CA للتركيز على القنوات الميزية البارزة، مما يحسن تحديد الحدود بين القرص والبصري والكأس البصري.
خسارة الهرم المتجمعة: دالة خسارة مبتكرة تشرف على مراحل متعددة من فك التشفير لتعزيز تدفق التدرج وتجزئة التفاصيل الدقيقة.
التصميم الهجين الفعال: نموذج يستفيد من نقاط القوة المتكاملة لـ UNet والمحولات وآليات الانتباه لتحقيق دقة عالية مع عدد أقل بكثير من المعلمات (parameters).
النتائج التجريبية
تم تقييم النموذج على مجموعتي بيانات REFUGE2 و ORIGA.
الأداء الكمي
مجموعة بيانات REFUGE2: حقق OptiModNet أداءً رائداً (SOTA)، متفوقاً على الطرق الموجودة بأكثر من 2.5%.
القرص البصري: درجة دايس (Dice Score) بلغت 99.45% و IoU بلغت 98.16%.
الكأس البصري: درجة دايس بلغت 93.23% و IoU بلغت 89.41%.
تجاوزت هذه النتائج أفضل طريقة سابقة (MedSegDiff-V2) بنسبة 2.75% (في درجة Dice) و 2.26% (في IoU) لتجزئة القرص.
مجموعة بيانات ORIGA: حقق النموذج أعلى الدرجات المسجلة للتصنيف ثلاثي الفئات (الكأس البصري، القرص البصري، الخلفية)، مع IoU بنسبة 98.92% ودرجة دايس بنسبة 99.31%.
الكفاءة والتعقيد
المعلمات (Parameters): يستخدم OptiModNet 1.93 مليون معلمة فقط، مما يمثل انخفاضاً بنسبة 96% مقارنة بالنماذج المعقدة مثل MedSegDiff-V2 (46 مليون معلمة).
التكلفة الحوسبية: يتطلب النموذج 3.73 GFLOPs فقط، وهو ما يعادل تقريباً 0.38% من التكلفة الحوسبية لنموذج MedSegDiff-V2 (983 GFLOPs).
دراسات الاستئصال (Ablation Studies)
تحليل المكونات: أدى إزالة إما انتباه القنوات أو انتباه الاستعلام المجموعاتي إلى انخفاض ملموس في الأداء، مما يؤكد أن كلا الوحدتين تساهمان بشكل فعال في الدقة النهائية.
دالة الخسارة: أظهر تغيير الأوزان (λ) في خسارة الهرم المتجمعة أن مخطط الوزن المحدد المستخدم (0.2، 0.3، 0.5 للمراحل الثلاث المتوسطة و 1.0 للمخرج النهائي) قد حقق النتائج المثلى.
الأهمية والادعاءات
تدعي الورقة أن OptiModNet ينجح في معالجة المقايضة بين دقة التجزئة وكفاءة الحوسبة. من خلال دمج انتباه الاستعلام المجموعاتي وانتباه القنوات ضمن إطار عمل UNetR، يحقق النموذج تحديداً فائقاً للحدود ونمذجة للسياق العالمي دون العبء الحوسبي الثقيل للمحولات القياسية أو نماذج الانتشار.
يؤكد المؤلفون أن كفاءة النموذج العالية (انخفاض المعلمات وGFLOPs) تجعله مناسباً بشكل خاص لعمليات الفحص السريع واسع النطاق وللنشر في البيئات السريرية محدودة الموارد، حيث يواجه تصوير قاع العين عالي الدقة غالباً قيوداً في الأجهزة. تشير النتائج إلى أن OptiModNet يوفر حلاً عملياً وعالي الأداء للفحص الآلي للجلوكوما، مما يقلل من التباين بين المراقبين ويسهل التشخيص المبكر.