فرق عميقة غير متجانسة مكدسة باستخدام CatBoost لتصنيف أورام الدماغ متعددة الفئات القابلة للتفسير بواسطة التصوير بالرنين المغناطيسي CatBoost-Stacked Heterogeneous Deep Ensembles for Explainable Multi-Class Brain Tumor MRI Classification

شارك:
المجلة: Journal of Medical and Health Studies، المجلد: 7، العدد: 5
DOI: https://doi.org/10.32996/jmhs.2026.7.5.4
تاريخ النشر: 2026-03-08
المؤلف: Partha Chakraborty
الموضوع الرئيسي: كشف وتصنيف أورام الدماغ

نظرة عامة

تقدم البحث نهجًا جديدًا لفحص أورام الدماغ بدقة من صور الرنين المغناطيسي، مستهدفًا بشكل خاص أربع فئات: الورم الدبقي، ورم السحايا، أورام الغدة النخامية، وعدم وجود ورم. يقدم المؤلفون نموذج تجميع متكامل يدمج بين مستخرجات الميزات المختلفة، بما في ذلك EfficientNetB0، MobileNetV2، GoogLeNet، شبكة الكبسولة متعددة المستويات، وشبكة CNN، مع متعلم CatBoost. تم تصميم هذا النموذج لمعالجة التحديات مثل التباين من مصادر متعددة، عدم التوازن بين الفئات، والحاجة إلى نتائج قابلة للتفسير سريريًا. تم التحقق من صحة المنهجية على مجموعة بيانات M2، التي تتكون من 24,618 صورة رنين مغناطيسي، باستخدام تقسيم 80/5/15 والتحقق المتقاطع المنهجي 10 مرات، مع تنفيذ تقنيات التخفيف من عدم التوازن حصريًا أثناء التدريب.

حقق النموذج المقترح أداءً مثيرًا للإعجاب، بما في ذلك دقة تبلغ 98.99±0.41%، ودرجة F1 الدقيقة 98.32±0.49%، وPR-AUC 99.33±0.27%، وMCC 94.14±0.13%. تتجاوز هذه النتائج تلك الخاصة بالنماذج الفردية، حيث تكشف تدقيقات Grad-CAM أن النموذج يركز بشكل فعال على المناطق ذات الصلة بالورم في معظم الحالات. ومع ذلك، يعترف المؤلفون بأن زيادة التكلفة الحاسوبية لاستدلال التجميع واستمرار التحيزات بسبب تباين الماسحات والبروتوكولات لا تزال تمثل تحديات. تشمل اتجاهات البحث المستقبلية تعزيز دعم القرار الواعي بالشك، والتحقق من صحة النموذج على مجموعات سريرية خارجية، وتحسين المعايرة لتقدير موثوق للثقة في التطبيقات العملية.

مقدمة

تسلط مقدمة هذه الورقة البحثية الضوء على الدور الحاسم للتشخيص الدقيق وفي الوقت المناسب لأورام الدماغ، والذي يؤثر بشكل كبير على استراتيجيات العلاج ونتائج المرضى. يتم التأكيد على التصوير بالرنين المغناطيسي (MRI) كأداة حيوية في علم الأورام العصبية، ومع ذلك، غالبًا ما تعيق فعاليته الاعتماد على تفسير الخبراء، مما يؤدي إلى تباين وعدم كفاءة، خاصة في البيئات ذات الموارد المحدودة. يقترح المؤلفون نهج تصنيف تلقائي متعدد الفئات لأورام الدماغ باستخدام الرنين المغناطيسي، مصنفين إياها إلى ورم دبقي، ورم سحائي، ورم غدة نخامية، وعدم وجود ورم، مما يعكس كل من الأهمية السريرية والحاجة إلى تمييز موثوق بين أنواع الأورام.

