Loading [MathJax]/extensions/TeX/mhchem.js
Transfer Learning and Fine-Tuning for Facial Expression Recognition with Class Balancing | IEEE Conference Publication | IEEE Xplore

Transfer Learning and Fine-Tuning for Facial Expression Recognition with Class Balancing


Abstract:

Facial expression recognition benefits from deep learning models because of their ability to automatically extract features. However, these models face three important ch...Show More

Abstract:

Facial expression recognition benefits from deep learning models because of their ability to automatically extract features. However, these models face three important challenges: first, training tends to take longer times than with traditional machine learning models. Second, obtaining and labeling enough data samples can become a heavy burden due to the feature complexity usually involved in these problems. Third, it is also common to face class imbalance challenges. In this paper, we address these challenges by implementing transfer learning, oversampling and fine tuning to a facial expression recognition use case. Combining transfer learning with the use of a GPU helped us complete the training for our models in just about one hour. Furthermore, we achieved a 65.75% accuracy with one of the models. We provide measurements for metrics that are helpful when dealing with imbalanced data to assess that the models are not biased like precision, recall, F1 score and loss.
Date of Conference: 12-16 August 2024
Date Added to IEEE Xplore: 08 October 2024
ISBN Information:

ISSN Information:

Conference Location: Buenos Aires, Argentina

References

References is not available for this document.