TRANSFER LEARNING MODEL FOR BRAIN TUMOR DETECTION WITH SNAKE SEGMENTATION

ICTACT Journal on Data Science and Machine Learning ( Volume: 5 , Issue: 4 )

Abstract

As medical standards advance, the role of medical imaging in clinical diagnosis and monitoring becomes increasingly vital. This paper presents a novel approach for the detection and classification of brain tumors using MRI imaging, leveraging digital image processing and artificial intelligence with a focus on Transfer Learning. The methodology begins with pre-processing MRI scans using a Gaussian filter, followed by snake segmentation for image partitioning. The images are then de-noised using a Parallel Non-Local Mean filter. For classification, a hybrid model combining a pre-trained VGG16 network with a Convolutional Neural Network is proposed, capitalizing on the advantages of transfer learning. This model’s efficacy is benchmarked against conventional machine learning algorithms—SVM, KNN, RF—and other deep learning architectures, including CNN variations with AlexNet, self-attention, and additive attention. Implemented in Python on Google Colab, the model was trained and tested on a Kaggle dataset comprising 5,712 images. Metrics employed for evaluation included accuracy, precision, and recall, alongside execution time. The findings demonstrate that our proposed transfer learning-enhanced model achieved superior performance with an accuracy of 96.7%, precision of 96%, and recall of 95%, thus outstripping the other examined models in brain tumor detection efficiency.

Authors

Manisha Tyagi, Priyanka
Deenbandhu Chhotu Ram University of Science and Technology, India

Keywords

Brain Tumor, Snake Segmentation, PNLM Filter, VGG16, CNN

Published By
ICTACT
Published In
ICTACT Journal on Data Science and Machine Learning
( Volume: 5 , Issue: 4 )
Date of Publication
September 2024
Pages
680 - 689

ICT Academy is an initiative of the Government of India in collaboration with the state Governments and Industries. ICT Academy is a not-for-profit society, the first of its kind pioneer venture under the Public-Private-Partnership (PPP) model

Contact Us

ICT Academy
Module No E6 -03, 6th floor Block - E
IIT Madras Research Park
Kanagam Road, Taramani,
Chennai 600 113,
Tamil Nadu, India

For Journal Subscription: journalsales@ictacademy.in

For further Queries and Assistance, write to us at: ictacademy.journal@ictacademy.in