Personality Trait Recognition using ECG Spectrograms and Deep Learning.

Journal: Annual International Conference of the IEEE Engineering in Medicine and Biology Society. IEEE Engineering in Medicine and Biology Society. Annual International Conference
PMID:

Abstract

This paper presents an innovative approach to recognizing personality traits using deep learning (DL) methods applied to electrocardiogram (ECG) signals. The research explores the potential of ECG-derived spectrograms as informative features in detecting the Big Five personality traits model encompassing extraversion, neuroticism, agreeableness, conscientiousness, and openness. Optimal window sizes for spectrogram generation are determined, and a convolutional neural network (CNN), specifically Resnet-18, and visual transformer (ViT) are employed for feature extraction and personality trait classification. The study utilizes the publicly available ASCERTAIN dataset, which comprises various physiological signals, including ECG recordings, collected from 58 participants while presenting video stimuli categorized by valence and arousal levels. The outcomes of this study demonstrate noteworthy performance in personality trait classification, consistently achieving F1-scores exceeding 0.9 across different window sizes and personality traits. These results emphasize the viability of ECG signal spectrograms as a valuable modality for personality trait recognition, with Resnet-18 exhibiting effectiveness in discerning distinct personality traits.

Authors

  • Muhammad Mohsin Altaf
  • Saadat Ullah Khan
  • Muhammad Majid
    Department of Computer Engineering, University of Engineering and Technology Taxila, Taxila, 47050, Pakistan. m.majid@uettaxila.edu.pk.
  • Syed Muhammad Anwar
    Software Engineering Department, University of Engineering and Technology, Taxila, Pakistan.