Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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

About

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.

Wenzhi Cao, Vahid Mirjalili, Sebastian Raschka• 2019

Related benchmarks

TaskDatasetResultRank
Age EstimationMorph (test)
MAE (Years)6.13
52
Age EstimationMORPH II (test)
MAE2.62
51
Age EstimationUTKFace (test)
MAE5.39
36
Age EstimationCACD (test)
MAE5.24
25
Age EstimationFG-NET (test)
MAE6.12
24
Age EstimationUTKFace
MAE5.47
13
Age EstimationMORPH II (Random Split (RS) protocol)
MAE2.64
13
Age EstimationAFAD (test)
MAE3.42
12
Age EstimationKANFace (test)
MAE7.88
10
Ordinal RegressionMessidor (test)
Accuracy62.2
9
Showing 10 of 16 rows

Other info

Code

Follow for update