cs.LGJun 23, 2026

Efficient Adaptive Data Acquisition via Pretrained Belief Representations

Authors: Daolang HuangZhuoyue HuangConor HassanLuigi AcerbiSamuel KaskiTom Rainforth

Abstract

Learning effective policies for adaptive data acquisition remains challenging: posterior-based methods rely on surrogate models and posterior approximations that can be misspecified or biased, while direct policy-learning methods map from historical observations and fail to exploit available model representations, making learning harder. We introduce policy learning with belief representations (POLAR), based on the insight that optimal data acquisition depends on the observation history only through a sufficient belief state. Specifically, POLAR decouples representation learning from policy learning by leveraging pretrained predictive foundation models as belief-state encoders, training a policy head on top of their representations. This yields a simple, unified amortised policy learning framework for Bayesian experimental design, Bayesian optimisation, and active learning, differing only in the task-specific utility used to train the policy. Empirically, we find that POLAR outperforms state-of-the-art amortised methods across diverse tasks while requiring far fewer training samples, demonstrating a significant step in the scalability and efficiency of amortised data acquisition.

Explore similar work

CardsList
  1. Non-Myopic Active Feature Acquisition via Pathwise Policy Gradients

    May 6, 2026Linus Aronsson, Morteza Haghir ChehreghaniActive Learning

  2. Bayesian Experimental Design via Score Matching

    Jul 9, 2026Angus Phillips, Gavin Kerrigan, Tom RainforthBayesian Experimental DesignBayesian