Back to Search Start Over

Continuous Output Personality Detection Models via Mixed Strategy Training

Authors :
Wang, Rong
Sun, Kun
Publication Year :
2024

Abstract

The traditional personality models only yield binary results. This paper presents a novel approach for training personality detection models that produce continuous output values, using mixed strategies. By leveraging the PANDORA dataset, which includes extensive personality labeling of Reddit comments, we developed models that predict the Big Five personality traits with high accuracy. Our approach involves fine-tuning a RoBERTa-base model with various strategies such as Multi-Layer Perceptron (MLP) integration, and hyperparameter tuning. The results demonstrate that our models significantly outperform traditional binary classification methods, offering precise continuous outputs for personality traits, thus enhancing applications in AI, psychology, human resources, marketing and health care fields.

Details

Database :
arXiv
Publication Type :
Report
Accession number :
edsarx.2406.16223
Document Type :
Working Paper