Abstract
Brain tumor classification using Magnetic Resonance Imaging (MRI) is a pivotal area in medical diagnostics, with the potential to influence early detection and subsequent treatment strategies. Over the years, various machine learning and deep learning models have been proposed to enhance the accuracy of MRI-based tumor detection. In this evolving landscape, this paper introduces a distinctive deep-learning model that harnesses the power of a soft attention mechanism. We employ a meticulously designed Convolutional Neural Network (CNN) comprising four convolution layers. One of the significant innovations in our approach is the method of feature extraction. Instead of extracting features solely from the final layer, as is common in many models, our approach aggregates and combines features from all layers. This ensures that the vital characteristics intrinsic to each layer are not lost but rather amalgamated into a robust and comprehensive feature vector. The incorporation of a soft attention mechanism at the terminal stages ensures that the most salient and clinically relevant features are emphasized, enhancing classification accuracy. To validate the efficacy of our proposed model, we employed standard datasets for training and testing. A comparative analysis with existing state-of-the-art models affirms the superiority and potential of our approach in the realm of MRI-based brain tumor classification.
Similar content being viewed by others
References
Cheng J (2017) Brain tumor dataset. figshare. Dataset. https://doi.org/10.6084/m9.figshare.1512427.v5
Deepak S, Ameer PM (2019) Brain tumor classification using deep CNN features via transfer learning. Comput Biol Med 111:103345
Ghassemi N, Shoeibi A, Rouhani M (2020) Deep neural network with generative adversarial networks pre-training for brain tumor classification based on MR images. Biomed Signal Process Control 57:101678
Vankdothu R, Hameed MA, Fatima H (2022) A brain tumor identification and classification using deep learning based on CNN-LSTM method. Comput Electr Eng 101:107960
Sultan HH, Salem NM, Al-Atabany W (2019) Multi-classification of brain tumor images using deep neural network. IEEE access 7:6921569225
Raza A, Ayub H, Khan JA, Ahmad I, Salama SA, Daradkeh YI, Hamam H (2022) A hybrid deep learning-based approach for brain tumor classification. Electronics 11(7):1146
Saleh A, Sukaik R, Abu-Naser SS (2020) Brain tumor classification using deep learning. 2020 International Conference on Assistive and Rehabilitation Technologies (iCareTech). IEEE, pp 131–136
Kumar S, Mankame DP (2020) Optimization driven deep convolution neural network for brain tumor classification. Biocybern Biomed Eng 40(3):1190–1204
Abbood AA, Shallal QM, Fadhel MA, Shallal QM (2021) Automated brain tumor classification using various deep learning models: a comparative study. Indones J Electr Eng Computer Sci 22(1):252–259
Tandel GS, Tiwari A, Kakde OG (2021) Performance optimisation of deep learning models using majority voting algorithm for brain tumour classification. Comput Biol Med 135:104564
Asif S, Zhao M, Tang F, Zhu Y (2023) An enhanced deep learning method for multi-class brain tumor classification using deep transfer learning. Multimed Tools Appl 82:31709–31736
Siar M, Teshnehlab M (2019) Brain tumor detection using deep neural network and machine learning algorithm. 2019 9th international conference on computer and knowledge engineering (ICCKE). IEEE, pp 363–368
Pashaei A, Sajedi H, Jazayeri N (2018) Brain tumor classification via convolutional neural network and extreme learning machines. 2018 8th International conference on computer and knowledge engineering (ICCKE). IEEE, pp 314–319
Saba T, Mohamed AS, El-Affendi M, Amin J, Sharif M (2020) Brain tumor detection using fusion of hand crafted and deep learning features. Cogn Syst Res 59:221–230
Ismael SAA, Mohammed A, Hefny H (2020) An enhanced deep learning approach for brain cancer MRI images classification using residual networks. Artif Intell Med 102:101779
Çınar N, Kaya B, Kaya M (2022) Comparison of deep learning models for brain tumor classification using MRI images. 2022 International conference on decision aid sciences and applications (DASA). IEEE, pp 1382–1385
Rahman T, Islam MS (2022) Mri brain tumor classification using deep convolutional neural network. 2022 International Conference on Innovations in Science, Engineering and Technology (ICISET). IEEE, pp 451–456
Havaei M, Davy A, Warde-Farley D, Biard A, Courville A, Bengio Y, Larochelle H (2017) Brain tumor segmentation with deep neural networks. Med Image Anal 35:18–31
Öksüz C, Urhan O, Güllü MK (2022) Brain tumor classifica-¨ tion using the fused features extracted from expanded tumor region. Biomed Signal Process Control 72:103356
Ashwini K, Dash R (2022) Soft attention with convolutional neural network for grading diabetic retinopathy. 2022 IEEE 19th India Council International Conference (INDICON). IEEE, pp 1–6
Ghosal P, Nandanwar L, Kanchan S, Bhadra A, Chakraborty J, Nandi D (2019) Brain tumor classification using ResNet-101 based squeeze and excitation deep neural network. 2019 Second International Conference on Advanced Computational and Communication Paradigms (ICACCP). IEEE, pp 1–6
Kalaiselvi T, Padmapriya ST, Sriramakrishnan P et al (2020) Deriving tumor detection models using convolutional neural networks from MRI of human brain scans. Int j inf tecnol 12:403–408. https://doi.org/10.1007/s41870-020-00438-4
Bhairnallykar ST, Narawade V (2023) Segmentation of MR images using DN convolutional neural network. Int j inf tecnol. https://doi.org/10.1007/s41870-023-01461-x
Esmaeilzadeh Asl S, Chehel Amirani M, Seyedarabi H (2023) Brain tumors segmentation using a hybrid filtering with U-Net architecture in multimodal MRI volumes. Int j inf tecnol. https://doi.org/10.1007/s41870-023-01485-3
Singh R, Agarwal BB (2023) An automated brain tumor classification in MR images using an enhanced convolutional neural network. Int J Inf Tecnol 15:665–674. https://doi.org/10.1007/s41870-022-01095-5
Jain S, Jain V (2023) Novel approach to classify brain tumor based on transfer learning and deep learning. Int j inf tecnol 15:2031–2038. https://doi.org/10.1007/s41870-023-01259-x
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interest
The authors have no relevant financial or nonfinancial interests to disclose.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Mohanty, B.C., Subudhi, P.K., Dash, R. et al. Feature-enhanced deep learning technique with soft attention for MRI-based brain tumor classification. Int. j. inf. tecnol. 16, 1617–1626 (2024). https://doi.org/10.1007/s41870-023-01701-0
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s41870-023-01701-0