Format

Send to

Choose Destination
J Med Syst. 2017 Nov 17;42(1):2. doi: 10.1007/s10916-017-0845-x.

Alcoholism Detection by Data Augmentation and Convolutional Neural Network with Stochastic Pooling.

Author information

1
Department of Informatics, University of Leicester, Leicester, LE1 7RH, UK.
2
School of Electronic Science and Engineering, Nanjing University, Nanjing, Jiangsu, 210046, China.
3
Department of Psychiatry, Nanjing Brain Hospital, Nanjing Medical University, Nanjing, Jiangsu, 210029, China.
4
College of Computer Science, Inner Mongolia University, Hohhot, 010012, China. cs_liushuai@imu.edu.cn.
5
Key Laboratory of Behavior Sciences, Institute of Psychology, Chinese Academy of Sciences, Beijing, 100101, China. wangsujing@psych.ac.cn.
6
Department of Informatics, University of Leicester, Leicester, LE1 7RH, UK. yudongzhang@ieee.org.

Abstract

Alcohol use disorder (AUD) is an important brain disease. It alters the brain structure. Recently, scholars tend to use computer vision based techniques to detect AUD. We collected 235 subjects, 114 alcoholic and 121 non-alcoholic. Among the 235 image, 100 images were used as training set, and data augmentation method was used. The rest 135 images were used as test set. Further, we chose the latest powerful technique-convolutional neural network (CNN) based on convolutional layer, rectified linear unit layer, pooling layer, fully connected layer, and softmax layer. We also compared three different pooling techniques: max pooling, average pooling, and stochastic pooling. The results showed that our method achieved a sensitivity of 96.88%, a specificity of 97.18%, and an accuracy of 97.04%. Our method was better than three state-of-the-art approaches. Besides, stochastic pooling performed better than other max pooling and average pooling. We validated CNN with five convolution layers and two fully connected layers performed the best. The GPU yielded a 149× acceleration in training and a 166× acceleration in test, compared to CPU.

KEYWORDS:

Alcohol use disorder; Average pooling; Convolutional neural network; Data augmentation; Graphical processing unit; Max pooling; Stochastic pooling

PMID:
29159706
DOI:
10.1007/s10916-017-0845-x
[Indexed for MEDLINE]

Supplemental Content

Full text links

Icon for Springer
Loading ...
Support Center