TY - GEN
T1 - Facial Beauty Analysis Using Distribution Prediction and CNN Ensembles
AU - Ibrahim, Ahmed Aman
AU - Ugail, Noah Hassan
AU - Jayatileke, Tazkia Hoodh
AU - Saffery, Millie Hope
AU - Ugail, Hassan
PY - 2023/12/9
Y1 - 2023/12/9
N2 - 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.
AB - 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.
KW - Convolutional Neural Networks
KW - Discrete Probability Distribution
KW - Earth Mover's Distance
KW - Ensemble Learning
KW - Facial Beauty Prediction
UR - http://www.scopus.com/inward/record.url?scp=85184368468&partnerID=8YFLogxK
U2 - 10.1109/SKIMA59232.2023.10387332
DO - 10.1109/SKIMA59232.2023.10387332
M3 - Chapter in a published conference proceeding
AN - SCOPUS:85184368468
SN - 9798350316568
T3 - International Conference on Software, Knowledge Information, Industrial Management and Applications, SKIMA
SP - 130
EP - 135
BT - 2023 - 15th International Conference on Software, Knowledge, Information Management and Applications, SKIMA 2023
PB - IEEE
CY - U. S. A.
T2 - 15th International Conference on Software, Knowledge, Information Management and Applications, SKIMA 2023
Y2 - 8 December 2023 through 9 December 2023
ER -