2023, Volume 7 Issue 2
A Novel Brain Tumor Classification Model Using Machine Learning Techniques
AUTHOR(S)
Saikat Sundar Pal, Prithwish Raymahapatra, Soumyadeep Paul, Sajal Dolui, Dr. Avijit Kumar Chaudhuri, Sulekha Das
DOI: https://doi.org/10.46647/ijetms.2023.v07i02.011
ABSTRACT
The objective of this research work is to classify brain tumor images into 4 different classes by using Convolutional Neural Network (CNN) algorithm i.e. a deep learning method with VGG16 architecture. The four classes are pituitary, glioma, meningioma, and no tumor. The dataset used for this research is a publicly available MRI Image dataset of brain tumor with 7023 images. The methodology followed in this project includes data pre-processing, model building, and evaluation. The dataset is pre-processed by resizing the images to 64x64 and normalizing the pixel values. The VGG16 architecture is used to build the CNN model, and it is trained on the pre-processed data for 10 epochs with a batch size of 64. The model is evaluated using the area under the operating characteristic curve (AUC) metric of the receiver. The results of this project show that the CNN model with VGG16 architecture achieves an AUC of 0.92 for classifying brain tumor images into four different classes. The model performs best for classifying meningioma with AUC of 0.90, followed by pituitary with AUC of 0.91, glioma with AUC of 0.93, and no tumor with AUC of 0.89. In conclusion, the CNN model with VGG16 architecture is an effective approach for classifying brain tumor images into multiple classes. The model achieves high accuracy in identifying different types of brain tumors, which could potentially aid in early diagnosis and treatment of brain tumors.
Page No: 87 - 98
References:
[1]https://www.kaggle.com/datasets/masoudnickparvar/brain-tumor-mri-dataset?select=Training
[2]https://www.researchgate.net/publication/346391164_Study_on_image_processing_using_deep_learning_techniques
[3] Machhale K, Nandpuru HB, Kapur V, Kosta L. MRI brain cancer classification using hybrid classifier (SVM-KNN). In: 2015 International Conference on Industrial Instrumentation and Control (ICIC). IEEE; 2015, pp. 60–65
[4] Zacharaki EI, Wang S, Chawla S, et al. Classification of brain tumor type and grade using MRI texture and shape in a machine learning scheme. Magn Reson Med. 2009;62:1609- 1618. https://doi.org/10.1002/mrm.22147
[5] Zacharaki EI, Wang S, Chawla S, et al. Classification of brain tumor type and grade using MRI texture and shape in a machine learning scheme. Magn Reson Med. 2009;62:1609- 1618. https://doi.org/10.1002/mrm.22147
[6] Afshar P, Mohammadi A, Plataniotis KN. Brain tumor type classification via capsule networks. In: 2018 25th IEEE International Conference on Image Processing (ICIP). IEEE; 2018, pp. 3129–3133.
[7] Zia R, Akhtar P, Aziz A. A new rectangular window based image cropping method for generalization of brain neoplasm classification systems. Int J Imaging Syst Technol. 2018;28:153- 162. https://doi.org/10.1002/ima.22266
[8] Sajjad M, Khan S, Muhammad K, et al. Multi-grade brain tumor classification using deep CNN with extensive data augmentation. J Comput Sci. 2019;30:174-182. https://doi.org/10. 1016/j.jocs.2018.12.003
[9] Badža MM, Barjaktarovic MČ. Classification of brain tumors from MRI images using a convolutional neural network. Appl Sci. 2020;10:1999. https://doi.org/10.3390/app10061999
[10] Cheng J, Huang W, Cao S, et al. Enhanced performance of brain tumor classification via tumor region augmentation and partition. PLoS One. 2015;10:e0140381. https://doi.org/10. 1371/journal.pone.0140381
[11] Paul JS, Plassard AJ, Landman BA, Fabbri D. Deep learning for brain tumor classification. Proc SPIE. 2017;10137: 1013710
[12] Huang Z, Du X, Chen L, et al. Convolutional neural network based on complex networks for brain tumor image classification with a modified activation function. IEEE Access. 2020;8:89281-89290. https://doi.org/10.1109/ACCE SS.2020.2993618
[13] Afshar P, Plataniotis KN, Mohammadi A. Capsule networks for brain tumor classification based on MRI images and coarse tumor boundaries. In: ICASSP 2019 – 2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). IEEE; 2019, pp. 1368–1372.
[14] Hussain L, Awan IA, Aziz W, et al. Detecting congestive heart failure by extracting multimodal features and employing machine learning techniques. Biomed Res Int. 2020;2020:1-19. https://doi.org/10.1155/2020/4281243
[15] Hussain L, Saeed S, Awan IA, et al. Detecting brain tumor using machine learning techniques based on different features extracting strategies. Curr Med Imaging Rev. 2018;14:595-606. https://doi.org/10.2174/1573405614666180718123533
[16] Fazli S, Pour HM, Bouzari H. Particle filter based object tracking with sift and color feature. 2009 2nd Int Conf Mach Vision, ICMV. 2009;2009:89-93. https://doi.org/10.1109/ICMV. 2009.47
[17]Pal, S. S., Paul, S., Dey, R., Das, S., & Chaudhuri, A. K. Determining the probability of poverty levels of the Indigenous Americans and Black Americans in US using Multiple Regression.
[18]Bialynicky-Birula I, Mycielski J. Relations for information entropy in wave mechanics. Commun Math Phys. 1975;44: 129-132
[19] Chen T, Xu R, He Y, Wang X. Improving sentiment analysis via sentence type classification using BiLSTM-CRF and CNN. Expert Syst Appl. 2017;72:221-230. https://doi.org/10.1016/j. eswa.2016.10.065
[20] He K, Zhang X, Ren S, Sun J. Deep residual learning for image recognition. In: IEEE Conference on Computer Vision and Pattern Recognition; 2016.
[21]Ray, A., & Chaudhuri, A. K. (2021). Smart healthcare disease diagnosis and patient management: Innovation, improvement and skill development. Machine Learning with Applications, 3, 100011.
[22] Qassim, H., Verma, A., & Feinzimer, D. (2018, January). Compressed residual-VGG16 CNN model for big data places image recognition. In 2018 IEEE 8th annual computing and communication workshop and conference (CCWC) (pp. 169-175). IEEE.
[23] Girshick, R. (2015). Fast r-cnn. In Proceedings of the IEEE international conference on computer vision (pp. 1440-1448).
[24] Krizhevsky, A., Sutskever, I., & Hinton, G. E. (2017). Imagenet classification with deep convolutional neural networks. Communications of the ACM, 60(6), 84-90.
[25] Shorten, C., & Khoshgoftaar, T. M. (2019). A survey on image data augmentation for deep learning. Journal of big data, 6(1), 1-48.
[26] Lobo, J. M., Jiménez‐Valverde, A., & Real, R. (2008). AUC: a misleading measure of the performance of predictive distribution models. Global ecology and Biogeography, 17(2), 145-151.
[27] Saha, S., Mondal, J., Arnam Ghosh, M., Das, S., & Chaudhuri, A. K. Prediction on the Combine Effect of Population, Education and Unemployment on Criminal Activity Using Machine Learning.
[28] Jalal deen K; Karthigai Priya G; Magesh B; Kubendran R. "Identification of CT Lung Tumor Using Fuzzy Clustering Algorithm". International Research Journal on Advanced Science Hub, 3, Special Issue ICEST 1S, 2021, 30-33. doi: 10.47392/irjash.2021.016
[29] Hari Prasada Raju Kunadharaju; Sandhya N.; Raghav Mehra. "Detection of Brain Tumor Using Unsupervised Enhanced K-Means, PCA and Supervised SVM Machine Learning Algorithms". International Research Journal on Advanced Science Hub, 2, Special Issue ICSTM 12S, 2020, 62-67. doi: 10.47392/irjash.2020.262
[30] Naveenkumar C; Praveenkumar M; Praveenkumar S; Karthik V; Parthasarathy .. "Study on the Effect different materials in Traffic signal pole using numerical simulation approach". International Research Journal on Advanced Science Hub, 1, 1, 2019, 33-41. doi: 10.47392/irjash.2019.06
[31] Gladia Angeline P; Jagadesh S; Jeba Christina D; Nevetha G; Poornima L. "Efficient and Enhanced Data Encryption in Cloud Computing". International Research Journal on Advanced Science Hub, 2, 3, 2020, 1-4. doi: 10.47392/irjash.2020.16
How to Cite This Article:
Saikat Sundar Pal, Prithwish Raymahapatra, Soumyadeep Paul, Sajal Dolui, Dr. Avijit Kumar Chaudhuri, Sulekha Das
. A Novel Brain Tumor Classification Model Using Machine Learning Techniques
. ijetms;7(2):87-98. DOI: 10.46647/ijetms.2023.v07i02.011