Zero-Shot Active Feature Acquisition via LLM-Elicitation
For clinical decision-making where labeled data is scarce, this work enables zero-shot AFA by leveraging LLM knowledge without requiring sequential planning from the LLM.
The paper tackles zero-shot active feature acquisition (AFA) without labeled data by using LLMs to provide sufficient statistics for a Markov random field, resolving gauge ambiguity via maximum-entropy closure. On an IBD patient cohort, the method outperforms LLM baselines and existing methods, especially on the hardest patients.
Active feature acquisition (AFA) sequentially selects which features to observe to reach a classification or ranking decision. Its central limitation is reliance on large amount of labeled data to fit probabilistic models guiding acquisition. Large language models (LLMs) supply unsupervised domain knowledge, but are poor sequential planners. Asking one to both know and decide conflates capabilities best kept separate. Here, we develop a framework for zero-shot AFA through disciplined elicitation: asking the LLM only for what it can be trusted to return, the unary deviations and pairwise co-variations that are the sufficient statistics of a Markov random field (MRF). We apply our framework to two settings: binary classification and top-$k$ identification. In practice, the LLM reliably returns only discriminative statistics, what distinguishes the classes rather than each class in isolation, which precludes classical AFA. We apply a maximum-entropy closure that resolves this gauge ambiguity. We evaluate on a cohort of Inflammatory Bowel Disease (IBD) patients, an active clinical setting where diagnostic ambiguity and patient heterogeneity obstruct stable treatment strategies. Our framework outperforms the LLM both on real labels and on its own extracted beliefs. Where it matters most, on the hardest patients, our top-$k$ acquisition policy markedly outperforms all existing methods.