Abstract:BACKGROUND In emergency departments (ED), timely rescue is very important as patients’ conditions usually deteriorate rapidly. Early diagnosis can increase patients’ chances of survival. Early diagnosis can be improved by predictive models based on machine learning using Electronic Medical Record (EMR) data. However, ED data are usually imbalanced, having missing values and sparse features. These quality issues make it challenging to build early identification models for diseases in ED. OBJECTIVE The objective of this study is to propose a systematic approach to deal with missing, imbalanced and sparse feature problems of ED data. METHODS We used random forest and K-means algorithms to interpolate missing values and under-sample data. Regarding sparse features, we used principal component analysis to reduce dimensions. For continuous and discrete variables, the decision coefficient R2 and Kappa coefficient are used to evaluate the performance respectively. The area under the receiver operating characteristic curve (AUC) and the area under the precision-recall curve (AUPRC) are used to estimate the model performance. To further evaluate the proposed approach, we carried out a case study using an ED dataset extracted from Hainan Hospital of Chinese PLA General Hospital. A logistic regression model for patient condition worsening prediction was built out of the data processed by the proposed approach. RESULTS A total of 1085 patients with rescue record and 17959 patients without rescue record were collected, which were significantly imbalanced. 275, 402 and 891 variables are extracted from laboratory tests, medications and diagnosis, respectively. After data preprocessing, the median R2 of random forest interpolation for continuous variables is 0.623 (IQR: 0.647), and the median of Kappa coefficient for discrete variable interpolation is 0.444 (IQR: 0.285). The logistic regression model constructed using the initial diagnostic data has poor performance and variable separation, which is reflected in the abnormally high OR values of the two variables of cardiac arrest and respiratory arrest (27857.4 and 9341.6) and an abnormal confidence interval. Using the processed data, the recall of the model reaches 0.77, F1-SCORE is 0.74, and AUC is 0.64. CONCLUSIONS We proposed a machine learning method to deal with data quality issues such as missing data, data imbalance, and sparse features in emergency data, so as to improve data availability. A preliminary case study indicate the results produced by the proposed method can be used for building prediction model for emergency patients.

Feature Selection and Optimization of Random Forest Modeling

Research of Medical High-Dimensional Imbalanced Data Classification Ensemble Feature Selection Algorithm with Random Forest

Cost-sensitive feature selection using random forest: Selecting low-cost subsets of informative features

Research on Optimization of Random Forest Algorithm Based on Spark

Empirical Evaluation of the Performance of Feature Selection Approaches on Random Forest

CURE-SMOTE algorithm and hybrid algorithm for feature selection and parameter optimization based on random forests

Using random forest for reliable classification and cost-sensitive learning for medical diagnosis

Feature Selection Methods for Cost-Constrained Classification in Random Forests

Recursive Random Forests Enable Better Predictive Performance and Model Interpretation than Variable Selection by LASSO

Application of Random Forest Algorithm on Feature Subset Selection and Classification and Regression

Nonparametric feature selection by random forests and deep neural networks

Feature Selection and Feature Stability Measurement Method for High-Dimensional Small Sample Data Based on Big Data Technology

Decision Variants for the Automatic Determination of Optimal Feature Subset in RF-RFE

Feature Selection using Random Forest Classifier for Predicting Prostate Cancer

A Feature Selection Method Based on Feature Grouping and Genetic Algorithm

Acute coronary syndrome risk prediction based on gradient boosted tree feature selection and recursive feature elimination: A dataset-specific modeling study

Dealing with the Missing, Imbalanced and Sparse Features Problems in Emergency Data Using Random Forest, K-means and PCA Respectively (Preprint)

An Improved Feature Selection Method Based on Random Forest Algorithm for Wind Turbine Condition Monitoring

RANDOM FOREST AND SUPPORT VECTOR MACHINE ON FEATURES SELECTION FOR REGRESSION ANALYSIS

A comparison of random forest variable selection methods for classification prediction modeling

Random Forest Variable Importance-based Selection Algorithm in Class Imbalance Problem