FACIAL EMOTION RECOGNITION AND HANDLING DATA IMBALANCE IN MACHINE LEARNING

ICTACT Journal on Data Science and Machine Learning ( Volume: 5 , Issue: 1 )

Abstract

In this paper, we use deep learning to make an emotion recognition convolution neural network by customizing the EfficientNet model pretrained on the ImageNet dataset. We used the FER-2013 dataset available in the Wolfram repository. Seven classes of emotions are considered in the dataset: happy, sad, angry, surprise, disgust, fear, and neutral. We try out different methods to tackle class imbalance.

Authors

Diya Elizabeth1, Siria Sadeddin2
Indian Institute of Science Education and Research, Thiruvananthapuram, India1, Universidad Nacional de Colombia, Colombia2

Keywords

Undersampling, Weighted Sampling, Focal Loss

Published By
ICTACT
Published In
ICTACT Journal on Data Science and Machine Learning
( Volume: 5 , Issue: 1 )
Date of Publication
December 2023
Pages
547 - 549