Comparison of Wavelet Transformations to Enhance Convolutional Neural Network Performance in Brain Tumor Segmentation Publisher Pubmed



Hajiabadi M¹ ; Alizadeh Savareh B^{2, 3} ; Emami H⁴ ; Bashiri A⁵ Show Affiliations
Source: BMC Medical Informatics and Decision Making Published:2021

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. © 2021, The Author(s).