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

Shui Hua Wang, Yi Ding Lv, Yuxiu Sui, Shuai Liu, Su Jing Wang, Yu Dong Zhang*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

136 Citations (Scopus)

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.

Original languageEnglish
Article number2
JournalJournal of Medical Systems
Volume42
Issue number1
DOIs
Publication statusPublished - 1 Jan 2018
Externally publishedYes

Keywords

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

Fingerprint

Dive into the research topics of 'Alcoholism Detection by Data Augmentation and Convolutional Neural Network with Stochastic Pooling'. Together they form a unique fingerprint.

Cite this