Text based personality prediction from multiple social media data sources using pre-trained language model and model averaging

Journal of Big Data

Hans Christian¹  Derwin Suhartono²  Andry Chowanda²  Kamal Z. Zamli³ 

[1] Computer Science Department, BINUS Graduate Program, Master of Computer Science, Bina Nusantara University, 11480, Jakarta, Indonesia;Computer Science Department, School of Computer Science, Bina Nusantara University, 11480, Jakarta, Indonesia;Faculty of Computing, College of Computing and Applied Sciences, Universiti Malaysia Pahang, 26600, Pahang, Malaysia;

关键词: Personality prediction;    Natural language processing;    Social media;    Deep learning;    BERT;    Language model;   

DOI  :  10.1186/s40537-021-00459-1

来源: Springer

【 摘 要 】

The ever-increasing social media users has dramatically contributed to significant growth as far as the volume of online information is concerned. Often, the contents that these users put in social media can give valuable insights on their personalities (e.g., in terms of predicting job satisfaction, specific preferences, as well as the success of professional and romantic relationship) and getting it without the hassle of taking formal personality test. Termed personality prediction, the process involves extracting the digital content into features and mapping it according to a personality model. Owing to its simplicity and proven capability, a well-known personality model, called the big five personality traits, has often been adopted in the literature as the de facto standard for personality assessment. To date, there are many algorithms that can be used to extract embedded contextualized word from textual data for personality prediction system; some of them are based on ensembled model and deep learning. Although useful, existing algorithms such as RNN and LSTM suffers from the following limitations. Firstly, these algorithms take a long time to train the model owing to its sequential inputs. Secondly, these algorithms also lack the ability to capture the true (semantic) meaning of words; therefore, the context is slightly lost. To address these aforementioned limitations, this paper introduces a new prediction using multi model deep learning architecture combined with multiple pre-trained language model such as BERT, RoBERTa, and XLNet as features extraction method on social media data sources. Finally, the system takes the decision based on model averaging to make prediction. Unlike earlier work which adopts a single social media data with open and close vocabulary extraction method, the proposed work uses multiple social media data sources namely Facebook and Twitter and produce a predictive model for each trait using bidirectional context feature combine with extraction method. Our experience with the proposed work has been encouraging as it has outperformed similar existing works in the literature. More precisely, our results achieve a maximum accuracy of 86.2% and 0.912 f1 measure score on the Facebook dataset; 88.5% accuracy and 0.882 f1 measure score on the Twitter dataset.

【 授权许可】

CC BY