ICU Mortality Prediction: A Classification Algorithm for Imbalanced Datasets

Authors

  • Sakyajit Bhattacharya Xerox Research Centre India
  • Vaibhav Rajan Xerox Research Centre India
  • Harsh Shrivastava Xerox Research Centre India

DOI:

https://doi.org/10.1609/aaai.v31i1.10721

Keywords:

Mortality Prediction, Supervised Classification, Class Imbalance

Abstract

Determining mortality risk is important for critical decisions in Intensive Care Units (ICU). The need for machine learning models that provide accurate patient-specific prediction of mortality is well recognized. We present a new algorithm for ICU mortality prediction that is designed to address the problem of imbalance, which occurs, in the context of binary classification, when one of the two classes is significantly under--represented in the data. We take a fundamentally new approach in exploiting the class imbalance through a feature transformation such that the transformed features are easier to classify. Hypothesis testing is used for classification with a test statistic that follows the distribution of the difference of two chi-squared random variables, for which there are no analytic expressions and we derive an accurate approximation. Experiments on a benchmark dataset of 4000 ICU patients show that our algorithm surpasses the best competing methods for mortality prediction.

Downloads

Published

2017-02-12

How to Cite

Bhattacharya, S., Rajan, V., & Shrivastava, H. (2017). ICU Mortality Prediction: A Classification Algorithm for Imbalanced Datasets. Proceedings of the AAAI Conference on Artificial Intelligence, 31(1). https://doi.org/10.1609/aaai.v31i1.10721

Issue

Section

Main Track: Machine Learning Applications