Wavelet-enhanced convolutional neural network: a new idea in a deep learning paradigm

Biomed Tech (Berl). 2019 Apr 24;64(2):195-205. doi: 10.1515/bmt-2017-0178.

Abstract

Purpose: Manual brain tumor segmentation is a challenging task that requires the use of machine learning techniques. One of the machine learning techniques that has been given much attention is the convolutional neural network (CNN). The performance of the CNN can be enhanced by combining other data analysis tools such as wavelet transform.

Materials and methods: In this study, one of the famous implementations of CNN, a fully convolutional network (FCN), was used in brain tumor segmentation and its architecture was enhanced by wavelet transform. In this combination, a wavelet transform was used as a complementary and enhancing tool for CNN in brain tumor segmentation.

Results: Comparing the performance of basic FCN architecture against the wavelet-enhanced form revealed a remarkable superiority of enhanced architecture in brain tumor segmentation tasks.

Conclusion: Using mathematical functions and enhancing tools such as wavelet transform and other mathematical functions can improve the performance of CNN in any image processing task such as segmentation and classification.

Keywords: brain tumor; convolutional neural network; segmentation; wavelet transform.

MeSH terms

  • Brain Neoplasms / classification
  • Deep Learning*
  • Humans
  • Image Processing, Computer-Assisted / methods*
  • Machine Learning
  • Neural Networks, Computer
  • Wavelet Analysis