We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

stat.ML

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Statistics > Machine Learning

Title: Deep and interpretable regression models for ordinal outcomes

Abstract: Outcomes with a natural order commonly occur in prediction tasks and often the available input data are a mixture of complex data like images and tabular predictors. Deep Learning (DL) models are state-of-the-art for image classification tasks but frequently treat ordinal outcomes as unordered and lack interpretability. In contrast, classical ordinal regression models consider the outcome's order and yield interpretable predictor effects but are limited to tabular data. We present ordinal neural network transformation models (ONTRAMs), which unite DL with classical ordinal regression approaches. ONTRAMs are a special case of transformation models and trade off flexibility and interpretability by additively decomposing the transformation function into terms for image and tabular data using jointly trained neural networks. The performance of the most flexible ONTRAM is by definition equivalent to a standard multi-class DL model trained with cross-entropy while being faster in training when facing ordinal outcomes. Lastly, we discuss how to interpret model components for both tabular and image data on two publicly available datasets.
Comments: 41 pages (incl. appendix, figures and literature), 11 figures in main text, 4 figures in appendix
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG)
Cite as: arXiv:2010.08376 [stat.ML]
  (or arXiv:2010.08376v4 [stat.ML] for this version)

Submission history

From: Beate Sick [view email]
[v1] Fri, 16 Oct 2020 13:27:34 GMT (1410kb,D)
[v2] Mon, 26 Oct 2020 10:32:08 GMT (1488kb,D)
[v3] Fri, 11 Dec 2020 13:50:05 GMT (1637kb,D)
[v4] Tue, 20 Apr 2021 15:09:52 GMT (1628kb,D)

Link back to: arXiv, form interface, contact.