Skip to main content
Log in

Brain tumor detection using deep ensemble model with wavelet features

  • Original Paper
  • Published:
Health and Technology Aims and scope Submit manuscript

Abstract

Purpose

A person's healthy activities are determined by the state of his or her brain. The brain is in charge of all of a person's activities. If a small abnormality develops in the brain, it will have a negative impact on the person regardless of whether the other organs are in good condition. As a result, early detection of any abnormal growth in the brain is essential.

Methods

In this work, the authors have utilized data pre-processing using discrete wavelet transform (DWT) and segmentation, whereas, for detection, an ensemble learning technique is proposed. DWT and segmentation help in increasing the dataset size that is used to train the deep learning model. Segmentation using supervised Auto-encoder (AE) is used for data enhancement to strengthen the training process. The original data, outputs of DWT, and segmented images are utilized for the training of the ensemble model designed with three parallel-connected convolutional neural networks (CNNs).

Results

The detection results obtained from the ensemble of these recurrent models are then passed through the Multilayer Perceptron (MLP) for final detection. Kaggle brain MRI image dataset is used to complete the proposed method. Test accuracy, F1-score, precision, sensitivity, and specificity provided by this method are 98.08%, 0.9836, 1.0000, 0.9677, and 1.0000 respectively. In comparison to state-of-the-art models, the proposed model produces competitive outcomes.

Conclusion

In time detection of the tumor may lead to the survival of the patient. Automatic and accurate detection is another perspective of this field. For this purpose, we have proposed a deep ensemble model with wavelet features. The ensemble model provides increased performance in comparison to single models due to the parallel training.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9

Similar content being viewed by others

Availability of data and material

The data used in this work is cited.

Code availability

Not applicable.

