4 个月前

用于年龄估计的神经网络秩一致序回归

用于年龄估计的神经网络秩一致序回归

摘要

在许多现实世界的预测任务中,类别标签包含了标签之间的相对顺序信息,而这些信息通常无法通过常用的损失函数(如多分类交叉熵)捕捉到。近期,深度学习社区采用了有序回归框架来考虑这种顺序信息。通过将有序目标转换为二分类子任务,神经网络被赋予了有序回归的能力。然而,这种方法在不同的二分类器之间存在不一致性问题。为了解决这些不一致性,我们提出了具有强大理论保证的秩单调性和一致置信度分数的COnsistent RAnk Logits (CORAL) 框架。此外,所提出的方法与架构无关,可以扩展任意最先进的深度神经网络分类器以执行有序回归任务。对一系列用于年龄预测的人脸图像数据集进行的实证评估表明,与参考有序回归网络相比,所提出的秩一致方法显著降低了预测误差。

代码仓库

mshehrozsajjad/Age-Classification
pytorch
GitHub 中提及
axeber01/dold
GitHub 中提及
Raschka-research-group/coral-cnn
官方
pytorch
GitHub 中提及
ck37/coral-ordinal
tf
GitHub 中提及

基准测试

基准方法指标
age-estimation-on-afadCORAL
MAE: 3.48
age-estimation-on-cacdCORAL
MAE: 5.35
age-estimation-on-morph-album2CORAL
MAE: 2.59
age-estimation-on-utkfaceCORAL
MAE: 5.39

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
用于年龄估计的神经网络秩一致序回归 | 论文 | HyperAI超神经