Enhancing brain tumor diagnosis: an optimized CNN hyperparameter model for improved accuracy and reliability

• Published in: PeerJ. Computer science Vol. 10; p. e1878
• Main Authors: Asiri, Abdullah A, Shaf, Ahmad, Ali, Tariq, Aamir, Muhammad, Irfan, Muhammad, Alqahtani, Saeed
• Format: Journal Article
• Language: English
• Published: United States PeerJ. Ltd 14.03.2024; PeerJ Inc
• Subjects: Brain tumor diagnosis; Brain tumors; Computer Vision; Decision-making processes; Diagnosis; Diagnostic imaging; Feature extraction; Gliomas; Hyperparameter tuning; Model complexity; Neural Networks; Spatial resolution; Brain tumor diagnosis; Optimization techniques; Feature extraction; Decision-making processes; Model complexity; Spatial resolution; Hyperparameter tuning
• ISSN: 2376-5992; 2376-5992
• DOI: 10.7717/peerj-cs.1878

Hyperparameter tuning plays a pivotal role in the accuracy and reliability of convolutional neural network (CNN) models used in brain tumor diagnosis. These hyperparameters exert control over various aspects of the neural network, encompassing feature extraction, spatial resolution, non-linear mapping, convergence speed, and model complexity. We propose a meticulously refined CNN hyperparameter model designed to optimize critical parameters, including filter number and size, stride padding, pooling techniques, activation functions, learning rate, batch size, and the number of layers. Our approach leverages two publicly available brain tumor MRI datasets for research purposes. The first dataset comprises a total of 7,023 human brain images, categorized into four classes: glioma, meningioma, no tumor, and pituitary. The second dataset contains 253 images classified as "yes" and "no." Our approach delivers exceptional results, demonstrating an average 94.25% precision, recall, and F1-score with 96% accuracy for dataset 1, while an average 87.5% precision, recall, and F1-score, with accuracy of 88% for dataset 2. To affirm the robustness of our findings, we perform a comprehensive comparison with existing techniques, revealing that our method consistently outperforms these approaches. By systematically fine-tuning these critical hyperparameters, our model not only enhances its performance but also bolsters its generalization capabilities. This optimized CNN model provides medical experts with a more precise and efficient tool for supporting their decision-making processes in brain tumor diagnosis.