AFABench: A Generic Framework for Benchmarking Active Feature Acquisition

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Schütz, Valter, Wu, Han, Rezvan, Reza, Aronsson, Linus, Chehreghani, Morteza Haghir
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910029559365632
author Schütz, Valter
Wu, Han
Rezvan, Reza
Aronsson, Linus
Chehreghani, Morteza Haghir
author_facet Schütz, Valter
Wu, Han
Rezvan, Reza
Aronsson, Linus
Chehreghani, Morteza Haghir
contents In many real-world scenarios, acquiring all features of a data instance can be expensive or impractical due to monetary cost, latency, or privacy concerns. Active Feature Acquisition (AFA) addresses this challenge by dynamically selecting a subset of informative features for each data instance, trading predictive performance against acquisition cost. While numerous methods have been proposed for AFA, ranging from myopic information-theoretic strategies to non-myopic reinforcement learning approaches, fair and systematic evaluation of these methods has been hindered by a lack of standardized benchmarks. In this paper, we introduce AFABench, the first benchmark framework for AFA. Our benchmark includes a diverse set of synthetic and real-world datasets, supports a wide range of acquisition policies, and provides a modular design that enables easy integration of new methods and tasks. We implement and evaluate representative algorithms from all major categories, including static, myopic, and reinforcement learning-based approaches. To test the lookahead capabilities of AFA policies, we introduce a novel synthetic dataset, CUBE-NM, designed to expose the limitations of myopic selection. Our results highlight key trade-offs between different AFA strategies and provide actionable insights for future research. The benchmark code is available at: https://github.com/Linusaronsson/AFA-Benchmark.
format Preprint
id arxiv_https___arxiv_org_abs_2508_14734
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle AFABench: A Generic Framework for Benchmarking Active Feature Acquisition
Schütz, Valter
Wu, Han
Rezvan, Reza
Aronsson, Linus
Chehreghani, Morteza Haghir
Machine Learning
Artificial Intelligence
In many real-world scenarios, acquiring all features of a data instance can be expensive or impractical due to monetary cost, latency, or privacy concerns. Active Feature Acquisition (AFA) addresses this challenge by dynamically selecting a subset of informative features for each data instance, trading predictive performance against acquisition cost. While numerous methods have been proposed for AFA, ranging from myopic information-theoretic strategies to non-myopic reinforcement learning approaches, fair and systematic evaluation of these methods has been hindered by a lack of standardized benchmarks. In this paper, we introduce AFABench, the first benchmark framework for AFA. Our benchmark includes a diverse set of synthetic and real-world datasets, supports a wide range of acquisition policies, and provides a modular design that enables easy integration of new methods and tasks. We implement and evaluate representative algorithms from all major categories, including static, myopic, and reinforcement learning-based approaches. To test the lookahead capabilities of AFA policies, we introduce a novel synthetic dataset, CUBE-NM, designed to expose the limitations of myopic selection. Our results highlight key trade-offs between different AFA strategies and provide actionable insights for future research. The benchmark code is available at: https://github.com/Linusaronsson/AFA-Benchmark.
title AFABench: A Generic Framework for Benchmarking Active Feature Acquisition
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2508.14734