References

  1. Mohan G, Subashini MM. MRI based medical image analysis: Survey on brain tumor grade classification. Biomed Signal Process Control. 2018;39:139–61.

    Article  Google Scholar 

  2. Ker J, Singh SP, Bai Y, Rao J, Lim T, Wang L. Image thresholding improves 3-dimensional convolutional neural network diagnosis of different acute brain hemorrhages on computed tomography scans. Sensors. 2019;19(9):2167.

    Article  Google Scholar 

  3. Ker J, Bai Y, Lee HY, Rao J, Wang L. Automated brain histology classification using machine learning. J Clin Neurosci. 2019;66:239–45.

    Article  Google Scholar 

  4. Ker J, Wang L, Rao J, Lim T. Deep learning applications in medical image analysis. Ieee Access. 2017;6:9375–89.

    Article  Google Scholar 

  5. Bai X, Niwas SI, Lin W, Ju BF, Kwoh CK, Wang L, Sng CC, Aquino MC, Chew PT. Learning ECOC code matrix for multiclass classification with application to glaucoma diagnosis. J Med Syst. 2016;40(4):1–10.

    Article  Google Scholar 

  6. He G, B., Cui, J., Xiao, B., and Wang, X. Image analysis by two types of Franklin-Fourier moments. IEEE/CAA Journal of Automatica Sinica. 2019;6(4):1036–51.

    Article  MathSciNet  Google Scholar 

  7. Mitra S, Shankar BU. Medical image analysis for cancer management in natural computing framework. Inf Sci. 2015;306:111–31.

    Article  Google Scholar 

  8. Jyoti A, Mohanty MN, Kumar MP. Morphological based segmentation of brain image for tumor detection. In 2014 International Conference on Electronics and Communication Systems (ICECS) 2014;1–5. IEEE.

  9. Hochreiter S, Schmidhuber J. Long short-term memory. Neural Comput. 1997;9(8):1735–80.

    Article  Google Scholar 

  10. Chung J, Gulcehre C, Cho K, Bengio Y. Empirical evaluation of gated recurrent neural networks on sequence modeling. 2014. arXiv preprint arXiv:1412.3555.

  11. Das A, Patra, GR, Mohanty MN. LSTM based Odia handwritten numeral recognition. In 2020 international conference on communication and signal processing (ICCSP). 2020;0538–0541. IEEE.

  12. Jia Z, Chen D. Brain Tumor Identification and Classification of MRI images using deep learning techniques. IEEE Access. 2020.

  13. Majib MS, Rahman MM, Sazzad TS, Khan NI, Dey SK. VGG-SCNet: A VGG Net-Based Deep Learning Framework for Brain Tumor Detection on MRI Images. IEEE Access. 2021;9:116942–52.

    Article  Google Scholar 

  14. Mallick PK, Ryu SH, Satapathy SK, Mishra S, Nguyen GN, Tiwari P. Brain MRI image classification for cancer detection using deep wavelet autoencoder-based deep neural network. IEEE Access. 2019;7:46278–87.

    Article  Google Scholar 

  15. Khan P, Kader MF, Islam SR, Rahman AB, Kamal MS, Toha MU, Kwak KS. Machine Learning and Deep Learning Approaches for Brain Disease Diagnosis: Principles and Recent Advances. IEEE Access. 2021;9:37622–55.

    Article  Google Scholar 

  16. Çelik G, Talu MF. A new 3D MRI segmentation method based on Generative Adversarial Network and Atrous Convolution. Biomed Signal Process Control. 2022;71: 103155.

    Article  Google Scholar 

  17. Dikici E, Ryu JL, Demirer M, Bigelow M, White RD, Slone W, Prevedello LM. Automated brain metastases detection framework for T1-weighted contrast-enhanced 3D MRI. IEEE J Biomed Health Inform. 2020;24(10):2883–93.

    Article  Google Scholar 

  18. Takrouni W, Douik A. Improving geometric P-norm-based glioma segmentation through deep convolutional autoencoder encapsulation. Biomed Signal Process Control. 2022;71: 103232.

    Article  Google Scholar 

  19. Noreen N, Palaniappan S, Qayyum A, Ahmad I, Imran M, Shoaib M. A deep learning model based on concatenation approach for the diagnosis of brain tumor. IEEE Access. 2020;8:55135–44.

    Article  Google Scholar 

  20. Li M, Kuang L, Xu S, Sha Z. Brain tumor detection based on multimodal information fusion and convolutional neural network. IEEE Access. 2019;7:180134–46.

    Article  Google Scholar 

  21. Sekhar A, Biswas S, Hazra R, Sunaniya AK, Mukherjee A, Yang L. Brain tumor classification using fine-tuned GoogLeNet features and machine learning algorithms: IoMT enabled CAD system. IEEE J Biomed Health Inform. 2021.

  22. Hao Q, Pei Y, Zhou R, Sun B, Sun J, Li S, Kang X. Fusing Multiple Deep Models for In Vivo Human Brain Hyperspectral Image Classification to Identify Glioblastoma Tumor. IEEE Trans Instrum Meas. 2021;70:1–14.

    Google Scholar 

  23. Tang Z, Yap PT, Shen D. A new multi-atlas registration framework for multimodal pathological images using conventional monomodal normal atlases. IEEE Trans Image Process. 2018;28(5):2293–304.

    Article  MathSciNet  Google Scholar 

  24. Ghassemi N, Shoeibi A, Rouhani M. Deep neural network with generative adversarial networks pre-training for brain tumor classification based on MR images. Biomed Signal Process Control. 2020;57: 101678.

    Article  Google Scholar 

  25. Chandra SK, Bajpai MK. Fractional Crank-Nicolson finite difference method for benign brain tumor detection and segmentation. Biomed Signal Process Control. 2020;60: 102002.

    Article  Google Scholar 

  26. Nema S, Dudhane A, Murala S, Naidu S. RescueNet: An unpaired GAN for brain tumor segmentation. Biomed Signal Process Control. 2020;55: 101641.

    Article  Google Scholar 

  27. Ghahfarrokhi SS, Khodadadi H. Human brain tumor diagnosis using the combination of the complexity measures and texture features through magnetic resonance image. Biomed Signal Process Control. 2020;61: 102025.

    Article  Google Scholar 

  28. Karayegen G, Aksahin MF. Brain tumor prediction on MR images with semantic segmentation by using deep learning network and 3D imaging of tumor region. Biomed Signal Process Control. 2021;66: 102458.

    Article  Google Scholar 

  29. Rai HM, Chatterjee K, Dashkevich S. Automatic and accurate abnormality detection from brain MR images using a novel hybrid UnetResNext-50 deep CNN model. Biomed Signal Process Control. 2021;66: 102477.

    Article  Google Scholar 

  30. Hedayati R, Khedmati M, Taghipour-Gorjikolaie M. Deep feature extraction method based on ensemble of convolutional auto encoders: Application to Alzheimer’s disease diagnosis. Biomed Signal Process Control. 2021;66: 102397.

    Article  Google Scholar 

  31. Aghalari M, Aghagolzadeh A, Ezoji M. Brain tumor image segmentation via asymmetric/symmetric UNet based on two-pathway-residual blocks. Biomed Signal Process Control. 2021;69: 102841.

    Article  Google Scholar 

  32. Das A, Mohapatra SK, Mohanty MN. Design of deep ensemble classifier with fuzzy decision method for biomedical image classification. Appl Soft Comput. 2022;115: 108178.

    Article  Google Scholar 

  33. Das A, Mohanty MN, Mallick PK, Tiwari P, Muhammad K, Zhu H. Breast cancer detection using an ensemble deep learning method. Biomed Signal Process Control. 2021;70: 103009.

    Article  Google Scholar 

  34. Xiao L, Wan X, Han Z. PHY-layer authentication with multiple landmarks with reduced overhead. IEEE Trans Wireless Commun. 2017;17(3):1676–87.

    Article  Google Scholar 

  35. Xiao L, Wan X, Lu X, Zhang Y, Wu D. IoT security techniques based on machine learning: How do IoT devices use AI to enhance security? IEEE Signal Process Mag. 2018;35(5):41–9.

    Article  Google Scholar 

  36. Deng L, Platt J. Ensemble deep learning for speech recognition. In Proc. interspeech. 2014.

  37. Xu L, Jiang JH, Zhou YP, Wu HL, Shen GL, Yu RQ. MCCV stacked regression for model combination and fast spectral interval selection in multivariate calibration. Chemom Intell Lab Syst. 2007;87(2):226–30.

    Article  Google Scholar 

  38. Sakkis G, Androutsopoulos I, Paliouras G, Karkaletsis V, Spyropoulos CD, Stamatopoulos P. Stacking classifiers for anti-spam filtering of e-mail. 2001. arXiv preprint cs/0106040.

  39. Aversano T, L., Bernardi, M. L., and Cimitile, M. Water stress classification using Convolutional Deep Neural Networks. J Univ Comput Sci. 2022;28(3):311–28.

    Google Scholar 

  40. Taştimur CANAN, Akın E. Fastener Classification Using One-Shot Learning with Siamese Convolution Networks. J Univ Comput Sci. 2022;28(1).

  41. Suong LK, Kwon J. Detection of Potholes Using a Deep Convolutional Neural Network. J Univ Comput Sci. 2018;24(9):1244–1257

  42. Krstinic D, Braovic M, Bozic-Stulic D. Convolutional Neural Networks and Transfer Learning Based Classification of Natural Landscape Images. J Univ Comput Sci. 2020;26(2):244–67.

    Google Scholar 

  43. Das AK, Pati SK, Huang HH, Chen CK. Cancer Classification by Gene Subset Selection from Microarray Dataset. J Univers Comput Sci. 2018;24(6):682–710.

    MathSciNet  Google Scholar 

  44. Demirci MERVE, Sağıroğlu ŞEREF. TwitterBulletin: An Intelligent and Real-Time Automated News Categorization Tool for Twitter. J Univ Comput Sci. 2022;28.

  45. Panda S, Das A, Mishra S, Mohanty MN. Epileptic seizure detection using deep ensemble network with empirical wavelet transform. Measurement Science Review. 2021;21(4):110–6.

    Article  Google Scholar 

  46. Nguyen TT, Nguyen MP, Pham XC, Liew AWC. Heterogeneous classifier ensemble with fuzzy rule-based meta learner. Inf Sci. 2018;422:144–60.

    Article  Google Scholar 

  47. Das A, Mohanty MN. Design of Encoder in CNN for Effective Recognition of Odia Numerals. In 2021 1st Odisha International Conference on Electrical Power Engineering, Communication and Computing Technology (ODICON). 2021;1–3. IEEE.

  48. Chakrabarty N. Brain MRI Images for Brain Tumor Detection. 2019. Available: https://www.kaggle.com/navoneel/brain-mri-images-for-brain-tumor-detection.

  49. Huang Z, Du X, Chen L, Li Y, Liu M, Chou Y, Jin L. Convolutional neural network based on complex networks for brain tumor image classification with a modified activation function. IEEE Access. 2020;8:89281–90.

    Article  Google Scholar 

  50. Huda S, Yearwood J, Jelinek HF, Hassan MM, Fortino G, Buckland M. A hybrid feature selection with ensemble classification for imbalanced healthcare data: A case study for brain tumor diagnosis. IEEE access. 2016;4:9145–54.

    Article  Google Scholar 

  51. Ge C, Gu IYH, Jakola AS, Yang J. Enlarged training dataset by pairwise GANs for molecular-based brain tumor classification. IEEE Access. 2020;8:22560–70.

    Article  Google Scholar 

  52. Ertosun MG, Rubin DL. Automated grading of gliomas using deep learning in digital pathology images: A modular approach with ensemble of convolutional neural networks. In AMIA Annual Symposium Proceedings. 2015; 2015:1899. American Medical Informatics Association.

Download references

Funding

The authors did not receive support from any organization for the submitted work.

Author information

Authors and Affiliations

Authors

Contributions

D.K. Sahoo went through the literature survey, data collection and worked on designing the ensemble model, A. Das worked on the preprocessing steps and corrected the writing, S. Mishra and M.N. Mohanty formulated the problem and supervised the work.

Corresponding author

Correspondence to Mihir Narayan Mohanty.

Ethics declarations

Conflict of interest

The authors declare that they have no conflict of interest.

Additional information

Publisher's Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Springer Nature or its licensor 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.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Sahoo, D.K., Das, A., Mishra, S. et al. Brain tumor detection using deep ensemble model with wavelet features. Health Technol. 12, 1157–1167 (2022). https://doi.org/10.1007/s12553-022-00699-y

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s12553-022-00699-y

Keywords

Navigation