Pathological brain detection based on AlexNet and transfer learning

Abstract

The aim of this study is to automatically detect pathological brain in magnetic resonance images (MRI) based on deep learning structure and transfer learning. Deep learning is now the hottest topic both in academia and industry. However, the volume of brain MRI datasets are usually too small to train the entire deep learning structure. The training can be easily trapped into overfitting. Therefore, we introduced transfer learning to train the deep neural network. Firstly, we obtained the pre-trained AlexNet structure. Then, we replaced parameters of the last three layers with random weights and the rest parameters served as the initial values. Finally, we trained the modified model with our MRI dataset. Experiment results suggested that our method achieved accuracy of 100.00%, which outperformed state-of-the-art approaches