لمعالجة التحديات الحالية في تصنيف أورام الدماغ، مثل الإفراط في التكيف مع مجموعات بيانات معينة، وعدم التوازن بين الفئات، والقيود في القابلية للتفسير، يقدم المؤلفون طريقة تجميع باستخدام CatBoost كمتعلم رئيسي. يجمع هذا النهج بين التنبؤات من نماذج أساسية متنوعة لتعزيز أداء التصنيف والقدرة على التحمل ضد تباين المجال. تشير الورقة إلى تحسينات كبيرة في مقاييس التصنيف، حيث حققت دقة تبلغ 98.99% وأظهرت فعالية المنهجية المقترحة من خلال التحقق الواسع على مجموعة بيانات متعددة المصادر. كما يؤكد المؤلفون على أهمية القابلية للتفسير في التشخيص المدفوع بالذكاء الاصطناعي، مقدّمين تحليلًا منهجيًا للأخطاء لتعزيز الثقة السريرية وموثوقية النموذج.

طرق

تستند المنهجية لإطار تصنيف الرنين المغناطيسي للدماغ CNN على ثلاثة مبادئ رئيسية: التعلم بالنقل، استراتيجيات التجميع، ووحدات القابلية للتفسير. يستفيد التعلم بالنقل من الشبكات العميقة المدربة مسبقًا لإنشاء تمثيلات مستقرة، بينما تعزز طرق التجميع القدرة على التحمل في مهام تصنيف الأمراض الصعبة. بالإضافة إلى ذلك، يسمح دمج مستخرجات الميزات المعززة بالتحويل بالتقاط كل من القوام المحلي والمعلومات السياقية الأوسع. يدعم هذا النهج مجموعة من الدراسات عبر التصوير الطبي وتطبيقات الذكاء الاصطناعي، مما يبرز الحاجة إلى أنظمة قابلة للتوسع، فعالة، وقابلة للتكيف.

يبدأ الإطار المقترح بمعالجة خفيفة لشرائح الرنين المغناطيسي لتقليل الضوضاء الخلفية وتباين التباين. تولد مجموعة متنوعة من المتعلمين الأساسيين متجهات احتمالية الفئة لأربع فئات مستهدفة، والتي يتم دمجها بعد ذلك في تمثيل ميزات على مستوى ميتا مضغوط. تتم معالجة هذا التمثيل بواسطة CatBoost، الذي يطبق قاعدة تجميع غير خطية لإنتاج التنبؤ النهائي للتصنيف. لضمان الشفافية والقابلية للتفسير، يتم استخدام Grad-CAM لتصور المناطق في صور الرنين المغناطيسي التي تساهم في قرارات النموذج، مما يسمح بتحديد المناطق ذات الصلة بالورم وأنماط الفشل النموذجية.

نتائج

تقدم قسم النتائج نتائج نوعية من تصور متعدد الألواح (الشكل 2) الذي يوضح أداء نموذج في تصنيف صور الرنين المغناطيسي إلى أربع فئات: ورم دبقي، ورم سحائي، ورم غدة نخامية، وعدم وجود ورم، إلى جانب مثال على سوء التصنيف. تعرض كل لوحة شريحة الرنين المغناطيسي الأصلية جنبًا إلى جنب مع خريطة حرارة Grad-CAM المقابلة، التي تبرز المناطق في الصورة التي يركز عليها النموذج أثناء التصنيف. يتم توحيد خرائط الحرارة في القياس وتطبيع خريطة الألوان، مما يسمح بإجراء مقارنات مباشرة عبر فئات الأورام المختلفة.

يكشف التحليل أن التنبؤات الصحيحة لفئات الأورام تتميز بتنشيط محلي في المناطق التي تتوافق مع الآفات، بينما تظهر الحالات الطبيعية استجابات منخفضة الكثافة تكون منتشرة تشريحيًا، تفتقر إلى نقاط ساخنة بؤرية مميزة. على العكس، يوضح مثال الفشل سيناريو شائعًا لسوء التصنيف حيث يكون انتباه Grad-CAM إما غير متماشي مع حدود الآفة أو مفرط التعميم عبر الهياكل التشريحية ذات الصلة، مما يعكس الارتباك بين الأنواع الفرعية للأورام المتشابهة شكليًا. تؤكد هذه التقييمات النوعية على نقاط القوة والضعف في تصنيف الأورام بناءً على بيانات الرنين المغناطيسي.

مناقشة

