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

• Published in: Journal of big data Vol. 8; no. 1; pp. 1 - 20
• Main Authors: Christian, Hans, Suhartono, Derwin, Chowanda, Andry, Zamli, Kamal Z.
• Format: Journal Article
• Language: English
• Published: Cham Springer International Publishing 17.05.2021; Springer Nature B.V; SpringerOpen
• Subjects: Algorithms; BERT; Big Data; Communications Engineering; Computational Science and Engineering; Computer Science; Context; Data Mining and Knowledge Discovery; Data sources; Database Management; Datasets; Deep learning; Digital media; Feature extraction; Information Storage and Retrieval; Job satisfaction; Language model; Machine learning; Mathematical Applications in Computer Science; Natural language processing; Networks; Personality; Personality prediction; Personality tests; Prediction models; Social media; Social networks; Deep learning; Natural language processing; BERT; Language model; Social media; Personality prediction
• ISSN: 2196-1115; 2196-1115
• DOI: 10.1186/s40537-021-00459-1

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.