stat.MLDec 28, 2025

JADAI: Jointly Amortizing Adaptive Design and Bayesian Inference

Authors: Niels BracherLars KühmichelDesi R. IvanovaXavier IntesPaul-Christian BürknerStefan T. Radev

Organizations: 1Rensselaer Polytechnic Institute, USA · 2TU Dortmund University, Germany · University of Oxford, UK.

Abstract

We consider problems of parameter estimation where design variables can be actively optimized to maximize information gain. To this end, we introduce JADAI, a framework that jointly amortizes Bayesian adaptive design and inference by training a policy, a history network, and an inference network end-to-end. The networks minimize a generic loss that aggregates incremental reductions in posterior error along experimental sequences. Inference networks are instantiated with diffusion-based posterior estimators that can approximate high-dimensional and multimodal posteriors at every experimental step. Across standard adaptive design benchmarks, JADAI achieves superior or competitive performance.

Explore similar work

CardsList
  1. Deep Adaptive Bayesian Screening

    Jul 18, 2026Jade Lejeune Herman, Arno Strouwen, Johan A. K. Suykens +1Bayesian Experimental DesignBayesian