Brain tumor classification in MRI image using convolutional neural network.

Hassan Ali Khan, Wu Jue, Muhammad Mushtaq, Muhammad Umer Mushtaq
Author Information
  1. Hassan Ali Khan: School of Computer Science and Technology, Southwest Unversity of Science and Techonlogy, Mianyang 621010, China.
  2. Wu Jue: School of Computer Science and Technology, Southwest Unversity of Science and Techonlogy, Mianyang 621010, China.
  3. Muhammad Mushtaq: Insitute for Neuro and Bioinformatics, University of Lübeck, Germany.
  4. Muhammad Umer Mushtaq: School of Information Engineering, Southwest Unversity of Science and Techonlogy, Mianyang 621010, China.

Abstract

Brain tumor is a severe cancer disease caused by uncontrollable and abnormal partitioning of cells. Recent progress in the field of deep learning has helped the health industry in Medical Imaging for Medical Diagnostic of many diseases. For Visual learning and Image Recognition, task CNN is the most prevalent and commonly used machine learning algorithm. Similarly, in our paper, we introduce the convolutional neural network (CNN) approach along with Data Augmentation and Image Processing to categorize brain MRI scan images into cancerous and non-cancerous. Using the transfer learning approach we compared the performance of our scratched CNN model with pre-trained VGG-16, ResNet-50, and Inception-v3 models. As the experiment is tested on a very small dataset but the experimental result shows that our model accuracy result is very effective and have very low complexity rate by achieving 100% accuracy, while VGG-16 achieved 96%, ResNet-50 achieved 89% and Inception-V3 achieved 75% accuracy. Our model requires very less computational power and has much better accuracy results as compared to other pre-trained models.

Keywords

MeSH Term

Brain
Brain Neoplasms
Deep Learning
Humans
Image Processing, Computer-Assisted
Magnetic Resonance Imaging
Neural Networks, Computer

Word Cloud

Created with Highcharts 10.0.0learningCNNaccuracytumorMRImodelachievedBraindeepMedicalImageconvolutionalneuralnetworkapproachbraintransfercomparedpre-trainedVGG-16ResNet-50modelsresultseverecancerdiseasecauseduncontrollableabnormalpartitioningcellsRecentprogressfieldhelpedhealthindustryImagingDiagnosticmanydiseasesVisualRecognitiontaskprevalentcommonlyusedmachinealgorithmSimilarlypaperintroducealongDataAugmentationProcessingcategorizescanimagescancerousnon-cancerousUsingperformancescratchedInception-v3experimenttestedsmalldatasetexperimentalshowseffectivelowcomplexityrateachieving100%96%89%Inception-V375%requireslesscomputationalpowermuchbetterresultsclassificationimageusingVGGinceptionresnet

Similar Articles

Cited By