English

GRASP: group-Shapley feature selection for patients

Machine Learning 2026-05-01 v2 Artificial Intelligence

Abstract

Feature selection remains a major challenge in medical prediction, where existing approaches such as LASSO often lack robustness and interpretability. We introduce GRASP, a novel framework that couples Shapley value driven attribution with group L21L_{21} regularization to extract compact and non-redundant feature sets. GRASP first distills group level importance scores from a pretrained tree model via SHAP, then enforces structured sparsity through group L21L_{21} regularized logistic regression, yielding stable and interpretable selections. Extensive comparisons with LASSO, SHAP, and deep learning based methods show that GRASP consistently delivers comparable or superior predictive accuracy, while identifying fewer, less redundant, and more stable features.

Keywords

Cite

@article{arxiv.2602.11084,
  title  = {GRASP: group-Shapley feature selection for patients},
  author = {Yuheng Luo and Shuyan Li and Zhong Cao},
  journal= {arXiv preprint arXiv:2602.11084},
  year   = {2026}
}

Comments

5 pages, 4 figures, 2 tables. Accepted at IEEE ICASSP 2026

R2 v1 2026-07-01T10:32:15.935Z