mlsanityData to 14 Oct 2026

Model-Agnostic Dynamic Feature Selection with Uncertainty Quantification

arXiv 2508.02566 ↗First posted 4 Aug 2025cs.LG cs.AIcited by 1

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 21 of its 38 references that are in the map. First posted on arXiv 4 Aug 2025.

Builds on

21 of 38 references.

References · 21

  1. 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,897
  2. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
  3. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
  4. 2019Can You Trust Your Model's Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shiftcited by 1,287
  5. 2020Concept Bottleneck Modelscited by 1,032
  6. 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 913
  7. 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 742
  8. 2022Fine-Tuning can Distort Pretrained Features and Underperform Out-of-Distributioncited by 739
  9. 2020Simple and Principled Uncertainty Estimation with Deterministic Deep Learning via Distance Awarenesscited by 389
  10. 2018Efficient parametrization of multi-domain deep neural networkscited by 168
  11. 2020Uncertainty Quantification and Deep Ensemblescited by 139
  12. 2022Quantifying Aleatoric and Epistemic Uncertainty in Machine Learning: Are Conditional Entropy and Mutual Information Appropriate Measures?cited by 125
9 more
  1. 2018EDDI: Efficient Dynamic Discovery of High-Value Information with Partial VAEcited by 87
  2. 2022A Deeper Look into Aleatoric and Epistemic Uncertainty Disentanglementcited by 87
  3. 2023Learning to Maximize Mutual Information for Dynamic Feature Selectioncited by 56
  4. 2017Classification with Costly Features using Deep Reinforcement Learningcited by 34
  5. 2023Variational Information Pursuit for Interpretable Predictionscited by 25
  6. 2023Estimating Conditional Mutual Information for Dynamic Feature Selectioncited by 21
  7. 2022The Missing Indicator Method: From Low to High Dimensionscited by 19
  8. 2025What makes an Ensemble (Un) Interpretable?cited by 9
  9. 2025Dynamic Feature Selection from Variable Feature Sets Using Features of Featurescited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026Causally-Guided Diffusion for Stable Feature Selection
  2. 2024Knockoff-Guided Feature Selection via A Single Pre-trained Reinforced Agent
  3. 2026Rethinking Representativeness and Diversity in Dynamic Data Selection
  4. 2025Know When to Abstain: Optimal Selective Classification with Likelihood Ratios
  5. 2025Probabilistic Consistency in Machine Learning and Its Connection to Uncertainty Quantification
  6. 2026Sampling-guided exploration of active feature selection policies
  7. 2026Dynamic Distribution-Aware Uncertainty Tracking in Vision-Language Representation Learning
  8. 2025A Multi-Component Reward Function with Policy Gradient for Automated Feature Selection with Dynamic Regularization and Bias Mitigation