Classifier Pooling for Modern Ordinal Classification
Organizations: Department of Biomedical Engineering, Whiting School of Engineering, Johns Hopkins University, Baltimore, MD, USA. · Department of Radiology, School of Medicine, Johns Hopkins University, Baltimore, MD, USA. · 3*Department of Biostatistics, Bloomberg School of Public Health, Johns Hopkins University, Baltimore, MD, USA.
Abstract
Ordinal data is widely prevalent in clinical and other domains, yet there is a lack of both modern, machine-learning based methods and publicly available software to address it. In this paper, we present a model-agnostic method of ordinal classification, which can apply any non-ordinal classification method in an ordinal fashion. We also provide an open-source implementation of these algorithms, in the form of a Python package. We apply these models on multiple real-world datasets to show their performance across domains. We show that they often outperform non-ordinal classification methods, especially when the number of datapoints is relatively small or when there are many classes of outcomes. This work, including the developed software, facilitates the use of modern, more powerful machine learning algorithms to handle ordinal data.
Figures & tables
| Dataset 1: Fetal Health | Dataset 2: Car Quality | Dataset 3: Wine Quality | Dataset 4: RetinaMNIST radiomics | Dataset 5: MNIST radiomics (negative control) | ||||||||||||
| Method | Method Subtype | AW | PC | AUC OVR | AW | PC | AUC OVR | AW | PC | AUC OVR | AW | PC | AUC OVR | AW | PC | AUC OVR |
| Logistic Regression | OVR | 0.744 | 0.914 | 0.925 | 0.657 | 0.899 | 0.963 | 0.968 | 0.999 | 0.998 | 0.215 | 0.367 | 0.637 | 0.224 | 0.280 | 0.690 |
| Ordered logit model | 0.698 | 0.911 | 0.947 | 0.607 | 0.909 | 0.945 | 0.908 | 0.999 | 0.929 | |||||||
| Difference | 0.757 | 0.906 | 0.936 | 0.686 | 0.907 | 0.966 | 0.984 | 0.999 | 0.998 | 0.306 | 0.594 | 0.698 | 0.221 | 0.294 | 0.690 | |
| Tree-based | 0.763 | 0.912 | 0.935 | 0.662 | 0.913 | 0.964 | 0.984 | 0.999 | 0.997 | 0.320 | 0.604 | 0.708 | 0.191 | 0.325 | 0.648 | |
| Ordered probit model | 0.702 | 0.914 | 0.947 | 0.586 | 0.910 | 0.945 | 0.908 | 0.999 | 0.930 | 0.124 | 0.126 | 0.528 | ||||
| Method | Method Subtype | AW | PC | AUC OVR |
| Logistic Regression | OVR | -0.358 | -0.026 | 0.025 |
| Ordered logit model | -0.272 | 0.025 | 0.033 | |
| Difference | -0.322 | -0.010 | 0.027 | |
| Tree-based | -0.302 | 0.015 | 0.028 | |
| Ordered probit model | -0.311 | 0.012 | 0.031 | |
| Gaussian Naïve Bayes | Multiclass | -0.082 | 0.074 | 0.029 |
| Best split index calculation method | Even Split | Best Classifier (via 4-fold validation)** | Last index | Middle index | |||||||||
| Best split index | 4 | 0*** | 6 | 3 | |||||||||
| Method | Method Subtype | AW | PC | AUC OVR | AW | PC | AUC OVR | AW | PC | AUC OVR | AW | PC | AUC OVR |
| Logistic Regression | Difference | 0.310 | 0.767 | 0.824 | 0.310 | 0.767 | 0.824 | 0.310 | 0.767 | 0.824 | 0.310 | 0.767 | 0.824 |
| Tree-based | 0.268 | 0.764 | 0.813 | 0.307 | 0.746 | 0.815 | 0.269 | 0.782 | 0.816 | 0.285 | 0.747 | 0.814 | |
| Gaussian Naïve Bayes | Difference | 0.346 | 0.689 | 0.731 | 0.346 | 0.689 | 0.731 | 0.346 | 0.689 | 0.731 | 0.346 | 0.689 | 0.731 |
| Tree-based | 0.342 | 0.684 | 0.740 | 0.342 | 0.685 | 0.736 | 0.342 | 0.683 | 0.734 | 0.342 | 0.685 | 0.741 | |
Ordinal Regression
Ordinal regression predicts ordered categories while accounting for the fact that some mistakes are more severe than others.
See all 38 papers