تسلط المناقشة الضوء على التقدم في تصنيف أورام الدماغ المعتمد على الرنين المغناطيسي، مع التأكيد على هيمنة الشبكات العصبية التلافيفية العميقة (CNNs) وإمكانات شبكات الكبسولة (CapsNets) بسبب قدرتها على الحفاظ على علاقات الجزء-الكل. بينما أظهرت CNNs دقة عالية في مجموعات البيانات المسيطر عليها، قد لا تترجم أداؤها بشكل موثوق إلى السيناريوهات الواقعية، خاصة عند تقييمها على مصادر بيانات متجانسة. تدعو الورقة إلى استخدام نهج هجينة وتجميعية، مثل التجميع، التي يمكن أن تعزز استقرار التنبؤ من خلال الاستفادة من المتعلمين المتنوعين وتصحيح الأخطاء النظامية من خلال تركيبات غير خطية للمخرجات. يُلاحظ استخدام CatBoost للتعلم الميتا لفعاليته في التعامل مع التفاعلات المعقدة وتحسين نتائج التصنيف عبر مهام التصوير الطبي المختلفة.

يستخدم النموذج المقترح تجميعًا غير متجانس من النماذج الأساسية، بما في ذلك EfficientNetB0، MobileNetV2، GoogLeNet، وشبكة الكبسولة متعددة المستويات، لالتقاط أنماط الخطأ التكميلية. يتضمن بروتوكول التدريب تقنيات مثل Borderline-SMOTE لمعالجة عدم التوازن بين الفئات وزيادة تعزيز الحفاظ على التسمية لتعزيز قوة النموذج. تركز مقاييس التقييم على الدقة والتدابير المقاومة لعدم التوازن، مع نتائج تشير إلى أن النموذج المكدس باستخدام CatBoost حقق أداءً مثيرًا للإعجاب على مجموعة بيانات M2، متفوقًا على الأسس الفردية. تكشف تدقيقات Grad-CAM أن التنبؤات الناجحة تتماشى مع المناطق ذات الصلة سريريًا، بينما غالبًا ما تنشأ سوء التصنيفات من غموض شكل الورم. تختتم الورقة بالاعتراف بالتكاليف الحاسوبية المرتبطة بأساليب التجميع والحاجة إلى عمل مستقبلي لمعالجة عدم اليقين في اتخاذ القرار والتحقق من صحة النموذج على مجموعات سريرية متنوعة.

Journal: Journal of Medical and Health Studies, Volume: 7, Issue: 5
DOI: https://doi.org/10.32996/jmhs.2026.7.5.4
Publication Date: 2026-03-08
Author(s): Partha Chakraborty
Primary Topic: Brain Tumor Detection and Classification

Overview

The research presents a novel approach for accurate brain tumor screening from MRI scans, specifically targeting four classes: glioma, meningioma, pituitary tumors, and no tumor. The authors introduce a stacking ensemble model that integrates various feature extractors, including EfficientNetB0, MobileNetV2, GoogLeNet, a multi-level CapsuleNet, and a CNN, with a CatBoost meta-learner. This model is designed to address challenges such as multi-source variability, class imbalance, and the need for clinically interpretable outcomes. The methodology was validated on the M2 dataset, comprising 24,618 MR images, employing an 80/5/15 split and stratified 10-fold cross-validation, while implementing imbalance mitigation techniques exclusively during training.

The proposed ensemble model achieved impressive performance metrics, including an accuracy of 98.99±0.41%, a micro-F1 score of 98.32±0.49%, a PR-AUC of 99.33±0.27%, and an MCC of 94.14±0.13%. These results surpass those of individual backbone models, with Grad-CAM audits revealing that the model effectively focuses on tumor-relevant regions in most cases. However, the authors acknowledge that the increased computational cost of ensemble inference and the persistence of biases due to scanner and protocol variability remain challenges. Future research directions include enhancing uncertainty-aware decision support, validating the model on external clinical cohorts, and improving calibration for reliable confidence estimation in practical applications.

Introduction

The introduction of this research paper highlights the critical role of accurate and timely diagnosis of brain tumors, which significantly impacts treatment strategies and patient outcomes. Magnetic resonance imaging (MRI) is emphasized as a vital tool in neurooncology, yet its effectiveness is often hampered by reliance on expert interpretation, leading to variability and inefficiencies, particularly in resource-limited settings. The authors propose an automated multi-class classification approach for brain tumors using MRI, categorizing them into glioma, meningioma, pituitary tumor, and no tumor, which reflects both clinical relevance and the necessity for reliable differentiation among tumor types.

