Unimodal probability distributions for deep ordinal classification
About
Probability distributions produced by the cross-entropy loss for ordinal classification problems can possess undesired properties. We propose a straightforward technique to constrain discrete ordinal probability distributions to be unimodal via the use of the Poisson and binomial probability distributions. We evaluate this approach in the context of deep learning on two large ordinal image datasets, obtaining promising results.
Christopher Beckham, Christopher Pal• 2017
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Age Estimation | AFAD-Lite (test) | MAE3.56 | 7 | |
| Classification | ICIAR (test) | Mean Absolute Error0.6 | 7 | |
| Classification | HCI (test) | MAE0.71 | 7 | |
| Ordinal Regression | HCI | MAE0.62 | 5 | |
| Ordinal Regression | FG-NET | MAE0.46 | 5 | |
| Ordinal Regression | Adience | MAE0.53 | 5 | |
| Ordinal Regression | Retina MNIST | MAE0.78 | 5 | |
| Ordinal Regression | AAF | MAE0.44 | 5 | |
| Ordinal Regression | AFAD-LITE | MAE0.51 | 5 | |
| Ordinal Regression | EVA | MAE0.63 | 5 |
Showing 10 of 11 rows