An Effective Personality Prediction Using Bidirectional Encoder Representations from Transformers with AdamW Optimizer
In recent days, each and every individual personality reflects the individual behavior of person. Therefore, it is essential to assess the personality that enhances the behavior of individuals across the world. Previous researchers have used various machine learning and deep learning analyzes individuals' traits across psychology, human resources, marketing, and mental health. However, predicting personality is challenging due to the subjective nature of traits, ethical considerations around data privacy, and the need for models that work accurately across diverse backgrounds. This study suggests a novel approach to develop a deep learning model that resolves these challenges by combining various techniques such as Bidirectional Encoder Representations from Transformers (BERT), AdamW Optimizer (BERT-AdamW). Initially, the data is collected from the Myers-Briggs Type Indicator (MBTI) dataset. Further, the data is preprocessed using lowercasing, lemmatization, and stemming and processed through a pre-trained language model namely BERT for feature extraction of optimal features. In addition, the AdamW optimizer is utilized to enhance the performance of BERT and finally LR model is incorporated to predict the personality behavior of an individual. The BERT-AdamW model resulted with 97.89% accuracy when compared with Long Short-Term Memory (LSTM).
Authors
- K. R. Subhashini (ORCID: https://orcid.org/0009-0007-5535-7241)
- Ammar Hameed Shnain (ORCID: https://orcid.org/0000-0002-6991-4607)
- Haider Alabdeli
- G. Geetha Ramani
- T Devi.
Institutions
- Iraqi University (IQ)
- Saveetha University (IN)
Publication Details
- Published
- 2024-11-22
- DOI
- https://doi.org/10.1109/iciics63763.2024.10859743
- Citations
- 1
- Primary Topic
- Personality Traits and Psychology
- Type
- article
- Field-Weighted Citation Impact
- 0.64