Alcoholism Identification Based on an AlexNet Transfer Learning Model

Front Psychiatry. 2019 Apr 11:10:205. doi: 10.3389/fpsyt.2019.00205. eCollection 2019.

Abstract

Aim: This paper proposes a novel alcoholism identification approach that can assist radiologists in patient diagnosis. Method: AlexNet was used as the basic transfer learning model. The global learning rate was small, at 10-4, and the iteration epoch number was at 10. The learning rate factor of replaced layers was 10 times larger than that of the transferred layers. We tested five different replacement configurations of transfer learning. Results: The experiment shows that the best performance was achieved by replacing the final fully connected layer. Our method yielded a sensitivity of 97.44%± 1.15%, a specificity of 97.41 ± 1.51%, a precision of 97.34 ± 1.49%, an accuracy of 97.42 ± 0.95%, and an F1 score of 97.37 ± 0.97% on the test set. Conclusion: This method can assist radiologists in their routine alcoholism screening of brain magnetic resonance images.

Keywords: AlexNet; alcoholism; convolutional neural network; data augmentation; dropout; local response normalization; magnetic resonance imaging; transfer learning.