62

GRASP: group-Shapley feature selection for patients

Yuheng Luo
Shuyan Li
Zhong Cao
Main:4 Pages
4 Figures
Bibliography:1 Pages
2 Tables
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.

View on arXiv
Comments on this paper