MEAIMLFeb 26, 2025

AI-Powered Bayesian Inference

arXiv:2502.19231v34 citationsh-index: 3
Originality Incremental advance
AI Analysis

This work addresses the challenge of leveraging AI's inherent randomness for improved decision-making in domains requiring uncertainty quantification, though it appears incremental as it builds on existing non-parametric Bayesian methods.

The paper tackles the problem of integrating uncertain Generative AI predictions into decision-making by proposing a method to construct AI-driven prior distributions for Bayesian inference, enabling predictive inference and uncertainty quantification in a coherent probabilistic framework.

The advent of Generative Artificial Intelligence (GAI) has heralded an inflection point that changed how society thinks about knowledge acquisition. While GAI cannot be fully trusted for decision-making, it may still provide valuable information that can be integrated into a decision pipeline. Rather than seeing the lack of certitude and inherent randomness of GAI as a problem, we view it as an opportunity. Indeed, variable answers to given prompts can be leveraged to construct a prior distribution which reflects assuredness of AI predictions. This prior distribution may be combined with tailored datasets for a fully Bayesian analysis with an AI-driven prior. In this paper, we explore such a possibility within a non-parametric Bayesian framework. The basic idea consists of assigning a Dirichlet process prior distribution on the data-generating distribution with AI generative model as its baseline. Hyper-parameters of the prior can be tuned out-of-sample to assess the informativeness of the AI prior. Posterior simulation is achieved by computing a suitably randomized functional on an augmented data that consists of observed (labeled) data as well as fake data whose labels have been imputed using AI. This strategy can be parallelized and rapidly produces iid samples from the posterior by optimization as opposed to sampling from conditionals. Our method enables (predictive) inference and uncertainty quantification leveraging AI predictions in a coherent probabilistic manner.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes