International Journal of Computer Science & Network Security

International Journal of Computer Science & Network Security (국제컴퓨터통신보호논문지학회)
권호 표지
DOI QR코드

DOI QR Code

Enhanced CNN Model for Brain Tumor Classification

  • Received : 2022.05.05
  • Published : 2022.05.30
Download PDF
⟨ Previous Next ⟩

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

References

  1. 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. https://doi.org/10.1109/access.2019.2904145
  2. 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. https://doi.org/10.1016/j.patrec.2019.11.016
  3. 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. https://doi.org/10.1016/j.bspc.2019.101678
  4. 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. https://doi.org/10.1016/j.bspc.2019.101728
  5. 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. https://doi.org/10.1016/j.compbiomed.2020.103804
  6. 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. https://doi.org/10.1109/access.2019.2919122
  7. 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. https://doi.org/10.1007/s13534-017-0050-3
  8. 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. https://doi.org/10.1016/j.compbiomed.2019.103345
  9. 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. https://doi.org/10.1016/j.artmed.2019.101779
  10. 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. https://doi.org/10.1016/j.patrec.2019.11.019
  11. 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. https://doi.org/10.1016/j.jocs.2018.12.003
  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. https://doi.org/10.1016/j.bbe.2018.10.004
  13. 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. https://doi.org/10.1016/j.compmedimag.2019.05.001
  14. 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. https://doi.org/10.1007/s00138-019-01050-8
  15. 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. https://doi.org/10.1109/access.2020.2993618
  16. 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.
  17. 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).
  18. 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. https://doi.org/10.1109/tpami.2019.2893630
  19. 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.
  20. Cheng, Jun, et al. "Enhanced Performance of Brain Tumor Classification via Tumor Region Augmentation and Partition." PloS one 10.10 (2015).