Skip to yearly menu bar Skip to main content


Poster
in
Workshop: 3rd Workshop on Interpretable Machine Learning in Healthcare (IMLH)

Semi-supervised Ordinal Regression via Cumulative Link Models for Predicting In-Hospital Length-of-Stay

Alexander Lobo · Preetish Rath · Michael Hughes

Keywords: [ medical time-series data ] [ length-of-stay prediction ] [ cumulative link models ] [ ordinal regression ] [ Semi-supervised learning ]


Abstract:

Length-of-stay prediction has been widely studied as a classification task (e.g. will patients stay 0-3 days, 3-7 days, or more than 7 days?). Yet previous approaches neglect the natural ordering of these classes: standard multi-class classification treats classes as unordered, while methods that build separate binary classifiers for each class struggle to enforce coherent probabilistic predictions across classes. Instead, we suggest that cumulative link models, an ordinal approach long-known in statistics, is a naturally coherent approach well-suited to length-of-stay classification. We view ordinal regression as an output layer that can be integrated into any training pipeline based on automatic differentiation. We show this output layer yields improved predictions over binary classifier alternatives when paired with either neural net or hidden Markov model representations of patient vital sign history, all while requiring fewer parameters. Further experiments show promise in a semi-supervised setting, where only some patients have observed outcomes.

Chat is not available.