Can Artificial Neural Networks Predict Psychiatric Conditions Associated with Cannabis Use?

  • Daniel Stamate
  • Wajdi Alghamdi
  • Daniel Stahl
  • Alexander Zamyatin
  • Robin Murray
  • Marta di Forti
Conference paper
Part of the IFIP Advances in Information and Communication Technology book series (IFIPAICT, volume 519)


This data-driven computational psychiatry research proposes a novel machine learning approach to developing predictive models for the onset of first-episode psychosis, based on artificial neural networks. The performance capabilities of the predictive models are enhanced and evaluated by a methodology consisting of novel model optimisation and testing, which integrates a phase of model tuning, a phase of model post-processing with ROC optimisation based on maximum accuracy, Youden and top-left methods, and a model evaluation with the k-fold cross-testing methodology. We further extended our framework by investigating the cannabis use attributes’ predictive power, and demonstrating statistically that their presence in the dataset enhances the prediction performance of the neural network models. Finally, the model stability is explored via simulations with 1000 repetitions of the model building and evaluation experiments. The results show that our best Neural Network model’s average accuracy of predicting first-episode psychosis, which is evaluated with Monte Carlo, is above 80%.


Machine learning Neural networks Prediction modelling ROC optimisation Monte carlo Computational psychiatry Cannabis Psychosis 


  1. 1.
    United Nations Office on Drugs and Crime, World Drug Report, United Nations publication, Sales No. E.16.XI.7 (2016)Google Scholar
  2. 2.
    Radhakrishnan, R., Wilkinson, S., Dsouza, D.: Gone to pot: a review of the association between cannabis and psychosis. Front. Psychiatry 5 (2014)Google Scholar
  3. 3.
    Di Forti, M., Marconi, A., et al.: Proportion of patients in South London with first-episode psychosis attributable to use of high potency cannabis: a case-control study. Lancet Psychiatry 2(3), 233–238 (2015)CrossRefGoogle Scholar
  4. 4.
    Alghamdi, W., Stamate, D., et al.: A prediction modelling and pattern detection approach for the first-episode psychosis associated to cannabis use. In: 15th IEEE International Conference on Machine Learning and Applications, pp. 825–830 (2016)Google Scholar
  5. 5.
    Zhou, H., Tang, J., Zheng, H.: Machine learning for medical applications. Sci. World J., 20, 1 (2015)Google Scholar
  6. 6.
    Iniesta, R., Stahl, D., McGuffin P.: Machine learning, statistical learning and the future of biological research in psychiatry, psychological medicine (2016)CrossRefGoogle Scholar
  7. 7.
    Kuhn, M., Johnson, K.: Applied Predictive Modelling. Springer (2013)CrossRefGoogle Scholar
  8. 8.
    Katrinecz, A., Stamate, D., et al.: Predicting psychosis using the experience sampling method with mobile apps. In: 16th IEEE International Conference on Machine Learning and Applications (2017)Google Scholar
  9. 9.
    Belgrave, D., Cassidy, R., Stamate, D., et al.: Predictive modelling strategies to understand heterogeneous manifestations of asthma in early life. In: 16th IEEE International Conference on Machine Learning and Applications (2017)Google Scholar
  10. 10.
    Elveren, E., Yumuşak, N.: Tuberculosis disease diagnosis using artificial neural network trained with genetic algorithm. J. Med. Syst. 35, 329–332 (2011)CrossRefGoogle Scholar
  11. 11.
    Adjouadi, M., Ayala, M., et al.: Classification of leukaemia blood samples using neural networks. Ann. Biomed. Eng. 38(4), 1473–1482 (2010)CrossRefGoogle Scholar
  12. 12.
    Yan, Y., Qin, X., et al.: A restricted Boltzmann machine based two-lead electrocardiography classification. In: Proceedings of 12th International Conference on Wearable Implantable Body Sensor Networks (2015)Google Scholar
  13. 13.
    Liaw, A., Wiener, M.: Classification and regression by random forest. R News 2(3), 18–22 (2002)Google Scholar
  14. 14.
    Qazi, N., Raza, K.: Effect of feature selection, SMOTE and under sampling on class imbalance classification. In: 2012 UKSim 14th International Conference on Computer Modelling and Simulation, pp. 145–150 (2012)Google Scholar
  15. 15.
    Bohning, D., Bohning, W., et al.: Revisiting Youden’s index as a useful measure of the misclassification error in a meta-analysis of diagnostic studies. Stat. Methods Med. Res. 17, 543–554 (2008)MathSciNetCrossRefGoogle Scholar
  16. 16.
    Pepe, M.: The Statistical Evaluation of Medical Tests for Classification and Prediction. Oxford University Press, New York (2003)zbMATHGoogle Scholar
  17. 17.
    Perkins, N., Schisterman, F.: The inconsistency of “optimal” cutpoints obtained using two criteria based on the receiver operating characteristic curve. Am. J. Epidemiol. 163(7), 670–675 (2006)CrossRefGoogle Scholar
  18. 18. The Comprehensive R Archive Network (2018). Accessed 2 Jan 2018

Copyright information

© IFIP International Federation for Information Processing 2018

Authors and Affiliations

  • Daniel Stamate
    • 1
  • Wajdi Alghamdi
    • 1
  • Daniel Stahl
    • 2
  • Alexander Zamyatin
    • 3
  • Robin Murray
    • 4
  • Marta di Forti
    • 5
  1. 1.Data Science and Soft Computing Lab, Department of Computing, GoldsmithsUniversity of LondonLondonUK
  2. 2.Department of Biostatistics and Health Informatics, Institute of Psychiatry, Psychology and NeuroscienceKing’s College LondonLondonUK
  3. 3.Faculty of Informatics, Department of Applied InformaticsNational Research Tomsk State UniversityTomskRussia
  4. 4.Department of Psychosis Studies, Institute of Psychiatry, Psychology and NeuroscienceKing’s College LondonLondonUK
  5. 5.MRC Social, Genetic and Developmental Psychiatry Centre, Institute of Psychiatry, Psychology and NeuroscienceKing’s College LondonLondonUK

Personalised recommendations