Detection of brain tumors using a transfer learning-based optimized ResNet152 model in MR images.
Journal:
Computers in biology and medicine
Published Date:
Feb 13, 2025
Abstract
Brain tumors are incredibly harmful and can drastically reduce life expectancy. Most researchers use magnetic resonance (MR) scans to detect tumors because they can provide detailed images of the affected area. Recently, AI-based deep learning methods have emerged to enhance diagnostic accuracy through efficient data processing. This study investigates the effectiveness of deep transfer learning techniques for accurate brain tumor diagnosis. A preprocessing pipeline is used to enhance the image quality. This pipeline includes morphological operations such as erosion and dilation for shape refinement, Gaussian blurring for noise reduction, and thresholding for image cropping. Principal Component Analysis (PCA) is applied for dimensionality reduction, and data augmentation enriches the dataset. The dataset is partitioned into training (80 %) and testing (20 %). Pretrained ResNet152 and GoogleNet extract meaningful features from the images. These extracted features are then classified using conventional machine learning classifiers: Support Vector Machine (SVM), K-Nearest Neighbors (KNN), Classification and Regression Trees (CART), and Gaussian Naive Bayes (GNB). This study compares the performance of two pre-trained models for medical image analysis. Performance metrics such as accuracy, sensitivity, recall, and F1-Score evaluate the final classification results. ResNet152 outperforms GoogleNet, achieving a 98.53 % accuracy, an F1 score of 97.4 %, and a sensitivity of 96.52 %. This study highlights integrating deep learning and traditional machine-learning techniques in medical image analysis for effective brain tumor detection.