Denotational validation of higher-order Bayesian inference
arXiv:1711.03219 · doi:10.1145/3158148
Abstract
We present a modular semantic account of Bayesian inference algorithms for probabilistic programming languages, as used in data science and machine learning. Sophisticated inference algorithms are often explained in terms of composition of smaller parts. However, neither their theoretical justification nor their implementation reflects this modularity. We show how to conceptualise and analyse such inference algorithms as manipulating intermediate representations of probabilistic programs using higher-order functions and inductive types, and their denotational semantics. Semantic accounts of continuous distributions use measurable spaces. However, our use of higher-order functions presents a substantial technical difficulty: it is impossible to define a measurable space structure over the collection of measurable functions between arbitrary measurable spaces that is compatible with standard operations on those functions, such as function application. We overcome this difficulty using quasi-Borel spaces, a recently proposed mathematical structure that supports both function spaces and continuous distributions. We define a class of semantic structures for representing probabilistic programs, and semantic validity criteria for transformations of these representations in terms of distribution preservation. We develop a collection of building blocks for composing representations. We use these building blocks to validate common inference algorithms such as Sequential Monte Carlo and Markov Chain Monte Carlo. To emphasize the connection between the semantic manipulation and its traditional measure theoretic origins, we use Kock's synthetic measure theory. We demonstrate its usefulness by proving a quasi-Borel counterpart to the Metropolis-Hastings-Green theorem.
References in corpus (8)
- Church: a language for generative models
- A Convenient Category for Higher-Order Probability Theory
- Deep Probabilistic Programming
- Automated Variational Inference in Probabilistic Programming
- Automatic Variational Inference in Stan
- Bayesian State-Space Modelling on High-Performance Hardware Using LibBi
- Commutative monads as a theory of distributions
- Eilenberg--Moore Monoids and Backtracking Monad Transformers
Cited by in corpus (28)
- A synthetic approach to Markov kernels, conditional independence and theorems on sufficient statistics
- A Domain Theory for Statistical Probabilistic Programming
- Formal verification of higher-order probabilistic programs
- ADEV: Sound Automatic Differentiation of Expected Values of Probabilistic Programs
- Densities of Almost Surely Terminating Probabilistic Programs are Differentiable Almost Everywhere
- Correctness of Sequential Monte Carlo Inference for Probabilistic Programming Languages
- Automatic Alignment in Higher-Order Probabilistic Programming Languages
- Probabilistic Programming with Programmable Variational Inference
- Learning Proposals for Probabilistic Programs with Inference Combinators
- Probability monads with submonads of deterministic states - Extended version
- Domains and Stochastic Processes
- Effectuses in Categorical Quantum Foundations
- Higher-Order Bayesian Networks, Exactly (Extended version)
- Amortized Rejection Sampling in Universal Probabilistic Programming
- Concrete categories and higher-order recursion (With applications including probability, differentiability, and full abstraction)
- Effect Handlers for Programmable Inference
- Suspension Analysis and Selective Continuation-Passing Style for Universal Probabilistic Programming Languages
- String Diagrams with Factorized Densities
- Functional probabilistic programming for scalable Bayesian modelling
- Sound Probabilistic Inference via Guide Types
- Automatic Alignment of Sequential Monte Carlo Inference in Higher-Order Probabilistic Programs
- Compositional Inference Metaprogramming with Convergence Guarantees
- Approximations in Probabilistic Programs
- Parallelizable Feynman-Kac Models for Universal Probabilistic Programming
- Towards Verified Stochastic Variational Inference for Probabilistic Programs
- GradInf: Gradient Estimation as Probabilistic Inference
- Higher-order probabilistic adversarial computations: Categorical semantics and program logics
- De Finetti's construction as a categorical limit