arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1901.07884cs.LGstat.ML
收藏

神经网络的秩相容有序回归及其在年龄估计中的应用

Rank consistent ordinal regression for neural networks with application to age estimation

University of Wisconsin-Madison(威斯康星大学麦迪逊分校);Michigan State University(密歇根州立大学)

Wenzhi Cao, Vahid Mirjalili, Sebastian Raschka

摘要:

在许多现实世界的预测任务中,类标签包括关于标签之间的相对排序的信息,这不是由常用的损失函数(如多类别交叉熵)捕获的。最近,深度学习社区采用了有序回归框架来考虑这种排序信息。神经网络通过将有序目标转换为二进制分类子任务来配备有序回归能力。然而,这种方法遭受不同的二进制分类器之间的不一致。为了解决这些不一致性,我们提出了一致的RANK Logits(CORAL)框架,具有强有力的理论保证秩单调性和一致的置信分数。此外,所提出的方法是架构不可知的,可以扩展任意国家的最先进的深度神经网络分类器的顺序回归任务。对一系列用于年龄预测的面部图像数据集的经验评估表明,与参考有序回归网络相比,预测误差大幅降低。

英文摘要:

In many real-world prediction tasks, class labels include information about the relative ordering between labels, which is not captured by commonly-used loss functions such as multi-category cross-entropy. Recently, the deep learning community adopted ordinal regression frameworks to take such ordering information into account. Neural networks were equipped with ordinal regression capabilities by transforming ordinal targets into binary classification subtasks. However, this method suffers from inconsistencies among the different binary classifiers. To resolve these inconsistencies, we propose the COnsistent RAnk Logits (CORAL) framework with strong theoretical guarantees for rank-monotonicity and consistent confidence scores. Moreover, the proposed method is architecture-agnostic and can extend arbitrary state-of-the-art deep neural network classifiers for ordinal regression tasks. The empirical evaluation of the proposed rank-consistent method on a range of face-image datasets for age prediction shows a substantial reduction of the prediction error compared to the reference ordinal regression network.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