Design of an Optimal Convolutional Neural Network Architecture for MRI Brain Tumor Classification by Exploiting Particle Swarm Optimization
Open Access
- 24 January 2025
- journal article
- research article
- Published by MDPI AG in Journal of Imaging
- Vol. 11 (2) , 31
Abstract
The classification of brain tumors using MRI scans is critical for accurate diagnosis and effective treatment planning, though it poses significant challenges due to the complex and varied characteristics of tumors, including irregular shapes, diverse sizes, and subtle textural differences. Traditional convolutional neural network (CNN) models, whether handcrafted or pretrained, frequently fall short in capturing these intricate details comprehensively. To address this complexity, an automated approach employing Particle Swarm Optimization (PSO) has been applied to create a CNN architecture specifically adapted for MRI-based brain tumor classification. PSO systematically searches for an optimal configuration of architectural parameters—such as the types and numbers of layers, filter quantities and sizes, and neuron numbers in fully connected layers—with the objective of enhancing classification accuracy. This performance-driven method avoids the inefficiencies of manual design and iterative trial and error. Experimental results indicate that the PSO-optimized CNN achieves a classification accuracy of 99.19%, demonstrating significant potential for improving diagnostic precision in complex medical imaging applications and underscoring the value of automated architecture search in advancing critical healthcare technology.Keywords
All Related Versions
This publication has 34 references indexed in Scilit:
- Automatic Brain Tumor Segmentation Using Convolutional Neural Networks: U-Net Framework with PSO-Tuned Hyperparameters2024
- Advanced CNN based on genetic algorithm to automated femoral neck fracture classificationSignal, Image and Video Processing, 2024
- Development of hybrid models based on deep learning and optimized machine learning algorithms for brain tumor Multi-ClassificationExpert Systems with Applications, 2023
- Automation of surveillance systems using deep learning and facial recognitionInternational Journal of System Assurance Engineering and Management, 2023
- A deep autoencoder approach for detection of brain tumor imagesComputers and Electrical Engineering, 2022
- IntelliSwAS: Optimizing deep neural network architectures using a particle swarm-based approachExpert Systems with Applications, 2022
- A Comprehensive Survey of Loss Functions in Machine LearningAnnals of Data Science, 2020
- A survey of regularization strategies for deep modelsArtificial Intelligence Review, 2019
- Magnetic resonance imaging-based brain tumor grades classification and grading via convolutional neural networks and genetic algorithmsBiocybernetics and Biomedical Engineering, 2019
- Particle swarm optimization algorithm: an overviewSoft Computing, 2017