Logistic Regression with Missing Covariates -- Parameter Estimation, Model Selection and Prediction within a Joint-Modeling Framework
About
Logistic regression is a common classification method in supervised learning. Surprisingly, there are very few solutions for performing logistic regression with missing values in the covariates. We suggest a complete approach based on a stochastic approximation version of the EM algorithm to do statistical inference with missing values including the estimation of the parameters and their variance, derivation of confidence intervals and a model selection procedure. We also tackle the problem of prediction for new observations (on a test set) with missing covariate data. The methodology is computationally efficient, and its good coverage and variable selection properties are demonstrated in a simulation study where we contrast its performances to other methods. For instance, the popular approach of multiple imputation by chained equations can lead to estimates that exhibit meaningfully greater biases than the proposed approach. We then illustrate the method on a dataset of severely traumatized patients from Paris hospitals to predict the occurrence of hemorrhagic shock, a leading cause of early preventable death in severe trauma cases. The aim is to consolidate the current red flag procedure, a binary alert identifying patients with a high risk of severe hemorrhage. The methodology is implemented in the R package misaem.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Parameter Estimation | Synthetic Data 30% MCAR mechanism | RMSE (Imp)0.732 | 7 | |
| Parameter Recovery | Synthetic 50% MCAR (test) | RMSE (Imputation)0.8055 | 7 | |
| Binary Classification | Synthetic 50% MCAR (test) | AUC73.05 | 7 | |
| Classification | Synthetic Data 30% MCAR mechanism | AUC81.6 | 7 | |
| Parameter Estimation | Synthetic Dataset 60% MNAR (test) | RMSE (Imputed)1.3 | 7 | |
| Classification | Synthetic Dataset 60% MNAR (test) | AUC62.5 | 7 | |
| Classification | Synthetic data 60% MNAR | AUC62.5 | 7 | |
| Logistic regression with missing covariates | Synthetic datasets 50% MCAR (test) | Training Time3.89e+3 | 7 | |
| Missing data estimation | Synthetic data 60% MNAR | Training Time4.21e+3 | 7 | |
| Missing data handling | Synthetic data 60% MNAR Ntrain = 2000, Ntest = 500, p = 5 (test) | Training Time (s)4.21e+3 | 7 |