Home > Published Issues > 2024 > Volume 15, No. 9, 2024 >
JAIT 2024 Vol.15(9): 1035-1046
doi: 10.12720/jait.15.9.1035-1046

A Dual-Branch Lightweight Model for Extracting Characteristics to Classify Brain Tumors

Sangeetha G. 1,*, Vadivu G. 1, and Sundara Raja Perumal R. 2
1. Department of Data Science and Business Systems, School of Computing, SRM Institute of Science and Technology, Kattankulathur, Tamilnadu, India
2. Department of Radiology, SRM Medical College Hospital and Research Centre, Kattankulathur, Tamilnadu, India
Email: sg8517@srmist.edu.in (S.G.); vadivug@srmist.edu.in (V.G.); majsundp@srmist.edu.in (S.R.)
*Corresponding author

Manuscript received December 16, 2023; revised February 21, 2024; accepted April 15, 2024; published September 13, 2024.

Abstract—Brain tumors present a significant challenge in healthcare, necessitating prompt and accurate detection for effective treatment. Pre-trained models were utilized to classify brain tumors without segmentation. Traditional pre-trained architectures like VGG16, VGG19, and ResNet, despite their accuracy, suffer from slow processing speed which leads to use them impractical for rapid diagnosis. Some of the other pre-trained models like Mobile Net and Efficient Net offer fast processing but overfitting problems occur in the small image dataset. To overcome these challenges a two-branch neural network model has been proposed which is lightweight feature extraction and multi-class classification of brain tumors. The proposed two-branch architecture begins with refining the size of the input images, then extraction of robust features, and concludes with a neural network classifier. The proposed model is also evaluated in the presence of image distortions including Gaussian, Poisson, and Speckle noise to ensure the robustness of the proposed solution. Experimental results state that the proposed model is capable of maintaining high accuracy in tumor classification when compared to the other pre-trained models with limited datasets and noise interferences.
 
Keywords—two-branch feature extraction, Magnetic resonance imaging (MRI), brain tumor, Convolutional Neural Network (CNN), VGG16, efficient net, mobile net, DenseNet21

Cite: Sangeetha G., Vadivu G., and Sundara Raja Perumal R., "A Dual-Branch Lightweight Model for Extracting Characteristics to Classify Brain Tumors," Journal of Advances in Information Technology, Vol. 15, No. 9, pp. 1035-1046, 2024.

Copyright © 2024 by the authors. This is an open access article distributed under the Creative Commons Attribution License (CC BY-NC-ND 4.0), which permits use, distribution and reproduction in any medium, provided that the article is properly cited, the use is non-commercial and no modifications or adaptations are made.