Implementation of an Attention Mechanism Model for Facial Beauty Assessment Using Transfer Learning

An important consideration in medical plastic surgery is the evaluation of the patient’s facial symmetry. However, because facial attractiveness is a slightly individualized cognitive experience, it is difficult to determine face attractiveness manually. This study aimed to train a model for assessi...

Full description

Bibliographic Details
Main Authors: Chao-Tung Yang, Yu-Chieh Wang, Lun-Jou Lo, Wen-Chung Chiang, Shih-Ku Kuang, Hsiu-Hsia Lin
Format: Article
Language:English
Published: MDPI AG 2023-03-01
Series:Diagnostics
Subjects:
Online Access:https://www.mdpi.com/2075-4418/13/7/1291
_version_ 1797608161695236096
author Chao-Tung Yang
Yu-Chieh Wang
Lun-Jou Lo
Wen-Chung Chiang
Shih-Ku Kuang
Hsiu-Hsia Lin
author_facet Chao-Tung Yang
Yu-Chieh Wang
Lun-Jou Lo
Wen-Chung Chiang
Shih-Ku Kuang
Hsiu-Hsia Lin
author_sort Chao-Tung Yang
collection DOAJ
description An important consideration in medical plastic surgery is the evaluation of the patient’s facial symmetry. However, because facial attractiveness is a slightly individualized cognitive experience, it is difficult to determine face attractiveness manually. This study aimed to train a model for assessing facial attractiveness using transfer learning while also using the fine-grained image model to separate similar images by first learning features. In this case, the system can make assessments based on the input of facial photos. Thus, doctors can quickly and objectively treat patients’ scoring and save time for scoring. The transfer learning was combined with CNN, Xception, and attention mechanism models for training, using the SCUT-FBP5500 dataset for pre-training and freezing the weights as the transfer learning model. Then, we trained the Chang Gung Memorial Hospital Taiwan dataset to train the model based on transfer learning. The evaluation uses the mean absolute error percentage (MAPE) value. The root mean square error (RMSE) value is used as the basis for experimental adjustment and the quantitative standard for the model’s predictive. The best model can obtain 0.50 in RMSE and 18.5% average error in MAPE. A web page was developed to infer the deep learning model to visualize the predictive model.
first_indexed 2024-03-11T05:39:35Z
format Article
id doaj.art-26f1ccbecb434421be37dcc8785f9c9c
institution Directory Open Access Journal
issn 2075-4418
language English
last_indexed 2024-03-11T05:39:35Z
publishDate 2023-03-01
publisher MDPI AG
record_format Article
series Diagnostics
spelling doaj.art-26f1ccbecb434421be37dcc8785f9c9c2023-11-17T16:30:32ZengMDPI AGDiagnostics2075-44182023-03-01137129110.3390/diagnostics13071291Implementation of an Attention Mechanism Model for Facial Beauty Assessment Using Transfer LearningChao-Tung Yang0Yu-Chieh Wang1Lun-Jou Lo2Wen-Chung Chiang3Shih-Ku Kuang4Hsiu-Hsia Lin5Department of Computer Science, Tunghai University, Taichung City 407224, TaiwanDepartment of Computer Science, Tunghai University, Taichung City 407224, TaiwanDepartment of Plastic and Reconstructive Surgery, Craniofacial Research Center, Chang Gung Memorial Hospital, No. 123, Dinghu Rd., Guishan Township, Taoyuan City 333423, TaiwanDepartment of Tourism and Recreation Management, Hsiuping University of Science and Technology, No. 11, Gongye Rd., Dali District, Taichung City 412406, TaiwanImaging Laboratory, Craniofacial Research Center, Chang Gung Memorial Hospital, No. 123, Dinghu Rd., Guishan Township, Taoyuan City 333423, TaiwanImaging Laboratory, Craniofacial Research Center, Chang Gung Memorial Hospital, No. 123, Dinghu Rd., Guishan Township, Taoyuan City 333423, TaiwanAn important consideration in medical plastic surgery is the evaluation of the patient’s facial symmetry. However, because facial attractiveness is a slightly individualized cognitive experience, it is difficult to determine face attractiveness manually. This study aimed to train a model for assessing facial attractiveness using transfer learning while also using the fine-grained image model to separate similar images by first learning features. In this case, the system can make assessments based on the input of facial photos. Thus, doctors can quickly and objectively treat patients’ scoring and save time for scoring. The transfer learning was combined with CNN, Xception, and attention mechanism models for training, using the SCUT-FBP5500 dataset for pre-training and freezing the weights as the transfer learning model. Then, we trained the Chang Gung Memorial Hospital Taiwan dataset to train the model based on transfer learning. The evaluation uses the mean absolute error percentage (MAPE) value. The root mean square error (RMSE) value is used as the basis for experimental adjustment and the quantitative standard for the model’s predictive. The best model can obtain 0.50 in RMSE and 18.5% average error in MAPE. A web page was developed to infer the deep learning model to visualize the predictive model.https://www.mdpi.com/2075-4418/13/7/1291transfer learningdeep learningfacial attractiveness predictionattention mechanismvisualization
spellingShingle Chao-Tung Yang
Yu-Chieh Wang
Lun-Jou Lo
Wen-Chung Chiang
Shih-Ku Kuang
Hsiu-Hsia Lin
Implementation of an Attention Mechanism Model for Facial Beauty Assessment Using Transfer Learning
Diagnostics
transfer learning
deep learning
facial attractiveness prediction
attention mechanism
visualization
title Implementation of an Attention Mechanism Model for Facial Beauty Assessment Using Transfer Learning
title_full Implementation of an Attention Mechanism Model for Facial Beauty Assessment Using Transfer Learning
title_fullStr Implementation of an Attention Mechanism Model for Facial Beauty Assessment Using Transfer Learning
title_full_unstemmed Implementation of an Attention Mechanism Model for Facial Beauty Assessment Using Transfer Learning
title_short Implementation of an Attention Mechanism Model for Facial Beauty Assessment Using Transfer Learning
title_sort implementation of an attention mechanism model for facial beauty assessment using transfer learning
topic transfer learning
deep learning
facial attractiveness prediction
attention mechanism
visualization
url https://www.mdpi.com/2075-4418/13/7/1291
work_keys_str_mv AT chaotungyang implementationofanattentionmechanismmodelforfacialbeautyassessmentusingtransferlearning
AT yuchiehwang implementationofanattentionmechanismmodelforfacialbeautyassessmentusingtransferlearning
AT lunjoulo implementationofanattentionmechanismmodelforfacialbeautyassessmentusingtransferlearning
AT wenchungchiang implementationofanattentionmechanismmodelforfacialbeautyassessmentusingtransferlearning
AT shihkukuang implementationofanattentionmechanismmodelforfacialbeautyassessmentusingtransferlearning
AT hsiuhsialin implementationofanattentionmechanismmodelforfacialbeautyassessmentusingtransferlearning