Brain Tumor Detection using Artificial Neural Network

Download Full-Text PDF Cite this Publication

Text Only Version

Brain Tumor Detection using Artificial Neural Network

Soham Mehta Department of Computer Science Vellore Institute of Technology

Vellore, India

AbstractBrain Tumor is to be considered one of the deadliest diseases which occur in the human brain and can destroy someones ability to perform tasks well, read, write communicate , think and much more as brain is the main part of a human body which helps in functioning and giving orders to the other parts and organs of the body.The detection and extraction if the tumor region from the MR is the main concern and it is a very lengthy task depending on the radiologist. Hence the role of computers and other electronic devices readily available or invented out there comes into play to improve the performance and reduce the complexity in the segmentation process. As the role of machines are there, following it the prediction and detection would be there done by machine learning and deep learning. This task can be performed with various machine learning techniques such as SVM, random forest, regression etc but for getting a better accuracy the neural networks have been used. The main motive of the paper is to provide a better accuracy.This shows a higher degree of overlap between the automated (machine) derived tumour zone and the manually extracted tumour region by radiologists In compared to state-of-the-art methodologies, the simulation results demonstrate the significance in terms of quality parameters and accuracy.

KeywordsAnalysis, machine learning,brain tumor prediction, deep learning, artificial neural network


    Brain, being the controller of the entire body, plays the most important role in making a body function properly The disease that can occur in brain is known as the brain tumour which is nothing but basically a mass growth of abnormal cells in the brain. The disease is very harmful as there are various types of brain tumours, some of them are non-cancerous while some of them are cancerous. These can also be known as benign which are non-cancerous and malignant which are cancerous. They generally occur in the brain, and also they are so harmful that they start spreading to the other parts of the brain. The main problem is how quickly the tumor can grow is the determining factor for the functioning of the nervous system. The treatment is varied as it depends on the type of the tumor as well as the location of the tumor as where it is happening.Some of the symptoms and signs of this tumors can be difficulty with balance, speech difficulties, feeling very tired after doing a small work, unable to follow what one says etc. There have been many cases where the cause of tumor isnt clear, hence some of the ways it can occur is through exposure to radiation or if the family has brain tumors and it is transferred through the genes. Now the Detection is easy as there are modern computerized

    techniques such as machin learning and deep learning techniques which play a very crucial role in detecting the disease and helps in the prediction at an early stage where the brain tumor can be avoided and kept to be a benign one.


    Much work has been done in this following field and some of it are, in [1] the main was to prove the results that had been obtained which was an average of 0.82 dice similarity index and it was better because it indicated better overlap between the extracted tumor region with manually extracted tumor by the radiologists.

    In [2] the author tests different deep learning methods which are ResNet, DenseNet,Mobile Net and after researching all and performing the models he got an accuracy of 91.8, 92.8, 92.9 respectively.In [3] The image segmentation has been performed through TK means clustering to detect the tumor. The results that has been show in that has better accuracy and a very reduced execution time than the other existing schemes.In [4]The main is to calculate the region of the tumer with different ensemble methods like KNN-RF-DT combined together and made into vote which make a good voting method. With a dataset of 2556 images they have received an accuracy of 97.3%.


    1. DATA PREPARATION Data Being the most important part of the entire model that needs to be performed needs to be carefully collected or taken. As the collection process is not an easy task, it is better to take a dataset readily available. The dataset that is taken is from Brain MRI images for brain tumor detection. The dataset consists of high quality images of the MRI scan acquired from the patients and the classification out there in the database is based on the two classes no tumor-0 or tumor-1.In figure 1 we can see the glimpse of the dataset with two classes- tumor and no tumor.

      Figure 1: Dataset Overview


      The preprocessing plays an important role in determining the output of the model, i.e. the accuracy of the model. The features/attributes may have some noises and variations which need to be figured out and improved to make it a perfect dataset. In the dataset used by the authors data augmentation technique is used to increase the size of the dataset images which would be beneficial for the model to get the features from the images and perform better. Mainly, data augmentation is used to increase the amount of the data, which in our case is to increase the dataset image size by slightly modified copies of the existing or newly created data. It also plays a role as regularizer and helps reduce overfitting when training the model. In figure 2 a glimpse of the augmentation done of 1 image is done. After that the next step would be to divide the dataset in to training dataset and testing dataset. The main aim for doing so is to train the algorithm better so it can show actual results on the unknown data which is presented to the model in the form of testing data.

      Figure 2: Augmentation Example


      Features/attribuutes are the main components on which the model that has been used works as if there are no features the models wont be having a deciding factor for prediction as a result the accuracy would be very less hence more features means more the accuracy and better the model will perform. Visualization plays an important role in this as it helps to know which feature/attribute has a more weightage in determining the success of the model.In the figure 3 entire step of feature extraction has been displayed, in first step the original image is taken, followed by step 2 in which the biggest contour is found and then the extreme points are collected and finally cropping is done.

      Figure 3: Feature extraction steps


    ANN stands for Artificial Neural Network, and it is built of the structure and function of the biological neural network, which is the primary reason for the construction of the Artificial neural network architecture. It is often made up of neurons that are arranged in layers and neurons that are comparable to those found in the human brain. The input

    signal is absorbed by a processing unit, while the output layer provides network output. In these linked configurations, the input and output layers are always present, as they are in all network topologies[10]. The Hidden layer is the third layer, and it keeps neurons out of the input and output layers. These neurons are hidden from anyone engaging with the system and behave as a black box. The system's computing and processing capabilities can be increased by adding more hidden layers containing neurons, but the system's trainingphenomena grow more complicated at the same time. TensorFlow is used to execute the ANN; to begin, we would use the sequential function to initialize the ANN. After it, we'd add levels to it, starting with the input layer, which would contain an activation function called'relu.' The activation function is used to determine if the output of a neural network is yes or no. It translates numbers ranging from -1 to 1 or 0 to 1, and so on. The procedure would be repeated for the four hidden layers, and then the sigmoid would be used as the activation function for the output layer. The ANN is then compiled, with Adam optimizer as the loss function, binary crossentropy as the loss function, and accuracy as the metrics.


    From collecting or taking the dataset from the MRI images of the patients to augmenting the images to make the dataset clear for the algorithm imposed and then applying the deep learning neural networks i.e. Artificial neural networks and performing 100 epochs gives us an accuracy of 97 percent after the 100th epoch. As we know more the epochs, more it will be able to train the dataset well and then finally the testing of the dataset on the remaining test cases which were removed from the dataset gives us the accuracy of the model. After that a confusion matrix of the testing data is created which tells us about how the algorithm has performed and it is good aspect in terms of analysis.

    Figure 4: Accuracy of model

    Figure 6: Confusion Matrix


    The Brain tumour detection decision support system assists and helps doctors in making the best, most accurate, and quickest decisions possible while also reducing total treatment costs. By predicting strokes at an early stage, the suggested technique lowers treatment costs and improves quality of life. We were able to attain a stunning 97 percent accuracy on a specific dataset by using Artificial Neural Networks, which is a tremendous outcome in terms of science and creativity and will help us have fewer people die from malignancies and offer required aid at an early stage.


    This topic has played an important role in the research and the medical advancement that is happening out there in the world, hence it should be researched more by taking a real time dataset of different human beings and then analysing it on the basis of the model by doing the preprocessing and all. The next step would be to try the databases on different ensemble models, which can be used to get a better accuracy and can be helped in reducing the complexity of the segmentation in very less time than the techniques which are already present in the brain tumor research.



  2. Sadad T, Rehman A, Munir A, Saba T, Tariq U, Ayesha N, Abbasi

    R. Brain tumor detection and multi-classification using advanced deep learning techniques. Microsc Res Tech. 2021 Jun;84(6):1296- 1308. doi: 10.1002/jemt.23688. Epub 2021 Jan 5. PMID: 33400339.

  3. Md Khairul Islam, Md Shahin Ali, Md Sipon Miah, Md Mahbubur Rahman, Md Shahariar Alam, Mohammad Amzad Hossain,

  4. Brain tumor detection in MR image using superpixels, principal component analysis and template based K-means clustering algorithm,

  5. Machine Learning with Applications,Volume 5,2021,100044,ISSN 2666-8270,


  7. A. Wulandari, R. Sigit and M. M. Bachtiar, "Brain Tumor Segmentation to Calculate Percentage Tumor Using MRI," 2018 International Electronics Symposium on Knowledge Creation and Intelligent Computing (IES-KCIC), 2018, pp. 292-296, doi: 10.1109/KCIC.2018.8628591.

  8. R. Tamilselvi, A. Nagaraj, M. P. Beham and M. B. Sandhiya, "BRAMSIT: A Database for Brain Tumor Diagnosis and Detection," 2020 Sixth International Conference on Bio Signals, Images, and Instrumentation (ICBSII), 2020, pp. 1-5, doi: 10.1109/ICBSII49132.2020.9167530.

  9. M. Gurbin, M. Lascu and D. Lascu, "Tumor Detection and Classification of MRI Brain Image using Different Wavelet Transforms and Support Vector Machines," 2019 42nd International Conference on Telecommunications and Signal Processing (TSP), 2019, pp. 505-508, doi: 10.1109/TSP.2019.8769040.

  10. D. Hirahara, "Preliminary assessment for the development of CADe system for brain tumor in MRI images utilizing transfer learning in Xception model," 2019 IEEE 8th Global Conference on Consumer Electronics (GCCE), 2019, pp. 922-924, doi: 10.1109/GCCE46687.2019.9015529.

  11. S. Somasundaram and R. Gobinath, "Current Trends on Deep Learning Models for Brain Tumor Segmentation and Detection A Review," 2019 International Conference on Machine Learning, Big Data, Cloud and Parallel Computing (COMITCon), 2019, pp. 217- 221, doi: 10.1109/COMITCon.2019.8862209.

  12. T. M. Shahriar Sazzad, K. M. Tanzibul Ahmmed, M. U. Hoque and

    M. Rahman, "Development of Automated Brain Tumor Identification Using MRI Images," 2019 International Conference

    on Electrical, Computer and Communication Engineering (ECCE), 2019, pp. 1-4, doi: 10.1109/ECACE.2019.8679240.

  13. H. S. Abdulbaqi, M. Zubir Mat, A. F. Omar, I. S. Bin Mustafa and

    L. K. Abood, "Detecting brain tumor in Magnetic Resonance Images using Hidden Markov Random Fields and Threshold techniques," 2014 IEEE Student Conference on Research and Development, 2014, pp. 1-5, doi: 10.1109/SCORED.2014.7072963.

  14. V. Zeljkovic et al., "Automatic brain tumor detection and segmentation in MR images," 2014 Pan American Health Care Exchanges (PAHCE), 2014, pp. 1-1, doi: 10.1109/PAHCE.2014.6849645.

  15. N. Kurat and N. Özkaya, "Automaticly extracting brain tumor from MR image," 2014 22nd Signal Processing and Communications Applications Conference (SIU), 2014, pp. 1532-1535, doi: 10.1109/SIU.2014.6830533.

  16. S. Modi and M. H. Bohara, "Facial Emotion Recognition using Convolution Neural Network," 2021 5th International Conference on Intelligent Computing and Control Systems (ICICCS), 2021, pp. 1339-1344, doi: 10.1109/ICICCS51141.2021.9432156.

  17. A. Sorokin et al., "Multi-label classification of brain tumor mass spectrometry data In pursuit of tumor boundary detection method," 2017 International Conference on Intelligent Informatics and Biomedical Sciences (ICIIBMS), 2017, pp. 169-171, doi: 10.1109/ICIIBMS.2017.8279736.

  18. Y. Sharma and Y. K. Meghrajani, "Brain tumor extraction from MRI image using mathematical morphological reconstruction," 2014 2nd International Conference on Emerging Technology Trends in Electronics, Communication and Networking, 2014, pp. 1-4, doi: 10.1109/ET2ECN.2014.7044982.

  19. S. Becker, A. Mang, A. Toma and T. M. Buzug, "Approximating tumor induced brain deformation using directly manipulated free form deformation," 2010 IEEE International Symposium on Biomedical Imaging: From Nano to Macro, 2010, pp. 85-88, doi: 10.1109/ISBI.2010.5490409.

  20. V. Shah and S. Modi, "Comparative Analysis of Psychometric Prediction System," 2021 Smart Technologies, Communication and Robotics (STCR), 2021, pp. 1-5, doi: 10.1109/STCR51658.2021.9588950.

  21. H. Chetty, M. Shah, S. Kabaria and S. Verma, "A Survey on brain tumor extraction approach from mri images using image processing," 2017 2nd International Conference for Convergence in Technology (I2CT), 2017, pp. 534-538, doi: 10.1109/I2CT.2017.8226187.

  22. D. Marszalik and W. Rczka, "Surgical tool trajectory optimization in brain tumour resection," 2019 20th International Carpathian Control Conference (ICCC), 2019, pp. 1-4, doi: 10.1109/CarpathianCC.2019.8766052.

Leave a Reply

Your email address will not be published. Required fields are marked *