Ensemble neural network approach detecting pain intensity from facial expressions.

Journal: Artificial intelligence in medicine
Published Date:

Abstract

This paper reports on research to design an ensemble deep learning framework that integrates fine-tuned, three-stream hybrid deep neural network (i.e., Ensemble Deep Learning Model, EDLM), employing Convolutional Neural Network (CNN) to extract facial image features, detect and accurately classify the pain. To develop the approach, the VGGFace is fine-tuned and integrated with Principal Component Analysis and employed to extract features in images from the Multimodal Intensity Pain database at the early phase of the model fusion. Subsequently, a late fusion, three layers hybrid CNN and recurrent neural network algorithm is developed with their outputs merged to produce image-classified features to classify pain levels. The EDLM model is then benchmarked by means of a single-stream deep learning model including several competing models based on deep learning methods. The results obtained indicate that the proposed framework is able to outperform the competing methods, applied in a multi-level pain detection database to produce a feature classification accuracy that exceeds 89 %, with a receiver operating characteristic of 93 %. To evaluate the generalization of the proposed EDLM model, the UNBC-McMaster Shoulder Pain dataset is used as a test dataset for all of the modelling experiments, which reveals the efficacy of the proposed method for pain classification from facial images. The study concludes that the proposed EDLM model can accurately classify pain and generate multi-class pain levels for potential applications in the medical informatics area, and should therefore, be explored further in expert systems for detecting and classifying the pain intensity of patients, and automatically evaluating the patients' pain level accurately.

Authors

  • Ghazal Bargshady
    School of Management and Enterprise, University of Southern Queensland, Springfield, Qld 4300, Australia. Electronic address: ghazal.bargshady@usq.edu.au.
  • Xujuan Zhou
    Centre for Health Informatics, Australian Institute of Health Innovation, The University of New South Wales, Sydney, NSW 2052, Australia.
  • Ravinesh C Deo
    School of Agricultural Computational and Environmental Sciences, International Centre of Applied Climate Science (ICACS), University of Southern Queensland, Springfield, QLD, 4300, Australia. ravinesh.deo@usq.edu.au.
  • Jeffrey Soar
    School of Management and Enterprise, University of Southern Queensland, Springfield, Qld 4300, Australia. Electronic address: jeffrey.soar@usq.edu.au.
  • Frank Whittaker
    School of Management and Enterprise, University of Southern Queensland, Springfield, Qld 4300, Australia. Electronic address: Frank.Whittaker@usq.edu.au.
  • Hua Wang
    Department of Orthopaedics, The Second Xiangya Hospital of Central South University, Changsha, Hunan, China.