MS Identification in Brain Magnetic Resonance Images Using Wavelet Transfer Learning

Authors

1 Department of Computer Engineering, Rasht Branch, Islamic Azad University, Rasht, Iran

2 Department of Computer Engineering, Karaj Branch, Islamic Azad University, Karaj, Iran

Abstract

Multiple Sclerosis (MS) is one of the most important diseases of the central nervous system. This disease causes small lesions detectable in Magnetic Resonance Imaging (MRI) images of the patient’s brain. Because of the small size of the lesions, their distribution, and their similarity to some other diseases, the MS diagnosis can be difficult for specialists and may be mistaken. In this paper, we presented a new method based on deep learning for the automatic classification of MRI images. The proposed method is a combinational architecture from transfer learning and wavelet transform (WT). First, WT was applied to the input MRI image, and its four output sub-bands are used as the input of four fine-tuning networks based on EfficientNet-B3. Transfer learning networks perform feature extraction on all four sub-bands. Then, their outputs are combined, and the result is classified by a fully connected neural network. Due to the feature of WT to extract local features, it was possible to highlight the lesions in the images and subsequently classify it with higher accuracy and precision. Various criteria have been used to evaluate the proposed method. The results of the experiments show that the Values of accuracy, precision, sensitivity, and specificity are 98.91%, 99.20%, 99.20%, and 98.33%, respectively.

Keywords