about
Active Inference as Context Acquisition for AI Agents (arxiv.org)
3 points by Anon84 42 days ago | hide | past | pdf | discuss on HN

In plain words: It keeps a running guess at what's missing and picks the next question, fetch, or prompt trial that buys the most information per token, or acts and stops. Testing covers question-asking with 25 to 300 answers, plus clarification and prompt tuning under token budgets.

Abstract

Interactive AI agents must acquire the right context as efficiently as possible. When a user omits a constraint, preference, file, or task variable, an agent can proceed with a default assumption or spend tokens on a clarifying question, retrieval call, tool call, or prompt trial. We formulate this tradeoff as active inference for context acquisition. An inner inference step updates beliefs over a latent task state, and an outer decision selects the next context action, task action, or stop action to minimize expected free energy under cost. In deterministic settings, the epistemic term reduces to expected information gain, optionally normalized by token cost. We instantiate the framework in Optimal Question Asking (OQA), with exact posteriors and a dynamic programming oracle, and benchmark frontier language models on binary and multiway categorical tasks from 25 to 300 candidates. We also study clarification before generation and automated prompt optimization under token budgets. The formulation is model-agnostic and views active inference as a design principle for the context-acquisition layer of AI agents.

Sanchayan Dutta, Sai Niranjan Ramachandran, Suvrit Sra
arXiv:2608.19202 · cs.AI, cs.CL, cs.LG · submitted Jun 8, 2026
abstract · pdf · html

add comment on HN