Current browse context:
stat.ML
Change to browse by:
References & Citations
Statistics > Machine Learning
Title: A simple squared-error reformulation for ordinal classification
(Submitted on 2 Dec 2016 (v1), last revised 9 Jan 2017 (this version, v2))
Abstract: In this paper, we explore ordinal classification (in the context of deep neural networks) through a simple modification of the squared error loss which not only allows it to not only be sensitive to class ordering, but also allows the possibility of having a discrete probability distribution over the classes. Our formulation is based on the use of a softmax hidden layer, which has received relatively little attention in the literature. We empirically evaluate its performance on the Kaggle diabetic retinopathy dataset, an ordinal and high-resolution dataset and show that it outperforms all of the baselines employed.
Submission history
From: Christopher Beckham [view email][v1] Fri, 2 Dec 2016 17:57:04 GMT (366kb,D)
[v2] Mon, 9 Jan 2017 16:04:38 GMT (442kb,D)
Link back to: arXiv, form interface, contact.