Application of bi-modal signal in the classification and recognition of drug addiction degree based on machine learning.

Journal: Mathematical biosciences and engineering : MBE
Published Date:

Abstract

Most studies on drug addiction degree are made based on statistical scales, addicts' account, and subjective judgement of rehabilitation doctors. No objective, quantified evaluation has been made. This paper uses devises the synchronous bimodal signal collection and experimentation paradigm with electroencephalogram (EEG) and forehead high-density near-infrared spectroscopy (NIRS) device. The drug addicts are classified into mild, moderate and severe groups with reference to the suggestions of researchers and medical experts. Data of 45 drug addicts (mild: 15; moderate: 15; and severe: 15) is collected, and then used to design an addiction degree testing algorithm based on decision fusion. The algorithm is used to classify mild, moderate and severe addiction. This paper pioneers to use two types of Convolutional Neural Network (CNN) to abstract the EEG and NIR data of drug addicts, and introduces batch normalization to CNN, thus accelerating training process, reducing parameter sensitivity, and enhancing system robustness. The characteristics output by two CNNs are transformed into dimensions. Two new characteristics are assigned with a weight of 50% each. The data is used for decision fusion. In the networks, 27 subjects are used as training sets, 9 as validation sets, and 9 as testing sets. The 3-class accuracy remains to be 63.15%, preliminarily justifying this method as an effective approach to measure drug addiction degree. And the method is ready to use, objective, and offers results in real time.

Authors

  • Xuelin Gu
    School of Mechanical and Electrical Engineering and Automation, Shanghai University, Shanghai 200444, China.
  • Banghua Yang
    School of Mechatronic Engineering and Automation, Research Center of Brain Computer Engineering, Shanghai University, Shanghai, 200444, China. yangbanghua@126.com.
  • Shouwei Gao
    School of Mechanical and Electrical Engineering and Automation, Shanghai University, Shanghai 200444, China.
  • Lin Feng Yan
    Department of Radiology & Functional and Molecular Imaging Key Lab of Shaanxi Province, Tangdu Hospital, Fourth Military Medical University, Xi'an, Shaanxi 710038, China.
  • Ding Xu
    Shanghai Drug Rehabilitation Administration Bureau, Shanghai 200080, China.
  • Wen Wang
    Clinical and Research Center for Infectious Diseases, Beijing Youan Hospital, Capital Medical University, Beijing, China.