An Attention-Enhanced MobileViT Framework for Multi-Task Skin Lesion Classification and Segmentation
تقترح هذه الورقة إطار عمل MobileViT-XS خفيف الوزن ومعزز بالانتباه، والذي يوازن بفعالية بين الكفاءة الحسابية والموثوقية التشخيصية للتصنيف والتقسيم المشترك لآفات الجلد، محققاً أداءً تنافسياً على مجموعة بيانات ISIC 2019 بعدد أقل بكثير من المعلمات مقارنة بالنماذج الثقيلة.
في كل عام، يصاب الملايين من الناس حول العالم بآفات جلدية، تتراوح ما بين شامات غير ضارة إلى سرطانات خطيرة مثل الميلانوما (سرطان الجلد). إن اكتشاف هذه الحالات مبكراً هو مسألة حياة أو موت؛ فعندما يتم اكتشاف الميلانوما قبل انتشارها، تكون نسبة البقاء على قيد الحياة لمدة خمس سنوات مثالية تقريباً، ولكن بمجرد وصولها إلى الأعضاء البعيدة، تنخفض تلك الفرصة بشكل حاد. حالياً، غالباً ما يكون خط الدفاع الأول هو طبيب عام في عيادة محلية، وليس طبيب جلدية متخصصاً. يواجه مقدمو الرعاية الأولية هؤلاء واقعاً صعباً: إذ يتعين عليهم رصد العلامات الدقيقة للسرطان في صور الجلد دون امتلاك سنوات التدريب المتخصص التي يمتلكها طبيب الجلدية، كما أنهم غالباً ما يفتقرون إلى الأنظمة الحاسوبية الضخمة والمكلفة التي تشغل أكثر أدوات التشخيص تقدماً. التحدي إذن، يكمن في بناء مساعد رقمي ذكي بما يكفي ليرى ما يراه المتخصص، وفي الوقت نفسه خفيف بما يكفي ليعمل على الحواسيب المتواضعة الموجودة في عيادة قياسية أو حتى على جهاز محمول.
لقد عالج فريق من الباحثين هذا التحدي من خلال إنشاء نوع جديد من أنظمة الذكاء الاصطناعي المصممة خصيصاً لهذه البيئة. لقد بنوا إطار عمل يمكنه النظر إلى صورة لآفة جلدية والقيام بشيئين في آن واحد: تحديد نوع الآفة ورسم مخطط دقيق حولها. جوهر نظامهم هو شبكة عصبية خفيفة الوزن، وهي نوع من البرامج الحاسوبية المستوحاة من الدماغ البشري، والتي اختاروها لقدرتها على تحقيق التوازن بين الأداء العالي واستهلاك الطاقة المنخفض. ولجعل هذا النظام أكثر دقة، أضافوا آلية تساعد الحاسوب على التركيز على الأجزاء الأكثر أهمية في الصورة، تماماً كما تتجاهل العين البشرية طبيعياً الخلفية المزدحمة للتركيز على جسم معين. كما صمموا النظام ليكون حذراً؛ فإذا كان الحاسوب غير متأكد من التشخيص، فإنه مبرمج لتحديد الحالة لعرضها على طبيب بشري بدلاً من التخمين بثقة، وهي ميزة سلامة تهدف إلى منع الأخطاء الخطيرة.
اختبر الباحثون ابتكارهم باستخدام مجموعة كبيرة من صور الجلد من قاعدة بيانات طبية عامة. وأظهرت النتائج أن نظامهم خفيف الوزن يمكنه تحديد نوع الآفة الجلدية بشكل صحيح في أكثر من 84 بالمائة من الحالات، وهو مستوى من الدقة ينافس النماذج الأكبر والأكثر تعقيداً التي تتطلب أجهزة قوية ومكلفة. بالإضافة إلى تصنيف الآفة، نجح النظام في رسم حدود النمو في الصورة بدرجة عالية من الدقة. ولعل الأهم من ذلك، أن الدراسة أظهرت أن تعليم الحاسوب القيام بالمهمتين في وقت واحد — تحديد الآفة ورسم شكلها — جعل النظام في الواقع أفضل في تحديد نوع الآفة مما لو تم تدريبه على القيام بواحدة من هاتين المهمتين فقط. وهذا يشير إلى أن المهمتين تساعدان بعضهما البعض، حيث أن عملية رسم مخطط للآفة تساعد الحاسوب على فهم سماتها بشكل أعمق.
وعلى الرغم من هذه النجاحات، يحرص الباحثون على الإشارة إلى أن نظامهم ليس جاهزاً بعد ليحل محل الطبيب. فقد تم تدريب النموذج على صور التقطت في ظروف مثالية ومنضبطة، وقد يواجه صعوبة مع الإضاءة والزوايا المتنوعة وغير المنتظمة الموجودة في العيادات الواقعية أو في فحوصات العناية الذاتية المنزلية. علاوة على على ذلك، فإن النظام مصمم ليكون أداة فحص، وليس سلطة تشخيصية نهائية. فعندما يواجه الحاسوب صورة صعبة أو غامضة، فإن بروتوكول السلامة المدمج فيه يوجهه لإرسال الحالة إلى خبير بشري للمراجعة. ويتصور الفريق أن هذه التكنولوجيا هي وسيلة لتمكين الممارسين العامين، عبر منحهم رأياً ثانياً موثوقاً يمكن أن يساعد في رصد سرطانات الجلد الخطيرة في وقت مبكر، حتى في المناطق التي تندر فيها موارد المتخصصين. ومع ذلك، وقبل أن يتمكن هذا النوع من الأدوات من الاستخدام في المستشفيات، يجب أن يخضع لمزيد من الاختبارات في بيئات سريرية حقيقية لضمان عمله بأمان عبر مختلف أنواع البشر والحالات الطبية.
ملخص تقني: إطار عمل MobileViT معزز بالانتباه لتصنيف وتقسيم آفات الجلد متعدد المهام
1. بيان المشكلة
يتناول البحث تحدي نشر التشخيص الآلي لآفات الجلد في بيئات الرعاية الأولية محدودة الموارد. وبينما حقق التعلم العميق تقدماً في تحليل آفات الجلد، إلا أن هناك مقايضة كبيرة قائمة بين دقة التشخيص والكفاءة الحسابية.
النماذج الثقيلة (Heavyweight models): (مثل الـ Transformers القياسية، وU-Nets الضخمة) تقدم أداءً قوياً ولكنها تمتلك أعباءً مفرطة في عدد المعلمات (parameters)، مما يجعلها غير مناسبة للأجهزة المحمولة أو أجهزة الحافة (edge devices).
الشبكات خفيفة الوزن الموجودة (Existing lightweight networks): غالباً ما تفتقر إلى نمذجة الميزات العالمية القوية وتعزيز الميزات المكانية، مما يحد من موثوقيتها للمهام الدقيقة مثل التمييز بين حدود الآفات الغامضة.
المخاوف المتعلقة بالسلامة: معظم النماذج الحالية تقدم تنبؤات حتمية دون تحديد كمي صريح لعدم اليقين، مما قد يؤدي إلى مخرجات مفرطة في الثقة عند التعامل مع عينات غامضة أو غير نمطية، وهو ما يزيد من مخاطر التشخيص الخاطب سريرياً.
2. المنهجية
يقترح المؤلفون إطار عمل خفيف الوزن ومعزز بالانتباه متعدد المهام يعتمد على MobileViT-XS للقيام بالتقسيم الدلالي وتصنيف آفات الجلد بشكل مشترك.
بنية العمود الفقري (Backbone Architecture): يستخدم إطار العمل MobileViT-XS كعمود فقري مشترك. تجمع هذه البنية بين الشبكات العصبية الالتفافية (CNNs) لاستخراج الأنسجة المحلية، ومحولات الرؤية (ViTs) لنمذجة الاعتماد العالمي، مما يلتقط بفعالية الخصائص متعددة المقاييس والحدود الغامضة للآفات الجلدية.
آلية الانتباه (CBAM): تم دمج وحدة الانتباه الالتفافية (Convolutional Block Attention Module - CBAM) في نهاية العمود الفقري، قبل رؤوس المهام المحددة. تطبق CBAM بشكل متسلسل انتباه القنوات (تحديد القنوات المعلوماتية عبر التجميع العالمي وMLP) والانتباه المكاني (تجميع معلومات القنوات لتوليد قناع مكاني) لتنقية تمثيلات الميزات.
تصميم التعلم متعدد المهام (MTL): تتميز البنية بهيكل ثنائي المسار:
فرع التصنيف (Classification Branch): يخرج احتمالات لثماني فئات من الآفات (AK, BCC, BKL, DF, MEL, NV, SCC, VASC) باستخدام التجميع المتوسط العالمي، والإسقاط (dropout)، وطبقة متصلة بالكامل.
فرع التقسيم (Segmentation Branch): يخرج قناعاً للآفة على مستوى البكسل باستخدام فك تشفير (decoder) خفيف الوزن يقوم برفع دقة الميزات وتطبيق الالتفافات.
الترقية غير المتماثلة (Asymmetric Promotion): يستفيد التصميم من الإشراف على التقسيم على مستوى البكسل لمساعدة تحسين التصنيف على مستوى الصورة، مما يخلق "نموذج ترقية مهام غير متماثل" حيث يساعد التقسيم التصنيف بشكل أكبر مما يفعل العكس.
دالة الخسارة (Loss Function): إجمالي الخسارة هو مجموع موزون لخسائر التصنيف والتقسيم (Ltotal=λclsLcls+λsegLseg).
خسارة التصنيف: تم استخدام خسارة الإنتروبيا المتقاطعة الموزونة لمعالجة عدم توازن الفئات، مع أوزان تتناسب عكسياً مع تكرار الفئة.
خسارة التقسيم: تم استخدام مزيج من خسارة Dice وخسارة الإنتروبيا المتقاطعة الثنائية (BCE) لتحسين التداخل الإقليمي والدقة على مستوى البكسل.
الاستدلال التحفظي وتصنيف المخاطر (Conservative Inference & Risk Stratification): للتخفيف من فرط الثقة، يتم استخدام استراتيجية استدلال تعطي الأولوحة للسلامة.
عتبات المراجعة: يتم وضع علامة على العينات للمراجعة من قبل الخبراء إذا كانت أقصى احتمالية للفئة (pmax) أقل من 0.50 أو إذا كان الفارق (δ) بين أعلى فئتين أقل من 0.15.
تصنيف المخاطر: بالنسبة للعينات غير المراجعة، يتم حساب الاحتمال التراكمي للفئات الخبيثة (MEL, BCC, SCC, AK). تُصنف المخاطر إلى: عالية (متوقع أن تكون خبيثة)، متوسطة (تنبؤ حميد ولكن pmalignant≥0.25)، ومنخفضة.
3. المساهمات الرئيسية
التحقق من صحة MobileViT-XS: تثبت الدراسة بشكل منهجي أن MobileViT-XS يعد عموداً فقرياً خفيف الوزن متفوقاً لتحليل آفات الجلد، حيث يظهر قدرته على موازنة الأنسجة المحلية والسياق العالمي بشكل أفضل من الـ CNNs النقية (مثل EfficientNet-B0) أو الـ Transformers الضخمة.
نموذج المهام المتعددة المعزز بالانتباه: يضع هذا العمل إطار عمل تعاوني حيث يعمل الإشراف على التقسيم على تنقية الميزات المكانية للتصنيف بشكل كبير. ويوضح أن وحدة CBAM تحقق مكاسب كبيرة في الأداء تحديداً عند اقترانها بالتعلم متعدد المهام، متجاوزة بذلك مكاسب الانتباه المحدودة الملاحظة في أطر العمل أحادية المهمة.
التنبؤ الموجه نحو السلامة: يعالج إدخال قيد التنبؤ التحفظي وآلية تصنيف المخاطر مشكلة فرط الثقة في النموذج، من خلال وضع علامات على العينات غير المؤكدة لمراجعتها بشرياً لتعزيز سلامة التشخيص.
4. النتائج التجريبية
أُجريت التجارب على مجموعة بيانات ISIC 2019 (25,331 صورة، 8 فئات) مع أقنعة تقسيم مطابقة من ISIC 2018.
مقاييس الأداء:
التصنيف: حقق النموذج المقترح دقة بنسبة 84.37%، وF1-macro بنسبة 79.46%، وحساسية بنسبة 79.68%، ونوعية (Specificity) بنسبة 97.22%.
التقسيم: حقق النموذج معامل Dice بنسبة 86.37% وmIoU بنسبة 78.09%.
الكفاءة: يعمل النموذج بـ 2.62 مليون معلمة فقط.
التحليل المقارن:
تفوقت الطريقة المقترحة على النماذج الأساسية خفيفة الوزن مثل EfficientNet-B0 (دقة 81.63%) وتفوقت بشكل ملحوظ على ResNet-50 وGS-TransUNet.
بينما حققت نماذج التقسيم أحادية المهمة (U-Net, DeepLabV3+) درجات أعلى قليلاً في Dice/mIoU، إلا أن النموذج متعدد المهام المقترح يقدم مقايضة متفوقة بين الدقة والكفاءة من خلال أداء المهمتين في آن واحد بأقل عدد من المعلمات.
دراسات الاستئصال (Ablation Studies):
أدى إضافة CBAM إلى نموذج MobileViT-XS الأساسي أحادي المهمة إلى مكاسب طفيفة.
أدى إدخال التعلم متعدد المهام (MTL) إلى تعزيز دقة التصنيف بشكل كبير (من ~72% إلى ~83.6%).
أدى الجمع بين MTL وCBAM إلى تحقيق النتائج المثلى، مما يؤكد أن إشراف التقسيم الكثيف ينشط الإمكانات الكاملة لآلية الانتباه.
5. الأهمية والادعاءات
يدعي البحث تقديم مقايضة متوازنة جيداً بين الدقة، والكفاءة الحسابية، والقابلية للتفسير البصري، وسلامة التنبؤ السريري.
القابلية للتطبيق في الرعاية الأولية: يتم وضع إطار العمل كحل خفيف الوزن وقابل للنشر للفرز الأولي لآفات الجلد ولأجهزة الحافة المحمولة، حيث تكون الموارد الحسابية محدودة.
القابلية للتفسير: يعزز دمج خرائط الحرارة من نوع Grad-CAM والتصورات المتراكبة من قابلية تفسير النموذج، مما يسم يسمح للأطباء بفهم الأسباب الكامنة وراء التنبؤات.
السلامة: من خلال التعامل الصريح مع عدم اليقين عبر عتبات تحفظية، يهدف النظام إلى تقليل خطر التنبؤات الخطيرة المفرطة في الثقة في الحالات الغامضة.
القيود والنظرة المستقبلية: يقر المؤلفون بأن النموذج تم تدريبه حصرياً على مجموعات بيانات ISIC العامة تحت ظروف تصوير محكومة، مما قد يؤدي إلى تحولات في النطاق (domain shifts) في سيناريوهات الرعاية الأولية أو الفحص الذاتي في العالم الحقيقي. كما أن وزن الخسارة الحالي ثابت، وعتبات التنبؤ محددة تجريبياً. ويخلص البحث إلى أن التحقق السريري متعدد المراكز المستقبلي مطلوب قبل إمكانية تحقيق التطبيق السريري في العالم الحقيقي. سيركز العمل المستقبلي على جمع مجموعات بيانات سريرية متنوعة، واستكشاف آليات موازنة المهام الديناميكية، وتنفيذ النشر عبر Mobile/Edge باستخدام ONNX أو TFLite.