Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Model-Agnostic Dynamic Feature Selection with Uncertainty Quantification

About

Dynamic feature selection (DFS) addresses budget constraints in decision-making by sequentially acquiring features for each instance, making it appealing for resource-limited scenarios. However, existing DFS methods require models specifically designed for the sequential acquisition setting, limiting compatibility with models already deployed in practice. Furthermore, they provide limited uncertainty quantification, undermining trust in high-stakes decisions. In this work, we show that DFS introduces new uncertainty sources compared to the static setting. We formalise how model adaptation to feature subsets induces epistemic uncertainty, how standard imputation strategies bias aleatoric uncertainty estimation, and why predictive confidence fails to discriminate between good and bad selection policies. We also propose a model-agnostic DFS framework compatible with pre-trained classifiers, including interpretable-by-design models, through efficient subset reparametrization strategies. Empirical evaluation on tabular and image datasets demonstrates competitive accuracy against state-of-the-art greedy and reinforcement learning-based DFS methods with both neural and rule-based classifiers. We further show that the identified uncertainty sources persist across most existing approaches, highlighting the need for uncertainty-aware DFS.

Javier Fumanal-Idocin, Raquel Fernandez-Peralta, Javier Andreu-Perez• 2025

Related benchmarks

TaskDatasetResultRank
Image ClassificationMNIST (test)
Accuracy96.54
61
ClassificationWINE (test)
Accuracy98.15
29
Tabular ClassificationCirrhosis (test)
Accuracy76.32
14
Tabular ClassificationYeast (test)
Accuracy57.32
14
Tabular Classification with Feature SelectionWine
Accuracy90.28
14
Tabular Classification with Feature SelectionYeast
Accuracy0.3902
14
Tabular Classification with Feature SelectionHeart
Accuracy75.08
14
Tabular ClassificationDiabetes (test)
Accuracy61.36
14
Tabular ClassificationHeart (test)
Accuracy82.88
14
Tabular Classification with Feature SelectionCirrhosis
Accuracy70.12
14
Showing 10 of 12 rows

Other info

Follow for update