A Deep Learning Framework for Multi-Class Lung Disease Detection Using CXR Imaging
تقترح هذه الدراسة إطار عمل للتعلم العميق مدركاً للانحياز، يستخدم مجموعة بيانات صور أشعة سينية للصدر (CXR) متعددة المصادر، منسقة ومعالجة مسبقاً بشكل موحد، وفحصاً منطقياً للتنبؤ بالمصدر للتخفيف من انحياز أصل مجموعة البيانات، مما يثبت أن نموذج ResNet-38 مخصص يحقق دقة فائقة في اكتشاف أمراض الرئة متعددة الفئات تبلغ حوالي 93% مقارنة ببنيات ResNet الأكثر عمقاً.
المؤلفون الأصليون:Neha Sehgal, Md. Arquam, Tayyab Khan, Akanksha Mrinali
تُعد الرئتان البشريتان شبكة معقدة من الممرات الهوائية والأكياس، وهي ضرورية لكل نفس نأخذه. وعندما يتعرض هذا النظام الدقيق للعدوى أو الالتهاب أو النمو غير الطبيعي، يمكن أن تكون العواقب وخيمة وسريعة. لعقود من الزمن، اعتمد الأطباء على صور الأشعة السينية للصدر، وهي تقنية تصوير سريعة ومنخفضة التكلفة، للنظر داخل الصدر ورصد هذه المشكلات. ومع ذلك، فإن قراءة هذه الصور مهمة شاقة؛ فغالباً ما تبدو الأمراض المختلفة متشابهة بشكل صارخ في الأشعة السينية، كما أن الحجم الهائل للصور التي تنتجها المستشفيات يمكن أن يرهق حتى أمهر أخصائيي الأشعة. وهنا يأتي دور الذكاء الاصطناعي، ليقدم وعداً بكونه عيناً ثانية لا تمل. والهدف هو تعليم الحواسيب التعرف على الأنماط الدقيقة للمرض، ولكن لا تزال هناك عقبة رئيسية: ضمان أن تتعلم هذه الحواسيب من العلامات الفعلية للمرض بدلاً من الأدلة العرضية المخفية في البيانات نفسها.
قام فريق من الباحثين من الهند بتطوير نهج جديد لهذا التحدي، حيث ابتكروا نظاماً مصمماً للكشف عن ست حالات رئوية مختلفة من صور الأشعة السينية بموثوقية عالية. يركز عملهم على مشكلة حرجة في الذكاء الاصطناعي الطبي: وهي أن مجموعات البيانات غالباً ما تكون منحازة. فإذا تم تدريب الحاسوب على صور من مستشفيات مختلفة، فقد يتعلم بالخطأ تحديد المستشفى أو الجهاز المستخدم في التقاط الصورة، بدلاً من تحديد المرض. ولمنع حدوث ذلك، بنى الباحثون إطار عمل يعامل جميع الصور بنفس الطريقة، مما يجردها من أي تلميحات حول مصدرها. لقد جمعوا أكثر من 13,000 صورة من مصادر عامة متنوعة، تغطي حالات مثل الالتهاب الرئوي، والسل، وسرطان الرئة، والرئة السليمة الطبيعية. ثم قاموا بموازنة البيانات بعناقة، لضمان رؤية الحاسوب لأمثلة كافية عن الأمراض النادرة لتعلمها بشكل صحيح، تماماً كما رأى الأمراض الشائعة.
اختبر الباحثون ثلاث نسخ مختلفة من بنية ذكاء اصطناعي شهيرة تُعرف باسم "ResNet"، والتي تعمل بمثابة عقل النظام. تباينت هذه النسخ في عمقها، أو تعقيدها؛ حيث كانت إحداها شبكة عميقة ومعقدة مكونة من 101 طبقة، والأخرى مكونة من 50 طبقة، والثالثة ذات تصميم أكثر إيجازاً مكون من 38 طبقة. وغالباً ما تشير الحكمة التقليدية في الذكاء الاصطناعي إلى أن الشبكات الأعمق والأكثر تعقيداً هي الأفضل دائماً لأنها تستطيع تعلم ميزات أكثر تفصيلاً. ومع ذلك، وجدت هذه الدراسة العكس تماماً بالنسبة لهذه المهمة الطبية؛ إذ كان النموذج الأكثر نجاحاً هو ذا الـ 38 طبقة، حيث حقق دقة تقارب 93 بالمائة، محددًا المرض بشكل صحيح في الغالبية العظمى من الحالات. أما النماذج الأعمق، رغم قوتها، فقد كان أداؤها أقل قليلاً، حيث سجلت دقة بلغت حوالي 87 و89 بالمائة. ويشير الباحثون إلى أنه بالنسبة لحجم وطبيعة مجموعة البيانات الطبية هذه، فإن التعقيد الإضافي للشبكات الأعمق قد أدى إلى جعلها تحفظ صور التدريب بدلاً من تعلم القواعد العامة للمرض، وهي ظاهرة تُعرف باسم "الإفراط في التخصيص" (overfitting).
ولضمان أن النظام ينظر حقاً إلى الرئتين وليس مجرد التخمين بناءً على ضوضاء الخلفية، استخدم الباحثون فحصاً ذكياً؛ حيث حاولوا تدريب نموذج لتحديد المستشفى أو قاعدة البيانات التي تنتمي إليها الصورة. فشل النموذج في هذه المهمة، وكان أداؤه بالكاد أفضل من الصدفة العشوائية. وكان هذا الفشل في الواقع نجاحاً للدراسة؛ فقد أثبت أن الصور قد تم توحيدها بنجاح، وأن نظام تصنيف الأمراض كان يتعلم من الاعتلال المرضي الفعلي، وليس من مصدر البيانات. علاوة على ذلك، استخدم الفريق تقنية تصور تسمى "Grad-CAM"، والتي تسلط الضوء على المناطق المحددة في الأشعة السينية التي ركز عليها الحاسوب لاتخاذ قراره. وقد راجع خبير طبي مؤهل هذه التوضيحات وأكد أن الحاسوب كان ينظر بالفعل إلى الأجزاء ذات الصلة من الرئتين، مثل البقع الضبابية للالتهاب الرئوي أو الأشكال المتميزة للأورام، بدلاً من العيوب العشوائية.
كما أخضعت الدراسة النتائج لاختبارات إحصائية صارمة لضمان أن النتائج ليست محض صدفة. ومن خلال تقسيم البيانات مراراً وتكراراً إلى مجموعات تدريب واختبار مختلفة، أكدوا أن أداء نموذج الـ 38 طبقة ظل مستقراً ومتسقاً. كما اختبروا النظام على مجموعة منفصلة تماماً من الصور التي لم يسبق له رؤيتها من قبل، وحافظ على مستوى عالٍ من الدقة، مما أظهر قدرته على تعميم معرفته على مرضى جدد. وخلص الباحثون إلى أنه في مجال التصوير الطبي، ليس الأكبر هو الأفضل دائماً؛ فنموذج متوسط الحجم ومضبوط بعناية، مدعوم بعملية تدريب واعية بالانحياز، يمكن أن يتفوق على نظائره الأعمق والأكثر تعقيداً. وتقدم هذه النتيجة مساراً عملياً للمضي قدماً في تطوير أدوات ذكاء اصطناعي موثوقة يمكنها مساعدة الأطباء في تشخيص أمراض الرئة بسرعة ودقة، مما قد ينقذ الأرواح عبر اكتشاف هذه الحالات في مراحل مبكرة.
ملخص تقني: إطار عمل للتعلم العميق للكشف عن أمراض الرئة متعددة الفئات باستخدام صور الأشعة السينية للصدر (CXR)
بيان المشكلة يعد الكشف المبكر والدقيق عن أمراض الرئة من صور الأشعة السينية للصدر (CXR) أمراً بالغ الأهمية لاتخاذ القرارات السريرية. ومع ذلك، تواجه أنظمة التشخيص الآلية الحالية تحديات كبيرة، بما في ذلك تداخل الأنماط البصرية بين أمراض الرئة المختلفة (على سبيل المثال، التمييز بين الالتهاب الرئوي، وكوفيد-19، وعتمة الرئة) وتحيز البيانات. فالمناذج المدربة على مجموعات بيانات متعددة المصادر غالباً ما تتعلم "ميزات مختصرة" (shortcut features) تتعلق بمصدر الحصول على البيانات (مثل نوع الماسح الضوئي أو بروتوكول المستشفى) بدلاً من الميزات المرضية الحقيقية، مما يؤدي إلى ضعف التعميم في البيئات السريرية الواقعية. بالإضافة إلى ذلك، فإن عدم توازن الفئات في مجموعات البيانات الطبية غالباً ما يؤدي إلى انحياز أداء النموذج نحو الفئات الأكثر انتشاراً، مما يقلل من حساسية النماذج تجاه الأمراض الأقل تمثيلاً.
المنهجية يقترح المؤلفون إطار عمل للتعلم العميق واعٍ بالتحيز، مصمم للتصنيف متعدد الفئات لستة حالات رئوية: كوفيد-19، والالتهاب الرئوي، والسل، وسرطان الرئة، وعتمة الرئة، والحالة الطبيعية. وتتمحور المنهجية حول ثلاث ركائز أساسية:
تنسيق ومعالجة البيانات الواعية بالتحيز:
بناء مجموعة البيانات: تم إنشاء مجموعة بيانات موحدة مكونة من 13,825 صورة أشعة سينية للصدر عبر دمج بيانات من ثلاثة مستودعات عامة (قاعدة بيانات التصوير الشعاعي لكوفيد-19، ومجموعة بيانات سرطان الرئة IQ-OTH/NCCD، ومستودعات Kaggle).
توحيد المصادر: لإزالة المؤشرات المتعلقة بالمجلدات، تم تخزين جميع الصور في هيكل واحد مع ملف بيانات وصفية موحد. تم الاحتفاظ بملصقات المصدر فقط لأغراض تحليل التحيز، وليس لتدريب تصنيف الأمراض.
فحص السلامة: تم إجراء "فحص سلامة التنبؤ بالمصدر" عن طريق تدريب نموذج ResNet-50 للتنبؤ بأصل مجموعة البيانات. حقق النموذج دقة بنسبة 51.57% فقط مع استدعاء (recall) شديد الانحراف، وفشل في التمييز بين المصادر بشكل موثوق. تدعم هذه النتيجة الادعاء بأن المعالجة المسبقة نجحت في تقليل الآثار الناتجة عن عملية الاستحواذ الخاصة بكل مجموعة بيانات، مما أجبر نماذج تصنيف الأمراض على تعلم الميزات ذات الصلة بالمرض.
توازن الفئات: لمعالجة عدم توازن الفئات (حيث تسود عتمة الرئة والالتهاب الرئوي)، استخدم الإطار تقنيات تعزيز البيانات في الوقت الفعلي (الدوران، القلب، التكبير) ودوال فقدان موزونة للفئات أثناء التدريب.
بنية النموذج والتدريب:
قيمت الدراسة ثلاث بنيات ResNet تم تكييفها لتناسب المدخلات الرمادية (grayscale): وهي ResNet-38 (نسخة مخصصة وأقل عمقاً مشتقة من ResNet-34)، و ResNet-50، و ResNet-101.
استخدمت جميع النماذج أوزانًا مدربة مسبقاً على ImageNet، مع تعديلها لتقبل مدخلات أحادية القناة (رمادية) عبر تكرار القنوات أو تعديل الطبقات.
اعتمد التدريب على التحقق المتقاطع الطبقي ذي الـ 5 طيات (5-fold stratified cross-validation)، وتنظيم الإسقاط (dropout) بنسبة (0.3)، ومحسن Adam مع دالة فقدان التقاطع الموزون (weighted cross-entropy loss).
التحقق والتفسير:
التحقق الإحصائي: تم اختبار الأداء بصرامة باستخدام فترات الثقة لأسلوب Bootstrap، واختبارات جودة المطابقة "كا مربع" (Chi-square)، واختبارات McNemar للمقارنة الزوجية بين النماذج.
القابلية للتفسير: تم استخدام تقنية Grad-CAM (رسم خرائط التنشيط المعتمدة على التدرج) لتصور انتباه النموذج. وقد قام خبير طبي مؤهل (د. جوفيند ساران) بالتحقق النوعي من أن الخرائط الحرارية تركز على المناطق الرئوية المنطقية تشريحياً بدلاً من عيوب الخلفية.
التحقق الخارجي: تم اختبار النموذج الأفضل أداءً على مجموعة بيانات مستقلة وغير مرئية من Roboflow (800 صورة) دون إعادة تدريب لتقييم القدرة على التعميم.
المساهمات الرئيسية
إطار عمل واعٍ بالتحيز: تقدم الورقة خط معالجة مسبقة محدد يتضمن "فحص سلامة التنبؤ بالمصدر" للتحقق صراحة من أن النماذج لا تتعلم الاختصارات المرتبطة بأصل مجموعة البيانات.
رؤية معمارية (العمق مقابل التعميم): خلافاً للتوجه نحو زيادة عمق الشبكة، تُظهر الدراسة أن بنية أقل عمقاً (ResNet-38) تفوقت على نظيراتها الأكثر عمقاً (ResNet-50 و ResNet-101) في هذه المجموعة المحددة من البيانات الطبية متعددة الفئات وغير المتوازنة.
تقييم شامل: توفر الدراسة مجموعة تحقق إحصائي قوية (بما في ذلك اختبارات McNemar وتحليل bootstrap) بالإضافة إلى التحقق الخارجي على مجموعة بيانات غير مرئية، متجاوزة مجرد مقاييس الدقة لتقييم الموثوقية والتعميم.
النتائج التجريبية
الأداء: حقق ResNet-38 أعلى متوسط دقة تصنيف بلغ 93.34% (±0.0073) تحت التحقق المتقاطع ذي الـ 5 طيات. وقد تفوق هذا على ResNet-50 (86.97%) و ResNet-101 (89.40%).
الاستقرار: أظهر ResNet-38 تبايناً أقل عبر الطيات ومعايرة أفضل (احتمالات خلفية أكثر اتساقاً) مقارنة بالنماذج الأعمق، والتي أظهرت علامات على الإفراط في التخصيص (overfitting) حيث اقتربت دقة التدريب من 1.0 بينما تأخرت دقة التحقق.
أداء الفئات: أظهر ResNet-38 درجات F1 واستدعاء (recall) متفوقة عبر معظم الفئات، وخاصة للفئات الأقل تمثيلاً مثل السل وكوفيد-19، ويُعزى ذلك إلى التعامل الفعال مع عدم توازن الفئات.
التحقق الخارجي: على مجموعة بيانات Roboflow غير المرئية، حقق نموذج ResNet-38 دقة بنسبة 85%، حيث حدثت الأخطاء بشكل أساسي بين الحالات المتداخلة سريرياً (مثل الالتهاب الرئوي مقابل كوفيد-19)، مما يؤكد أن النموذج يتعلم الميزات المرضية بدلاً من عيوب مجموعة البيانات.
الكفاءة: تطلب ResNet-38 معاملات أقل بكثير (21 مليون معلمة) وموارد حوسبة أقل (FLOPs ~3.1G) مقارنة بـ ResNet-101 (44.5 مليون معلمة، ~7.8G FLOPs)، مما يوفر توازناً أفضل بين الدقة والكفاءة.
الأهمية والادعاءات تدعي الورقة أنه بالنسبة لتصنيف أمراض الرئة متعددة الفئات باستخدام صور الأشعة السينية للصدر، فإن زيادة عمق الشبكة لا تؤدي بالضرورة إلى تحسين الأداء، خاصة عند التعامل مع مجموعات بيانات متوسطة الحجم وغير متوازنة الفئات. وتجادل الدراسة بأن الشبكات الأعمق (مثل ResNet-101) عرضة للإفراط في التخصيص وتعلم الضوضاء، بينما تؤدي بنية أقل عمقاً ومحسنة بعناية (مثل ResNet-38)، جنباً إلى جنب مع استراتيجيات صارمة لتخفيف التحيز، إلى تحقيق تعميم واستقرار فائقين.
يؤكد المؤلفون أن إطار عملهم يوفر "طريقة واقعية للفحص القائم على الذكاء الاصطناعي للأشعة" من خلال إعطاء الأولوية للتصميم الواعي بالتحيز والتحقق الإحصائي على مجرد التعقيد المعماري. ويذكرون صراحة أنه بينما يقلل إطار العمل من الاعتماد على أصل مجموعة البيانات، فإنه لا يقضي على جميع التحولات في النطاق (domain shifts)، وأن التحقق الخارجي الإضافي على مجموعات بيانات مستشفيات متنوعة مطلوب للنشر السريري. وتخلص الدراسة إلى أن نموذج ResNet-38 يمثل التوازن الأمثل بين الدقة، وكفاءة الحوسبة، والموثوقية لتحليل الأشعة السينية للصدر في العالم الحقيقي.