Fusion-Based Brain Tumor Classification Using Deep Learning and Explainable AI, and Rule-Based Reasoning

2508.06891v1 eess.IV, cs.CV 2025-08-14
Авторы:

Melika Filvantorkaman, Mohsen Piri, Maral Filvan Torkaman, Ashkan Zabihi, Hamidreza Moradi

Резюме на русском

## Контекст Определение и классификация типов головного мозга важны для точного диагностического процесса и разработки эффективной терапии. Несмотря на развитие медицинской интеллектуальной системы, классификация головного мозга остается сложной, требующей высокой точности и удобоваримой транспарентности. Существуют технические вызовы, связанные с возможностью ошибки классификации и необходимостью объяснительных моделей для укрепления доверия клинических специалистов. Таким образом, целью этого исследования является разработка интерпретируемой системы классификации головного мозга, которая объединяет мощность глубокого обучения и интерпретируемость. ## Метод Предложена структура с использованием нескольких сверточных нейронных сетей (CNN), использующая MobileNetV2 и DenseNet121 для классификации трех типов головного мозга: glioma, meningioma и pituitary adenoma. Тренировка выполнялась на датасете Figshare с использованием кросс-валидации 5-классов. Для улучшения транспарентности и надежности классификации интегрирована Explainable AI (XAI) с Grad-CAM++ для визуализации классов, которая отображает важные области в изображениях MRI. Добавлена символическая модель Clinical Decision Rule Overlay (CDRO), чтобы связать предсказания модели с опытным познанием радиологов. ## Результаты Общая точность фреймворка составила 91.7%, причем точность, полнота и F1-меры равны 91.9%, 91.7% и 91.6% соответственно. Grad-CAM++ показали сильную схожесть с анатомическими регионами, с Dice-коэффициентом 0.88 и IoU 0.78. Обзоры клинических правил существенно подтвердили эти результаты, обобщая модельные выводы. Оценка интерпретируемости с помощью экспертов показала высокую оценку удобства и соответствия визуализаций к реальным радиологическим понятиям. ## Значимость Предложенная модель может быть применена в клинической практике для классификации головного мозга, повышая точность и надежность диагностики. Интеграция XAI и CDRO улучшает доверие к системе, а символическая модель помогает радиологам понять модельные выводы. Этот подход может быть применён в других областях диагностики нейрологических заболеваний. ## Выводы Разработанная система предлагает эффективное, интерпретируемое решение для классификации головного мозга. Дальнейшие исследования будут сфокусированы на повышении точности, улучшении обработки нестандартных изображений и расширении модели на другие типы головного мозга. Это будет оптимизировать диагностический процесс и улучшить выбор терапий.

Abstract

Accurate and interpretable classification of brain tumors from magnetic resonance imaging (MRI) is critical for effective diagnosis and treatment planning. This study presents an ensemble-based deep learning framework that combines MobileNetV2 and DenseNet121 convolutional neural networks (CNNs) using a soft voting strategy to classify three common brain tumor types: glioma, meningioma, and pituitary adenoma. The models were trained and evaluated on the Figshare dataset using a stratified 5-fold cross-validation protocol. To enhance transparency and clinical trust, the framework integrates an Explainable AI (XAI) module employing Grad-CAM++ for class-specific saliency visualization, alongside a symbolic Clinical Decision Rule Overlay (CDRO) that maps predictions to established radiological heuristics. The ensemble classifier achieved superior performance compared to individual CNNs, with an accuracy of 91.7%, precision of 91.9%, recall of 91.7%, and F1-score of 91.6%. Grad-CAM++ visualizations revealed strong spatial alignment between model attention and expert-annotated tumor regions, supported by Dice coefficients up to 0.88 and IoU scores up to 0.78. Clinical rule activation further validated model predictions in cases with distinct morphological features. A human-centered interpretability assessment involving five board-certified radiologists yielded high Likert-scale scores for both explanation usefulness (mean = 4.4) and heatmap-region correspondence (mean = 4.0), reinforcing the framework's clinical relevance. Overall, the proposed approach offers a robust, interpretable, and generalizable solution for automated brain tumor classification, advancing the integration of deep learning into clinical neurodiagnostics.

Ссылки и действия