
Premium content
Access to this content requires a subscription. You must be a premium user to view this content.

workshop paper
NU at WASSA 2024 Empathy and Personality Shared Task: Enhancing Personality Predictions with Knowledge Graphs; A Graphical Neural Network and LightGBM Ensemble Approach
keywords:
graphical neural networks
gats. gcns
llms
personality prediction
knowledge graphs
This paper proposes a novel ensemble ap- proach that combines Graph Neural Networks (GNNs) and LightGBM to enhance personal- ity prediction based on the personality Big 5 model. By integrating BERT embed- dings from user essays with knowledge graph- derived embeddings, our method accurately captures rich semantic and relational informa- tion. Additionally, a special loss function that combines Mean Squared Error (MSE), Pear- son correlation loss, and contrastive loss to im- prove model performance is introduced. The proposed ensemble model, made of Graph Convolutional Networks (GCNs), Graph At- tention Networks (GATs), and LightGBM, demonstrates superior performance over other models, with significant improvements in pre- diction accuracy for the Big Five personality traits achieved. Our system officially ranked 2nd at the Track 4: PER track