Lightweight CNN for Accurate Brain Tumor Detection from MRI with Limited Training Data

Abstract

Aim: This study aims to develop a robust and lightweight deep learning model for early brain tumor detection using magnetic resonance imaging (MRI), particularly under constraints of limited data availability. Objective: To design a CNN-based diagnostic model that accurately classifies MRI brain scans into tumor-positive and tumor-negative categories with high clinical relevance, despite a small dataset. Methods: A five-layer CNN architecture—comprising three convolutional layers, two pooling layers, and a fully connected dense layer—was implemented using TensorFlow and TFlearn. A dataset of 189 grayscale brain MRI images was used, with balanced classes. The model was trained over 10 epochs and 202 iterations using the Adam optimizer. Evaluation metrics included accuracy, precision, recall, F1 Score, and ROC AUC.Results: The proposed model achieved 99% accuracy in both training and validation. Key performance metrics, including precision (98.75%), recall (99.20%), F1-score (98.87%), and ROC-AUC (0.99), affirmed the model’s reliability. The loss decreased from 0.412 to near zero. A comparative analysis with a baseline TensorFlow model trained on 1,800 images showed the superior performance of the proposed model.Conclusion: The results demonstrate that accurate brain tumor detection can be achieved with limited data using a carefully optimized CNN. Future work will expand datasets and integrate explainable AI for enhanced clinical integration.