DeepAI AI Chat
Log In Sign Up

Explainable Multi-class Classification of Medical Data

by   YuanZheng Hu, et al.

Machine Learning applications have brought new insights into a secondary analysis of medical data. Machine Learning helps to develop new drugs, define populations susceptible to certain illnesses, identify predictors of many common diseases. At the same time, Machine Learning results depend on convolution of many factors, including feature selection, class (im)balance, algorithm preference, and performance metrics. In this paper, we present explainable multi-class classification of a large medical data set. We in details discuss knowledge-based feature engineering, data set balancing, best model selection, and parameter tuning. Six algorithms are used in this study: Support Vector Machine (SVM), Naïve Bayes, Gradient Boosting, Decision Trees, Random Forest, and Logistic Regression. Our empirical evaluation is done on the UCI Diabetes 130-US hospitals for years 1999-2008 dataset, with the task to classify patient hospital re-admission stay into three classes: 0 days, <30 days, or > 30 days. Our results show that using 23 medication features in learning experiments improves Recall of five out of the six applied learning algorithms. This is a new result that expands the previous studies conducted on the same data. Gradient Boosting and Random Forest outperformed other algorithms in terms of the three-class classification Accuracy.


page 6

page 18

page 19

page 20

page 21


Explainable Multi-class Classification of the CAMH COVID-19 Mental Health Data

Application of Machine Learning algorithms to the medical domain is an e...

Gradient-based Quadratic Multiform Separation

Classification as a supervised learning concept is an important content ...

hi-RF: Incremental Learning Random Forest for large-scale multi-class Data Classification

In recent years, dynamically growing data and incrementally growing numb...

The Effectiveness of Supervised Machine Learning Algorithms in Predicting Software Refactoring

Refactoring is the process of changing the internal structure of softwar...

Machine Learning Evaluation of the Echo-Chamber Effect in Medical Forums

We propose the Echo-Chamber Effect assessment of an online forum. Sentim...