Low-resource MobileBERT for emotion recognition in imbalanced text datasets mitigating challenges with limited resources

M Muhammad Hussain C Caikou Chen S Sami S. Albouq K Khlood Shinan F Fatmah Alanazi M Muhammad Waseem Iqbal M M. Usman Ashraf

Abstract

Modern dialogue systems rely on emotion recognition in conversation (ERC) as a core element enabling empathetic and human-like interactions. However, the weak correlation between emotions and semantics poses significant challenges to emotion recognition in dialogue. Semantically similar utterances can express different types of emotions, depending on the context or speaker. In order to tackle this challenge, our paper proposes a novel loss called Focal Weighted Loss (FWL) with adversarial training and the compact language model MobileBERT. Our proposed loss function handles the problem of imbalanced emotion classification through Focal Weighted Loss and adversarial training and does not require large batch sizes or more computational resources. Our approach has been employed on four text emotion recognition benchmark datasets, MELD, EmoryNLP, DailyDialog and IEMOCAP demonstrating competitive performance. Extensive experiments on these benchmark datasets validate the effectiveness of our proposed FWL with adversarial training. This enables more human-like interactions on digital platforms. Our approach shows its potential to deliver competitive performance under limited resource constraints, comparable to large language models.

Article Details

Journal PLoS ONE
Volume / Issue Vol. 20, Issue 1
Published January 24, 2025
Pages e0312867
ISSN 1932-6203
Publisher Public Library of Science

Journal Info

PLoS ONE

Public Library of Science

ISSN: 1932-6203 Open Access Health Sciences

Authors (7)

M

Muhammad Hussain

C

Caikou Chen

S

Sami S. Albouq

K

Khlood Shinan

F

Fatmah Alanazi

M

Muhammad Waseem Iqbal

M

M. Usman Ashraf