Pith. sign in

REVIEW 5 cited by

JADAI: Jointly Amortizing Adaptive Design and Bayesian Inference

Not yet reviewed by Pith; the record is open.

This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.

SPECIMEN: schema-true, not a live event

T0 review · schema-true

One-sentence machine reading of the paper's core claim.

pith:XXXXXXXX · record.json · timestamp

arxiv 2512.22999 v2 pith:YJHUBZ4A submitted 2025-12-28 stat.ML cs.AIcs.LG

JADAI: Jointly Amortizing Adaptive Design and Bayesian Inference

classification stat.ML cs.AIcs.LG
keywords designinferenceadaptivejadaibayesianexperimentaljointlynetwork
verification ladder T0 review T1 audit T2 compute T3 formal T4 reserved
0 comments
read the original 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.

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.

Forward citations

Cited by 5 Pith papers

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score.

  1. Bayesian Experimental Design via Score Matching

    stat.ML 2026-07 conditional novelty 7.0

    SCOREBED isolates EIG double intractability in a policy-independent score-matching stage, then trains design policies with a singly intractable gradient estimator, enabling cheap multi-policy selection.

  2. Action-BED: Task-Driven Bayesian Experimental Design with Singly Intractable Objectives

    stat.ML 2026-06 unverdicted novelty 7.0

    Action-BED recasts BED as expected future loss on actions, producing singly intractable objectives jointly optimized for design and action policies via stochastic gradients without explicit posterior estimation.

  3. FairBED: A Bayesian Experimental Design Approach to Gathering Fairer Data

    stat.ML 2026-06 unverdicted novelty 7.0

    FairBED quantifies dataset fairness as uninformative about sensitive attributes and uses fairness-aware BED to gather data yielding better fairness-accuracy trade-offs than random or standard BED acquisition.

  4. Efficient Adaptive Data Acquisition via Pretrained Belief Representations

    cs.LG 2026-06 unverdicted novelty 6.0

    POLAR uses pretrained predictive foundation models as fixed belief-state encoders and trains only a lightweight policy head on top for amortised Bayesian experimental design, optimisation, and active learning.

  5. Overcoming Selection Bias in Statistical Studies With Amortized Bayesian Inference

    stat.ML 2026-04 unverdicted novelty 6.0

    Embedding selection mechanisms into generative simulators enables amortized Bayesian inference to produce debiased, well-calibrated posteriors without tractable likelihoods.