English

DeepSelective: Interpretable Prognosis Prediction via Feature Selection and Compression in EHR Data

Machine Learning 2025-06-23 v2 Artificial Intelligence

Abstract

The rapid accumulation of Electronic Health Records (EHRs) has transformed healthcare by providing valuable data that enhance clinical predictions and diagnoses. While conventional machine learning models have proven effective, they often lack robust representation learning and depend heavily on expert-crafted features. Although deep learning offers powerful solutions, it is often criticized for its lack of interpretability. To address these challenges, we propose DeepSelective, a novel end to end deep learning framework for predicting patient prognosis using EHR data, with a strong emphasis on enhancing model interpretability. DeepSelective combines data compression techniques with an innovative feature selection approach, integrating custom-designed modules that work together to improve both accuracy and interpretability. Our experiments demonstrate that DeepSelective not only enhances predictive accuracy but also significantly improves interpretability, making it a valuable tool for clinical decision-making. The source code is freely available at http://www.healthinformaticslab.org/supp/resources.php .

Keywords

Cite

@article{arxiv.2504.11264,
  title  = {DeepSelective: Interpretable Prognosis Prediction via Feature Selection and Compression in EHR Data},
  author = {Ruochi Zhang and Qian Yang and Xiaoyang Wang and Tian Wang and Qiong Zhou and Ziqi Deng and Kewei Li and Yueying Wang and Yusi Fan and Jiale Zhang and Lan Huang and Chang Liu and Fengfeng Zhou},
  journal= {arXiv preprint arXiv:2504.11264},
  year   = {2025}
}
R2 v1 2026-06-28T22:59:13.871Z