Alcoholism Identification Based on an AlexNet Transfer Learning Model.
FRONTIERS IN PSYCHIATRY(2019)
摘要
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 con figurations 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.
更多查看译文
关键词
alcoholism,transfer learning,AlexNet,data augmentation,convolutional neural network,dropout,local response normalization,magnetic resonance imaging
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络