Browse > Article
http://dx.doi.org/10.22937/IJCSNS.2022.22.5.21

Enhanced CNN Model for Brain Tumor Classification  

Kasukurthi, Aravinda (CSBS department, RVR & JC College of Engineering)
Paleti, Lakshmikanth (CSE department, Kallam Haranadhareddy Institute of Technology)
Brahmaiah, Madamanchi (CSBS department, RVR & JC College of Engineering)
Sree, Ch.Sudha (CSBS department, RVR & JC College of Engineering)
Publication Information
International Journal of Computer Science & Network Security / v.22, no.5, 2022 , pp. 143-148 More about this Journal
Abstract
Brain tumor classification is an important process that allows doctors to plan treatment for patients based on the stages of the tumor. To improve classification performance, various CNN-based architectures are used for brain tumor classification. Existing methods for brain tumor segmentation suffer from overfitting and poor efficiency when dealing with large datasets. The enhanced CNN architecture proposed in this study is based on U-Net for brain tumor segmentation, RefineNet for pattern analysis, and SegNet architecture for brain tumor classification. The brain tumor benchmark dataset was used to evaluate the enhanced CNN model's efficiency. Based on the local and context information of the MRI image, the U-Net provides good segmentation. SegNet selects the most important features for classification while also reducing the trainable parameters. In the classification of brain tumors, the enhanced CNN method outperforms the existing methods. The enhanced CNN model has an accuracy of 96.85 percent, while the existing CNN with transfer learning has an accuracy of 94.82 percent.
Keywords
Multi-classification; CNN model; Grid search technic; Hyper parameter optimization;
Citations & Related Records
연도 인용수 순위
  • Reference
1 Iqbal, S., Khan, M.U.G., Saba, T. and Rehman, A., 2018. Computer-assisted brain tumor type discrimination using magnetic resonance imaging features. Biomedical Engineering Letters, 8(1), pp.5-28.   DOI
2 Ghassemi, N., Shoeibi, A. and Rouhani, M., 2020. Deep neural network with generative adversarial networks pre-training for brain tumor classification based on MR images. Biomedical Signal Processing and Control, 57, p.101678.   DOI
3 Sultan, H.H., Salem, N.M. and Al-Atabany, W., 2019. Multi-classification of brain tumor images using deep neural network. IEEE Access, 7, pp.69215-69225.   DOI
4 Deepak, S. and Ameer, P.M., 2019. Brain tumor classification using deep CNN features via transfer learning. Computers in biology and medicine, 111, p.103345.   DOI
5 Ismael, S.A.A., Mohammed, A. and Hefny, H., 2020. An enhanced deep learning approach for brain cancer MRI images classification using residual networks. Artificial Intelligence in Medicine, 102, p.101779.   DOI
6 Sajjad, M., Khan, S., Muhammad, K., Wu, W., Ullah, A. and Baik, S.W., 2019. Multi-grade brain tumor classification using deep CNN with extensive data augmentation. Journal of computational science, 30, pp.174-182.   DOI
7 Sharif, M.I., Li, J.P., Khan, M.A. and Saleem, M.A., 2020. Active deep neural network features selection for segmentation and recognition of brain tumors using MRI images. Pattern Recognition Letters, 129, pp.181-189.   DOI
8 Swati, Z.N.K., Zhao, Q., Kabir, M., Ali, F., Ali, Z., Ahmed, S. and Lu, J., 2019. Brain tumor classification for MR images using transfer learning and fine-tuning. Computerized Medical Imaging and Graphics, 75, pp.34-46.   DOI
9 Ronneberger, O., Fischer, P. and Brox, T., 2015, October. U-net: Convolutional networks for biomedical image segmentation. In International Conference on Medical image computing and computer-assisted intervention (pp. 234-241). Springer, Cham.
10 Kendall, A., Badrinarayanan, V. and Cipolla, R., 2015. Bayesian segnet: Model uncertainty in deep convolutional encoder-decoder architectures for scene understanding. arXiv preprint arXiv:1511.02680.
11 Yin, B., Wang, C. and Abza, F., 2020. New brain tumor classification method based on an improved version of whale optimization algorithm. Biomedical Signal Processing and Control, 56, p.101728.   DOI
12 Anaraki, A.K., Ayati, M. and Kazemi, F., 2019. Magnetic resonance imaging-based brain tumor grades classification and grading via convolutional neural networks and genetic algorithms. Biocybernetics and Biomedical Engineering, 39(1), pp.63-74.   DOI
13 Kaur, T. and Gandhi, T.K., 2020. Deep convolutional neural networks with transfer learning for automated brain image classification. Machine Vision and Applications, 31, pp.1-16.   DOI
14 Huang, Z., Du, X., Chen, L., Li, Y., Liu, M., Chou, Y. and Jin, L., 2020. Convolutional Neural Network Based on Complex Networks for Brain Tumor Image Classification With a Modified Activation Function. IEEE Access, 8, pp.89281-89290.   DOI
15 Gumaei, A., Hassan, M.M., Hassan, M.R., Alelaiwi, A. and Fortino, G., 2019. A hybrid feature extraction method with regularized extreme learning machine for brain tumor classification. IEEE Access, 7, pp.36266-36273.   DOI
16 Amin, J., Sharif, M., Gul, N., Yasmin, M. and Shad, S.A., 2020. Brain tumor classification based on DWT fusion of MRI sequences using convolutional neural network. Pattern Recognition Letters, 129, pp.115-122.   DOI
17 Tandel, G.S., Balestrieri, A., Jujaray, T., Khanna, N.N., Saba, L. and Suri, J.S., 2020. Multiclass magnetic resonance imaging brain tumor classification using artificial intelligence paradigm. Computers in Biology and Medicine, p.103804.   DOI
18 Lin, G., Milan, A., Shen, C. and Reid, I., 2017. Refinenet: Multi-path refinement networks for high-resolution semantic segmentation. In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 1925-1934).
19 Lin, G., Liu, F., Milan, A., Shen, C. and Reid, I., 2019. Refinenet: Multi-path refinement networks for dense prediction. IEEE transactions on pattern analysis and machine intelligence, 42(5), pp.1228-1242.   DOI
20 Cheng, Jun, et al. "Enhanced Performance of Brain Tumor Classification via Tumor Region Augmentation and Partition." PloS one 10.10 (2015).