Abstract:Brain tumors are a diverse group of neoplasms that are challenging to detect and classify due to their varying characteristics. Deep learning techniques have proven to be effective in tumor classification. However, there is a lack of studies that compare these techniques using a common methodology. This work aims to analyze the performance of convolutional neural networks in the classification of brain tumors. We propose a network consisting of a few convolutional layers, batch normalization, and max-pooling. Then, we explore recent deep architectures, such as VGG, ResNet, EfficientNet, or ConvNeXt. The study relies on two magnetic resonance imaging datasets with over 3000 images of three types of tumors -gliomas, meningiomas, and pituitary tumors-, as well as images without tumors. We determine the optimal hyperparameters of the networks using the training and validation sets. The training and test sets are used to assess the performance of the models from different perspectives, including training from scratch, data augmentation, transfer learning, and fine-tuning. The experiments are performed using the TensorFlow and Keras libraries in Python. We compare the accuracy of the models and analyze their complexity based on the capacity of the networks, their training times, and image throughput. Several networks achieve high accuracy rates on both datasets, with the best model achieving 98.7% accuracy, which is on par with state-of-the-art methods. The average precision for each type of tumor is 94.3% for gliomas, 93.8% for meningiomas, 97.9% for pituitary tumors, and 95.3% for images without tumors. VGG is the largest model with over 171 million parameters, whereas MobileNet and EfficientNetB0 are the smallest ones with 3.2 and 5.9 million parameters, respectively. These two neural networks are also the fastest to train with 23.7 and 25.4 seconds per epoch, respectively. On the other hand, ConvNext is the slowest model with 58.2 seconds per epoch. Our custom model obtained the highest image throughput with 234.37 images per second, followed by MobileNet with 226 images per second. ConvNext obtained the smallest throughput with 97.35 images per second. ResNet, MobileNet, and EfficientNet are the most accurate networks, with MobileNet and EfficientNet demonstrating superior performance in terms of complexity. Most models achieve the best accuracy using transfer learning followed by a fine-tuning step. However, data augmentation does not contribute to increasing the accuracy of the models in general.

Implementation of Convolutional Neural Network with EfficientNet-B0 Architecture for Brain Tumor Classification

Classification of Brain Tumors on MRI Images Using Convolutional Neural Network Model EfficientNet

Classification of Brain Image Tumor using EfficientNet B1-B2 Deep Learning

Brain Tumor Classification Using Dense Efficient-Net

Detection and classification of brain tumor using hybrid deep learning models

Detection and Classification of Brain tumors Using Deep Convolutional Neural Networks

Brain Tumor Classification Using Convolutional Neural Network and Feature Extraction

Efficient Feature Extraction and Classification Architecture for MRI-Based Brain Tumor Detection

Comparative Analysis of Different Deep Convolutional Neural Network Architectures for Classification of Brain Tumor on Magnetic Resonance Images

Performance of convolutional neural networks for the classification of brain tumors using magnetic resonance imaging

Computational Intelligence Approach to Improve the Classification Accuracy of Brain Neoplasm in MRI Data

An efficient brain tumor detection and classification using pre-trained convolutional neural network models

Convolutional Block Attention Module-based Deep Learning Model for MRI Brain Tumor Identification (ResNet-CBAM)

Brain Tumor Detection and Classification Using an Optimized Convolutional Neural Network

Enhanced brain tumor classification using graph convolutional neural network architecture

Brain Tumor Diagnosis and Classification via Pre-Trained Convolutional Neural Networks

Efficient Brain Tumor Detection Based on Deep Learning Models

Brain Tumor Classification from MRI Using Image Enhancement and Convolutional Neural Network Techniques

Enhancing brain tumor classification in MRI scans with a multi-layer customized convolutional neural network approach

An Effective and Novel Approach for Brain Tumor Classification Using AlexNet CNN Feature Extractor and Multiple Eminent Machine Learning Classifiers in MRIs

Brain tumor MRI classification and identification using an image classification model via Convolutional Neural Networks