BMC Medical Informatics and Decision Making (Nov 2021)

Comparison of wavelet transformations to enhance convolutional neural network performance in brain tumor segmentation

  • Mohamadreza Hajiabadi,
  • Behrouz Alizadeh Savareh,
  • Hassan Emami,
  • Azadeh Bashiri

DOI
https://doi.org/10.1186/s12911-021-01687-4
Journal volume & issue
Vol. 21, no. 1
pp. 1 – 11

Abstract

Read online

Abstract Introduction and goal to background Due to the importance of segmentation of MRI images in identifying brain tumors, various methods including deep learning have been introduced for automatic brain tumor segmentation. On the other hand, using a combination of methods can improve their performance. Among them is the use of wavelet transform as an auxiliary element in deep networks. The analysis of the requirements of such combinations has been addressed in this study. Method In this developmental study, different wavelet functions were used to compress brain MRI images and finally as an auxiliary element in improving the performance of the convolutional neural network in brain tumor segmentation. Results Based on the results of the tests performed, the Daubechies1 function was most effective in enhancing network performance in segmenting MRI images and was able to balance the performance and computational overload. Conclusion Choosing the wavelet function to optimize the performance of a convolutional neural network should be based on the requirements of the problem, also taking into account some considerations such as computational load, processing time, and performance of the wavelet function in optimizing CNN output in the intended task.

Keywords