Personality Prediction Model: An Enhanced Machine Learning Approach

In today’s digital era, social media platforms like Instagram have become deeply embedded in daily life, generating billions of content items each day. This vast stream of publicly accessible data presents a unique opportunity for researchers to gain insights into human behaviour and personality. However, leveraging such unstructured and highly variable data for psychological analysis introduces significant challenges, including data sparsity, noise, and ethical considerations around privacy. This study addresses these challenges by exploring the potential of machine learning to infer personality traits from Instagram content. Motivated by the growing demand for scalable, non-intrusive methods of psychological assessment, we developed a personality prediction system combining convolutional neural networks (CNNs) and random forest (RF) algorithms. Our model is grounded in the Big Five Personality framework, which includes Extraversion, Agreeableness, Conscientiousness, Neuroticism, and Openness. Using data collected with informed consent from 941 participants, we extracted visual features from their Instagram images using two pretrained CNNs, which were then used to train five RF models, each targeting a specific trait. The proposed system achieved an average mean absolute error of 0.1867 across all traits. Compared to the PAN-2015 benchmark, our method demonstrated competitive performance. These results highlight that using social media data for personality prediction offers potential applications in personalized content delivery, mental health monitoring, and human–computer interactions.