Facial Beauty Analysis Using Distribution Prediction and CNN Ensembles

Ahmed Aman Ibrahim, Noah Hassan Ugail, Tazkia Hoodh Jayatileke, Millie Hope Saffery, Hassan Ugail

Research output: Chapter or section in a book/report/conference proceedingChapter in a published conference proceeding

1 Citation (SciVal)

Abstract

Facial Beauty Prediction (FBP) is a computer vision task of quantifying the beauty of a face. Several solutions to this problem have benefitted immensely from the recent developments in deep learning. However, the majority of current methods train machine learning models to purely predict mean beauty scores, treating FBP solely as a regression task. In addition, deep learning based FBP approaches so far use transfer learning from models trained on general classification tasks such as ImageNet. We propose fine-tuning an ensemble of convolutional neural network (CNN) models originally trained on face verification tasks using a variety of loss functions such as Earth Mover's Distance (EMD) based loss. With this approach, our method can predict the entire beauty score distribution rather than just the mean, and the predicted mean scores have a higher Pearson Correlation (PC) compared to the ground truth scores. This method achieves state of the art results on the MEBeauty dataset in terms of mean absolute error, root mean squared error and PC between the predicted and the ground truth mean scores.

Original languageEnglish
Title of host publication2023 - 15th International Conference on Software, Knowledge, Information Management and Applications, SKIMA 2023
Place of PublicationU. S. A.
PublisherIEEE
Pages130-135
Number of pages6
ISBN (Electronic)9798350316551
ISBN (Print)9798350316568
DOIs
Publication statusPublished - 9 Dec 2023
Externally publishedYes
Event15th International Conference on Software, Knowledge, Information Management and Applications, SKIMA 2023 - Kuala Lumpur, Malaysia
Duration: 8 Dec 20239 Dec 2023

Publication series

NameInternational Conference on Software, Knowledge Information, Industrial Management and Applications, SKIMA
ISSN (Print)2373-082X
ISSN (Electronic)2573-3214

Conference

Conference15th International Conference on Software, Knowledge, Information Management and Applications, SKIMA 2023
Country/TerritoryMalaysia
CityKuala Lumpur
Period8/12/239/12/23

Keywords

  • Convolutional Neural Networks
  • Discrete Probability Distribution
  • Earth Mover's Distance
  • Ensemble Learning
  • Facial Beauty Prediction

ASJC Scopus subject areas

  • Computer Science Applications
  • Software
  • Information Systems
  • Information Systems and Management

Fingerprint

Dive into the research topics of 'Facial Beauty Analysis Using Distribution Prediction and CNN Ensembles'. Together they form a unique fingerprint.

Cite this