Author:
Bhattacharya Sakyajit,Rajan Vaibhav,Shrivastava Harsh
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.
Publisher
Association for the Advancement of Artificial Intelligence (AAAI)
Cited by
18 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献
1. Tools for Healthcare Data Lake Infrastructure Benchmarking;Information Systems Frontiers;2024-01-17
2. Neural Graph Revealers;Machine Learning for Multimodal Healthcare Data;2023-11-26
3. Neural Graphical Models;Lecture Notes in Computer Science;2023-11-19
4. Machine Learning for Benchmarking Critical Care Outcomes;Healthcare Informatics Research;2023-10-31
5. ExpertNet: A Deep Learning Approach to Combined Risk Modeling and Subtyping in Intensive Care Units;IEEE Journal of Biomedical and Health Informatics;2023-10