Post-hoc reliability assessment and visual explanation for deep learning-based oral mucosal lesion detection
تقدم هذه الورقة إطار عمل لاحقاً (post-hoc) يعزز موثوقية وقابلية تفسير نموذج كشف YOLOv11 مجمد لآفات الغشاء المخاطي للفم من خلال تحليل كثافة تضمين الميزات لتحديد التنبؤات غير الموثوقة واستخدام استرجاع أقرب الجيران لتقديم تفسيرات بصرية، وكل ذلك دون الحاجة إلى إعادة تدريب النموذج.
المؤلفون الأصليون:Ekaterina Tolstaya, Julia Schwärzler, Dinara Kayumova, Stefan Kolek, Antonin Tichy, Sebastian Paris, Gitta Kutyniok, Falk Schwendicke
في عالم التصوير الطبي، أصبح الذكاء الاصطناعي أداة قوية لرصد الأمراض التي قد تخطئها العين البشرية. تُدرب هذه الأنظمة على آلاف الصور، لتتعلم كيفية التعرف على الأنماط التي تشير إلى المرض. ومع ذلك، تظل هناك مشكلة كبيرة: فعندما تقدم هذه البرامج الحاسوبية تنبؤاً، فإنها غالباً ما تقدم درجة ثقة قد تكون مضللة. فقد يعلن النظام عن تشخيص بنسبة يقين تصل إلى 99 بالمائة بينما ينظر إلى صورة تختلف تماماً عن أي شيء رآه من قبل. وفي المجالات الحساسة للسلامة مثل الطب، حيث يمكن أن يؤدي التخمين الخاطئ إلى تأخير العلاج المنقذ للحياة، فإن معرفة متى يمكن الوثوق بالآلة لا تقل أهمية عن قدرة الآلة على الرؤية نفسها. ويكمن التحدي في إيجال طريقة للتحقق من موثوقية التنبؤ دون الحاجة إلى إعادة بناء النظام بأكره.
لقد تصدى باحثون من عدة مؤسسات ألمانية ودولية لهذه القضية عبر تطوير طريقة جديدة لتدقيق قرارات ذكاء اصطناعي مصمم للكشف عن آفات الغشاء المخاطي للفم. وهذه الآفات هي حالات تصيب الأنسجة الرخوة داخل الفم، وتتراوح بين البقع البيضاء غير الضارة والنمو السرطاني المحتمل. ركز الفريق على نوع محدد من كواشف الذكاء الاصطناعي يسمى "YOLO"، وهو ممتاز في العثور على الأشياء في الصور ورسم مربعات حولها. وبينما كان الكاشف مدرباً بالفعل على تحديد ثلاثة أنواع من الآفات والأنسجة السليمة، أراد الباحثون فهم "السبب" وراء كل قرار وتحديد ما إذا كان الذكاء الاصطناعي واثقاً حقاً أم أنه مجرد تخمين. لم يعيدوا تدريب النموذج أو تغيير كوده الداخلي، بل نظروا بدلاً من ذلك إلى البصمات الرياضية، المعروفة باسم "التضمينات" (embeddings)، التي أنشأها الذكاء الاصطناعي لكل صورة قام بتحليلها. تمثل هذه البصمات الميزات البصرية التي استخدمها الذكاء الاصطناعي لاتخاذ قراره.
قام الباحثون بأخذ هذه البصمات عالية الأبعاد ورسم خرائط لها على مشهد بسيط ثنائي الأبعاد. على هذه الخريطة، تتجمع الصور التي تبدو متشابهة جداً مع بعضها البعض، بينما تشكل أنواع الآفات المختلفة مجموعات متميزة خاصة بها. ومن خلال تحليل مدى ازدحام أو ندرة منطقة معينة على هذه الخريطة، استطاعوا حساب درجة الموثوقية. فإذا وقع تنبؤ الذكاء الاصطناعي في تجمع كثيف من الصور التدريبية المتشابهة والمحددة بشكل صحيح، اعتبر النظام التنبؤ موثوقاً. أما إذا وقع التنبؤ في منطقة نادرة وفارغة بعيدة عن أي أمثلة معروفة، فإنه يصنف التنبؤ كونه غير مؤكد. سمح هذا النهج بتقييم مدى موثوقية كل عملية كشف دون تغيير الكاشف الأصلي. وقد وجدوا أن درجة الكثافة هذه كانت أفضل بكثير في تحديد التنبؤات غير المؤكدة من أرقام الثقة القياسية التي يوفرها الذكاء الاصطناعي عادةً. وفي اختبارات شملت أكثر من 3,500 صورة، نجحت هذه الطريقة في التمييز بين التنبؤات الموثوقة وغير الموثوقة بمساحة تحت المنحنى (AUC) تتراوح بين 0.92 و0.95 عبر فئات الأمراض المختلفة.
ولجعل هذه الدرجات المجردة مفيدة للأطباء، أضاف الفريق أداة تفسير بصري. فعندما يصنف النظام تنبؤاً ما على أنه غير مؤكد، يمكنه استرجاع وعرض الصور الأربع الأكثر تشابهاً من قاعدة بياناته التدريبية فوراً. خلق هذا الأمر فسيفساء تظهر للطبيب بالضبط ما كان يقارن به الذكاء الاصطناعي حالة المريض الحالية. فعلى سبيل المثال، إذا أخطأ الذكاء الاصطناعي في تحديد نوع معين من البقع البيضاء، فإن التفسير البصري سيظهر أن النظام كان يقارن المريض بحالات أخرى مشابهة بصرياً ولكنها مختلفة. سمح هذا للأطباء برؤية عملية التفكير الخاصة بالذكاء الاصطناعي وهي تعمل. وكشفت الدراسة أن الذكاء الاصطناعي غالباً ما يعاني مع الآفات التي تتشابه فيما بينها، مثل بعض أشكال "اللوحاق" (leukoplakia) و"الحزاز المسطح" (lichen planus)، والتي تشترك في ميزات بصرية متداخلة. وفي هذه الحالات الحدية، حددت الطريقة الجديدة بشكل صحيح أن الذكاء الاصطناعي يعمل في منطقة من الغموض، في حين أن درجة الثقة القياسية كانت ستظل مرتفعة بشكل مضلل.
كما اختبر الباحثون النظام على صور لآفات لم تكن جزءاً من مجموعة التدريب الأصلية، مثل الأمراض المسببة للفقاعات أو التقرن الاحتكاكي. ووجدوا أن الذكاء الاصطناعي يخصص أحياناً هذه الحالات غير المعروفة لأقرب فئة معروفة، وأن درجة الكثافة لم تكن دائماً تصنفها كحالات غير معروفة تماماً. وهذا يسلط الضضوء على قصور حاسم: فالنظام يقيس مدى ملاءمة التنبؤ لبيانات التدريب، وليس ما إذا كانت الصورة تمثل مرضاً جديداً تماماً. ومع ذلك، أثبتت أداة الاسترجاع البصري قيمتها هنا أيضاً؛ فحتى عندما يرتكب الذكاء الاصطناعي خطأً، فإن الجيران التدريبيين المعروضين يساعدون الخبير البشري على إدراك أن التشابه البصري كان مضللاً وأن التشخيص يتطلب ما هو أكثر من مجرد صورة فوتوغرافية. وتخلص الدراسة إلى أنه من خلال الجمع بين فحص الموثوقية القائم على الكثافة وبين الأمثلة البصرية، يمكن للمتخصصين الطبيين فهم متى يمكنهم الوثوق بتشخيص الذكاء الاصطناعي ومتى يجب عليهم طلب رأي ثانٍ، وكل ذلك دون الحاجة إلى إعادة تدريب النماذج المعقدة المستخدمة بالفعل.
ملخص تقني: تقييم الموثوقية البعدي والتفسير البصري للكشف عن آفات الغشاء المخاطي الفموي القائم على التعلم العميق
بيان المشكلة تنتج أدوات الكشف القائمة على التعلم العميق لآفات الغشاء المخاطي الفموي، مثل تلك القائمة على بنيات YOLO، تنبؤات بالفئات ودرجات ثقة (softmax)؛ ومع ذلك، تفشل هذه المخرجات في الكشف عن الأنماط البصرية التي تدعم التنبؤ أو في الإشارة إلى ما إذا كان تنبؤ معين موثوقاً به أم لا. في المجالات الطبية الحرجة للسلامة، من المعروف أن درجات (softmax) تفتقر إلى المعايرة الجيدة وغالباً ما تكون مفرطة في الثقة عند التعامل مع مدخلات تختلف عن توزيع بيانات التدريب (المدخلات خارج التوزيع أو OOD). تتطلب أساليب تقدير عدم اليقين الحالية، مثل إسقاط "مونت كارلو" (Monte-Carlo dropout) أو الشبكات العصبية البايزية، تعديلات هيكلية أو عمليات تمرير متعددة للأمام، مما يجعلها غير مناسبة لأدوات الكشف المدربة مسبقاً والمجمدة. كما أظهرت مناهج التنبؤ المطابق (conformal prediction) القياسية قيوداً في الموثوقية تحت تأثير التحولات التوزيعية وفي البيئات الطبية متعددة الفئات. ويتمثل التحدي في تطوير منهجية "بعدية" (post-hoc) لتقييم الموثوقية وتوفير تفسيرات بصرية دون إعادة تدريب النموذج أو تغيير بنيته.
المنهجية يقترح المؤلفون إطار عمل بعدي يُطبق على كاشف YOLOv11 مجمد تم تدريبه على 3,104 صورة داخل الفم تغطي الحزاز المسطح الفموي (OLP)، والتحول الورمي المتصلب/الورم الأبيض (PVL/Leukoplakia)، وسرطان الخلايا الحرشفية الفموية (OSCC)، والغشاء المخاطي السليم. تتكون المنهجية من ثلاث مراحل رئيسية:
1. استخراج التضمينات (Embedding Extraction):
يتم استخراج تضمينات السمات من أعمق طبقة في العمود الفقري (Backbone) لنموذج YOLOv11 (الطبقة 21، "C2f") لكل صندوق إحاطة مكتشف.
تمت مقارنة أربع استراتيجيات استخراج: نقطة المركز (center-point)، والقمة الواحدة (من خرائط بروز EigenCAM)، وتجميع الأوزان المعتمد على CAM، ومتوسط الصندوق (box-averaging).
تم استخدام EigenCAM لتوليد خرائط البروز دون الحاجة إلى الانتشار العكسي للتدرج (gradient backpropagation)، مما يسمح بتحديد المناطق المكانية الأكثر معلوماتية داخل صندوق الإحاطة.
تم اختيار متوسط الصندوق (Box-averaging) (المتوسط الموحد لجميع ناقلات السمات داخل الصندوق المكتشف) كالإستراتيجية المثلى بناءً على التقييم الكمي، حيث حققت أكثر التجمعات الفئوية إحكاماً وتباعداً في فضاء التضمين.
2. إسقاط المتشعب وتقدير الكثافة (Manifold Projection and Density Estimation):
يتم إسقاط التضمينات عالية الأبعاد (384 بُعداً) على متشعب ثنائي الأبعاد لتسهية نمذجة الكثافة والتصور.
تمت مقارنة طريقتين للإسقاط: تحليل المكونات المجاورة (NCA) (خاضع للإشراف) و (UMAP) (غير خاضع للإشراف).
يتم تطبيق تقدير كثافة النواة (KDE) على عينات التدريب التي تم التنبؤ بها بشكل صحيح لكل فئة ضمن الفضاء المسقط ثنائي الأبعاد.
لأي كشف جديد، يتم حساب درجة كثافة بناءً على نماذج KDE المناسبة لكل فئة؛ وتعمل هذه الدرجة كوكيل لـ "دعم توزيع التدريب"، مما يشير إلى ما إذا كان الكشف يقع في منطقة كثيفة من الأمثلة المعروفة والمصنفة بشكل صحيح، أم في منطقة نادرة أو حدودية.
3. التفسير البنائي عبر استرجاع أقرب الجيران (Visual Explanation via Nearest-Neighbor Retrieval):
لاستكمال درجة الكثافة القياسية، يقوم النظام بعملية استرجاع أقرب k جار (k-NN) في فضاء التضمين.
بالنسبة لكشف اختبار، يقوم النظام باسترجاع أفضل k حالة تدريب متشابهة بصرياً (بناءً على مسافة جيب التمام/cosine distance).
يولد هذا "فسيفساء استرجاع" (retrieval mosaic) تسمح للأطباء بمراجعة منطق النموذج بصرياً من خلال مقارنة الصورة المستعلم عنها مع حالات التدريب الأكثر تشابهاً.
النتائج الرئيسية
أداء الكشف: حقق نموذج YOLOv11 درجات F1 لكل فئة بلغت 0.88 (OLP)، و0.76 (PVL/Leukoplakia)، و0.73 (OSCC)، و0.93 (Healthy).
إستراتيجية التضمين: تفوق "متوسط الصندوق" على استراتيجيات الاستخراج الأخرى، محققاً درجة "سيلويت" (silhouette score) قدرها 0.45 (مقارنة بـ 0.26 لنقطة المركز) ودقة أعلى في استرجاع أقرب جار (Precision@5 = 0.84).
تقييم الموثوقية: أظهرت درجات الموثوقية القائمة على الكثافة قدرة تمييزية قوية، حيث تراوحت قيم المساحة تحت منحنى خصائص التشغيل المستقبلي (AUROC) بين 0.92 و0.95 عبر جميع الفئات الأربع.
تفوقت إسقاطات NCA قليلاً على UMAP في قيم AUROC لبعض الفئات (مثل OSCC: 0.94 مقابل 0.94؛ وPVL: 0.94 مقابل 0.92)، رغم أن UMAP قدم تمثيلاً أكثر دقة لهندسة النموذج المتعلمة دون تحيز للعلامات (labels).
حددت درجة الكثافة باستمرار التنبؤات منخفضة الثقة التي فشلت درجة (softmax) القياسية في رصدها.
التفسيرات البصرية: نجح استرجاع أقرب الجيران في توضيح حدود القرار للنموذج. في حالات التصنيف الخاطئ (مثل توقع PVL على أنه OLP)، كانت الجيران المسترجعة تشبه بصرياً الفئة المتوقعة، مما أكد أن خطأ النموذج كان مبنياً على التشابه البصري وليس على ضجيج عشوائي. كما سلطت الطريقة الضوء على القيود، مثل عدم القدرة على استعادة الفرضيات المتنافسة التي تم قمعها بواسطة عملية "إزالة التكرار غير الأقصى" (NMS).
الأهمية والادعاءات يدعي البحث سد فجوة في الأدبيات من خلال تكييف الكشف عن البيانات خارج التوزيع (OOD) القائم على التضمين، والذي تمت دراسته سابقاً للمصنفات الصورية، ليناسب كواشف الكائنات أحادية المرحلة. وتتمثل المساهمات الرئيسية فيما يلي:
القابلية للتطبيق البعدي: تعمل الطريقة على كاشف مجمد دون الحاجة لإعادة التدريب، أو تغييرات هيكلية، أو بيانات تدريب إضافية (مثل عينات الوكيل لـ OOD).
القابلية للتفسير: توفر مخرجاً مزدوجاً: درجة موثوقية كمية (الكثافة) وتفسيراً بصرياً نوعياً (فسيفساء أقرب جار)، مما يدعم المراجعة السريرية.
دعم توزيع التدريب: تقيس درجة الكثافة مدى دعم التنبؤ بتوزيع التدريب، مما يوفر حماية ضد التنبؤات مفرطة الثقة في المدخلات غير النمطية.
الفائدة السريرية: بينما يؤكد المؤلفون أن درجة الكثافة ليست أداة تشخيصية قائمة بذاتها (لأنها تفتقر إلى السياق السريري مثل اللمس أو التاريخ المرضي)، إلا أنها تعمل كأداة فرز قيمة؛ حيث يمكنها تحديد التنبؤات منخفضة الدعم لمراجعة المتخصصين، وتساعد في تنقية البيانات عبر تحديد العينات غير النمطية أو التي قد تكون مصنفة بشكل خاطئ في التدريب.
يخلص المؤلفون إلى أن تحليل كثافة فضاء التضمين، مقترناً بالاسترجاع البصري، يكمل ثقة (softmax) ويدعم المراجعة السريرية البعدية للكشف عن آفات الغشاء المخاطي الفموي.