3 个月前

基于分布预测与CNN集成的面部美感分析

基于分布预测与CNN集成的面部美感分析

摘要

面部美感预测(Facial Beauty Prediction, FBP)是计算机视觉领域中一项量化人脸美感的任务。近年来,深度学习的快速发展为该问题的解决带来了显著进展。然而,目前大多数方法仅训练机器学习模型以预测平均美感得分,将FBP视为单一的回归任务。此外,现有的基于深度学习的FBP方法普遍采用在通用分类任务(如ImageNet)上预训练的模型进行迁移学习。本文提出一种新方法:对原本在人脸验证任务上训练的卷积神经网络(CNN)集成模型进行微调,并采用多种损失函数(如基于地球移动距离,Earth Mover's Distance, EMD的损失函数)进行优化。该方法不仅能够预测完整的美感得分分布,而不仅仅是均值,而且预测的均值得分与真实值之间的皮尔逊相关系数(Pearson Correlation, PC)显著提升。在MEBeauty数据集上,该方法在平均绝对误差(Mean Absolute Error, MAE)、均方根误差(Root Mean Squared Error, RMSE)以及预测均值与真实均值之间的皮尔逊相关系数(PC)三项指标上均达到当前最优水平,实现了该任务的最新性能。

基准测试

基准方法指标
facial-beauty-prediction-on-mebeautyCNN + Earth Mover Distance
MAE: 0.616
Pearson Corr: 0.795
Root mean square error (RMSE): 0.794

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于分布预测与CNN集成的面部美感分析 | 论文 | HyperAI超神经