To address existing challenges in brain tumor classification, such as overfitting to specific datasets, class imbalance, and limitations in explainability, the authors introduce a stacking ensemble method utilizing CatBoost as a meta-learner. This approach combines predictions from diverse base models to enhance classification performance and robustness against domain variability. The paper reports significant improvements in classification metrics, achieving an accuracy of 98.99% and demonstrating the effectiveness of the proposed methodology through extensive validation on a multi-source dataset. The authors also emphasize the importance of explainability in AI-driven diagnostics, providing systematic error analysis to foster clinical trust and model reliability.

Methods

The methodology for the Brain MRI CNN classification framework is founded on three key principles: transfer learning, ensemble strategies, and explainability modules. Transfer learning leverages pretrained deep networks to establish stable representations, while ensemble methods enhance robustness in challenging disease classification tasks. Additionally, the integration of transformer-enhanced feature extractors allows for the capture of both local textures and broader contextual information. This approach is supported by a range of studies across medical imaging and AI applications, emphasizing the need for scalable, efficient, and adaptable systems.

The proposed framework begins with lightweight preprocessing of MRI slices to minimize background noise and contrast variability. A diverse set of base learners generates class-probability vectors for four target categories, which are then combined into a compact meta-level feature representation. This representation is processed by CatBoost, which applies a non-linear stacking rule to produce the final classification prediction. To ensure transparency and interpretability, Grad-CAM is utilized to visualize the areas of the MRI scans that contribute to the model’s decisions, allowing for the identification of tumor-relevant regions and typical failure patterns.

Results

The results section presents qualitative findings from a multi-panel visualization (Fig. 2) that illustrates the performance of a model in classifying MRI scans into four categories: glioma, meningioma, pituitary tumor, and no tumor, along with an example of misclassification. Each panel displays the original MRI slice alongside its corresponding Grad-CAM heatmap, which highlights areas of the image that the model focuses on during classification. The heatmaps are standardized in scaling and colormap normalization, allowing for direct comparisons across different tumor classes.

The analysis reveals that correct predictions for tumor classes are characterized by localized activation in regions corresponding to the lesions, while normal cases show low-intensity responses that are anatomically diffuse, lacking distinct focal hotspots. Conversely, the failure example illustrates a common misclassification scenario where the Grad-CAM attention is either misaligned with the lesion boundary or overly generalized across relevant anatomical structures, reflecting confusion among morphologically similar tumor subtypes. This qualitative assessment underscores the model’s strengths and weaknesses in tumor classification based on MRI data.

Discussion

The discussion highlights the advancements in MRI-based brain tumor classification, emphasizing the dominance of deep convolutional neural networks (CNNs) and the potential of capsule networks (CapsNets) due to their ability to maintain part-whole relationships. While CNNs have demonstrated high accuracy in controlled datasets, their performance may not translate reliably to real-world scenarios, particularly when evaluated on homogeneous data sources. The paper advocates for hybrid and ensemble approaches, such as stacking, which can enhance predictive stability by leveraging diverse learners and correcting systematic errors through non-linear combinations of outputs. The use of CatBoost for meta-learning is noted for its effectiveness in handling complex interactions and improving classification outcomes across various medical imaging tasks.

The proposed model employs a heterogeneous ensemble of backbones, including EfficientNetB0, MobileNetV2, GoogLeNet, and a multilevel CapsuleNet, to capture complementary error patterns. The training protocol incorporates techniques like Borderline-SMOTE for class imbalance and label-preserving augmentation to enhance model robustness. Evaluation metrics focus on accuracy and imbalance-robust measures, with results indicating that the CatBoost-stacked model achieved impressive performance on the M2 dataset, outperforming single-backbone baselines. Grad-CAM audits reveal that successful predictions align with clinically relevant areas, while misclassifications often stem from ambiguous tumor morphology. The paper concludes by acknowledging the computational costs associated with ensemble methods and the need for future work to address uncertainty in decision-making and validate the model on diverse clinical cohorts.