Increasing efficiency of SVMp+ for handling missing values in healthcare prediction

PLOS Digit Health. 2023 Jun 29;2(6):e0000281. doi: 10.1371/journal.pdig.0000281. eCollection 2023 Jun.

Abstract

Missing data presents a challenge for machine learning applications specifically when utilizing electronic health records to develop clinical decision support systems. The lack of these values is due in part to the complex nature of clinical data in which the content is personalized to each patient. Several methods have been developed to handle this issue, such as imputation or complete case analysis, but their limitations restrict the solidity of findings. However, recent studies have explored how using some features as fully available privileged information can increase model performance including in SVM. Building on this insight, we propose a computationally efficient kernel SVM-based framework (l2-SVMp+) that leverages partially available privileged information to guide model construction. Our experiments validated the superiority of l2-SVMp+ over common approaches for handling missingness and previous implementations of SVMp+ in both digit recognition, disease classification and patient readmission prediction tasks. The performance improves as the percentage of available privileged information increases. Our results showcase the capability of l2-SVMp+ to handle incomplete but important features in real-world medical applications, surpassing traditional SVMs that lack privileged information. Additionally, l2-SVMp+ achieves comparable or superior model performance compared to imputed privileged features.

Grants and funding

This material is based upon work supported by the National Science Foundation under Grant No. 1722801 and Grant No. 2014003. YZ received funding from National Science Foundation (NSF) Grant No. 2014003, ZG received funding from NSF Grant No. 1722801. EW and KN both received funding from NSF Grant No. 1722801 and Grant No. 2014003. The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.