NeuTra-lizing Bad Geometry in Hamiltonian Monte Carlo Using Neural Transport
Not on the map. Why?
Abstract (arXiv)
Hamiltonian Monte Carlo is a powerful algorithm for sampling from difficult-to-normalize posterior distributions. However, when the geometry of the posterior is unfavorable, it may take many expensive evaluations of the target distribution and its gradient to converge and mix. We propose neural transport (NeuTra) HMC, a technique for learning to correct this sort of unfavorable geometry using inverse autoregressive flows (IAF), a powerful neural variational inference technique. The IAF is trained to minimize the KL divergence from an isotropic Gaussian to the warped posterior, and then HMC sampling is performed in the warped space. We evaluate NeuTra HMC on a variety of synthetic and real problems, and find that it significantly outperforms vanilla HMC both in time to reach the stationary distribution and asymptotic effective-sample-size rates.
Key facts. Cited by 73 papers in the map (data through 7 Oct 2026). First posted on arXiv 9 Mar 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 14; 2022: 8; 2023: 11; 2024: 16; 2025: 10; 2026: 14.
73 papers.
Sort:
By area:
- 2021Deep Generative Modelling: A Comparative Review of VAEs, GANs, Normalizing Flows, Energy-Based and Autoregressive Modelscited by 251
- 2021Path Integral Sampler: a stochastic control approach for samplingcited by 175
- 2024Particle Denoising Diffusion Samplercited by 63
- 2021Density Ratio Estimation via Infinitesimal Classificationcited by 59
- 2024Improved off-policy training of diffusion samplerscited by 47
- 2021Solution of Physics-based Bayesian Inverse Problems with Deep Generative Priorscited by 29
- 2021Towards Understanding the Generative Capability of Adversarially Robust Classifierscited by 26
- 2022Aspects of scaling and scalability for flow-based sampling of lattice QCDcited by 26
- 2023Advances in machine-learning-based sampling motivated by lattice quantum chromodynamicscited by 23
- 2021Local-Global MCMC kernels: the best of both worldscited by 21
- 2023On Sampling with Approximate Transport Mapscited by 20
- 2023Improving Gradient-guided Nested Sampling for Posterior Inferencecited by 14
38 more
- 2023Energy-Guided Continuous Entropic Barycenter Estimation for General Costscited by 12
- 2024Amortized Bayesian Multilevel Modelscited by 12
- 2021Learning Equivariant Energy Based Models with Equivariant Stein Variational Gradient Descentcited by 10
- 2022Deterministic Langevin Monte Carlo with Normalizing Flows for Bayesian Inferencecited by 10
- 2021Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Samplingcited by 9
- 2022Wide Bayesian neural networks have a simple weight posterior: theory and accelerated samplingcited by 9
- 2024Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flowscited by 9
- 2022Transport Score Climbing: Variational Inference Using Forward KL and Adaptive Neural Transportcited by 8
- 2023Transition Path Sampling with Boltzmann Generator-based MCMC Movescited by 8
- 2021Sampling in Combinatorial Spaces with SurVAE Flow Augmented MCMCcited by 7
- 2021Deep Learning Hamiltonian Monte Carlocited by 7
- 2023Normalizing flow sampling with Langevin dynamics in the latent spacecited by 7
- 2024Importance Corrected Neural JKO Samplingcited by 7
- 2022Latent Space Diffusion Models of Cryo-EM Structurescited by 6
- 2024Hamiltonian Score Matching and Generative Flowscited by 6
- 2021Entropy-based adaptive Hamiltonian Monte Carlocited by 5
- 2021Bootstrap Your Flowcited by 5
- 2022Transport Reversible Jump Proposalscited by 5
- 2024Quasi-Bayes meets Vinescited by 5
- 2024Disentangling impact of capacity, objective, batchsize, estimators, and step-size on flow VIcited by 5
- 2022Principal Manifold Flowscited by 4
- 2023Automatically Marginalized MCMC in Probabilistic Programmingcited by 4
- 2023Balanced Training of Energy-Based Models with Adaptive Flow Samplingcited by 4
- 2023Self-Tuning Hamiltonian Monte Carlo for Accelerated Samplingcited by 4
- 2023Efficient Multimodal Sampling via Tempered Distribution Flowcited by 3
- 2024Amortized Bayesian Workflowcited by 3
- 2025Model-Informed Flows for Bayesian Inferencecited by 3
- 2025Learning Geometry and Topology via Multi-Chart Flowscited by 3
- 2021Reparameterized Sampling for Generative Adversarial Networkscited by 2
- 2025FlowVAT: Normalizing Flow Variational Inference with Affine-Invariant Temperingcited by 2
- 2021Semi-Empirical Objective Functions for MCMC Proposal Optimizationcited by 1
- 2022Reconstructing the Universe with Variational self-Boosted Samplingcited by 1
- 2024Fast Bayesian Inference for Neutrino Non-Standard Interactions at Dark Matter Direct Detection Experimentscited by 1
- 2024Ai-Sampler: Adversarial Learning of Markov kernels with involutive mapscited by 1
- 2024SoftCVI: Contrastive variational inference with self-generated soft labelscited by 1
- 2024Hamiltonian Monte Carlo Inference of Marginalized Linear Mixed-Effects Modelscited by 1
- 2024Empirical evaluation of normalizing flows in Markov Chain Monte Carlocited by 1
- 2025Enhanced Importance Sampling through Latent Space Exploration in Normalizing Flowscited by 1
50 most cited of 73 All 73 (JSON)
- 2026Transformed Samplers with Variance Reduction7 Oct 2026
- 2026Posterior sampling by source-space MCMC via prior-based few-step transport maps1 Oct 2026
- 2026World Models with Predictable Long-Horizon Marginals26 Sep 2026
- 2026Neural Transport Nested Sampling24 Sep 2026
- 2026Exact Global MCMC with Denoising Diffusion31 Aug 2026
- 2026A Hybrid Nested Harness for Decoupling Structure and Parameters in LLM-Driven Optimization8 Aug 2026
- 2026ReFP-AD: Rectified Flow Preconditioning for Energy-Based Anomaly Detection3 Aug 2026
- 2026LatentFlow: A General Framework for Conditioning Stochastic Processes14 Jul 2026
- 2026Folded Transport MCMC: Eliminating Label Switching by Sampling on a Fundamental Domain3 Jun 2026
- 2026Non-Vacuous Certification of Transport MCMC via Oscillation-Controlled Normalizing Flows31 May 2026
- 2026Self-Certifying Transport MCMC via Dual Spectral-Gap Certificates29 May 2026
- 2026Corrected Integrated Laplace Approximation for Bayesian Inference in Latent Gaussian Models19 May 2026
38 more
- 2026Latent-IMH: Efficient Bayesian Inference for Inverse Problems with Approximate Operators28 Jan 2026
- 2026Neural Architectures for Amortized Bayesian Inference: Statistical Foundations and Empirical Assessments12 Jan 2026
- 2025A Novel Framework Using Variational Inference with Normalizing Flows to Train Transport Reversible Jump Proposals14 Dec 2025
- 2025Particle Monte Carlo methods for Lattice Field Theory19 Nov 2025
- 2025Reducing normalizing flow complexity for MCMC preconditioning4 Nov 2025
- 2025Rotated Mean-Field Variational Inference and Iterative Gaussianization9 Oct 2025
- 2025Learning Geometry and Topology via Multi-Chart Flows30 May 2025
- 2025Model-Informed Flows for Bayesian Inference30 May 2025
- 2025FlowVAT: Normalizing Flow Variational Inference with Affine-Invariant Tempering15 May 2025
- 2025Efficiently Vectorized MCMC on Modern Accelerators20 Mar 2025
- 2025Sample-Efficient Optimization over Generative Priors via Coarse Learnability10 Mar 2025
- 2025Enhanced Importance Sampling through Latent Space Exploration in Normalizing Flows6 Jan 2025
- 2024Empirical evaluation of normalizing flows in Markov Chain Monte Carlo22 Dec 2024
- 2024Disentangling impact of capacity, objective, batchsize, estimators, and step-size on flow VI11 Dec 2024
- 2024Hamiltonian Monte Carlo Inference of Marginalized Linear Mixed-Effects Models31 Oct 2024
- 2024Hamiltonian Score Matching and Generative Flows27 Oct 2024
- 2024Amortized Bayesian Workflow6 Sep 2024
- 2024Amortized Bayesian Multilevel Models23 Aug 2024
- 2024Importance Corrected Neural JKO Sampling29 Jul 2024
- 2024SoftCVI: Contrastive variational inference with self-generated soft labels22 Jul 2024
- 2024Sequential Kalman Tuning of the t-preconditioned Crank-Nicolson algorithm: efficient, adaptive and gradient-free inference for Bayesian inverse problems10 Jul 2024
- 2024Quasi-Bayes meets Vines18 Jun 2024
- 2024Ai-Sampler: Adversarial Learning of Markov kernels with involutive maps4 Jun 2024
- 2024Fast Bayesian Inference for Neutrino Non-Standard Interactions at Dark Matter Direct Detection Experiments23 May 2024
- 2024Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows23 May 2024
- 2024MCMC-driven learning14 Feb 2024
- 2024Particle Denoising Diffusion Sampler9 Feb 2024
- 2024Improved off-policy training of diffusion samplers7 Feb 2024
- 2023Transition Path Sampling with Boltzmann Generator-based MCMC Moves8 Dec 2023
- 2023Improving Gradient-guided Nested Sampling for Posterior Inference6 Dec 2023
- 2023Energy-Guided Continuous Entropic Barycenter Estimation for General Costs2 Oct 2023
- 2023Self-Tuning Hamiltonian Monte Carlo for Accelerated Sampling24 Sep 2023
- 2023Advances in machine-learning-based sampling motivated by lattice quantum chromodynamics3 Sep 2023
- 2023Learning variational autoencoders via MCMC speed measures26 Aug 2023
- 2023Balanced Training of Energy-Based Models with Adaptive Flow Sampling1 Jun 2023
- 2023Normalizing flow sampling with Langevin dynamics in the latent space20 May 2023
- 2023Efficient Multimodal Sampling via Tempered Distribution Flow8 Apr 2023
- 2023On Sampling with Approximate Transport Maps9 Feb 2023
50 newest of 73 All 73 (JSON)
- 2024Particle Denoising Diffusion Samplercited by 63
- 2024Improved off-policy training of diffusion samplerscited by 47
- 2024Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flowscited by 9
- 2022Latent Space Diffusion Models of Cryo-EM Structurescited by 6
- 2024Hamiltonian Score Matching and Generative Flowscited by 6
- 2026Exact Global MCMC with Denoising Diffusioncited by 1
- 2026ReFP-AD: Rectified Flow Preconditioning for Energy-Based Anomaly Detectioncited by 0
- 2026Exact Global MCMC with Denoising Diffusion31 Aug 2026
- 2026ReFP-AD: Rectified Flow Preconditioning for Energy-Based Anomaly Detection3 Aug 2026
- 2024Hamiltonian Score Matching and Generative Flows27 Oct 2024
- 2024Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows23 May 2024
- 2024Particle Denoising Diffusion Sampler9 Feb 2024
- 2024Improved off-policy training of diffusion samplers7 Feb 2024
- 2022Latent Space Diffusion Models of Cryo-EM Structures25 Nov 2022
- 2026World Models with Predictable Long-Horizon Marginalscited by 0
- 2026World Models with Predictable Long-Horizon Marginals26 Sep 2026
- 2021Deep Generative Modelling: A Comparative Review of VAEs, GANs, Normalizing Flows, Energy-Based and Autoregressive Modelscited by 251
- 2021Path Integral Sampler: a stochastic control approach for samplingcited by 175
- 2021Density Ratio Estimation via Infinitesimal Classificationcited by 59
- 2021Solution of Physics-based Bayesian Inverse Problems with Deep Generative Priorscited by 29
- 2021Towards Understanding the Generative Capability of Adversarially Robust Classifierscited by 26
- 2022Aspects of scaling and scalability for flow-based sampling of lattice QCDcited by 26
- 2023Advances in machine-learning-based sampling motivated by lattice quantum chromodynamicscited by 23
- 2021Local-Global MCMC kernels: the best of both worldscited by 21
- 2023On Sampling with Approximate Transport Mapscited by 20
- 2023Improving Gradient-guided Nested Sampling for Posterior Inferencecited by 14
- 2023Energy-Guided Continuous Entropic Barycenter Estimation for General Costscited by 12
- 2024Amortized Bayesian Multilevel Modelscited by 12
12 most cited of 65 All 65 (JSON)
- 2026Transformed Samplers with Variance Reduction7 Oct 2026
- 2026Posterior sampling by source-space MCMC via prior-based few-step transport maps1 Oct 2026
- 2026Neural Transport Nested Sampling24 Sep 2026
- 2026A Hybrid Nested Harness for Decoupling Structure and Parameters in LLM-Driven Optimization8 Aug 2026
- 2026LatentFlow: A General Framework for Conditioning Stochastic Processes14 Jul 2026
- 2026Folded Transport MCMC: Eliminating Label Switching by Sampling on a Fundamental Domain3 Jun 2026
- 2026Non-Vacuous Certification of Transport MCMC via Oscillation-Controlled Normalizing Flows31 May 2026
- 2026Self-Certifying Transport MCMC via Dual Spectral-Gap Certificates29 May 2026
- 2026Corrected Integrated Laplace Approximation for Bayesian Inference in Latent Gaussian Models19 May 2026
- 2026Latent-IMH: Efficient Bayesian Inference for Inverse Problems with Approximate Operators28 Jan 2026
- 2026Neural Architectures for Amortized Bayesian Inference: Statistical Foundations and Empirical Assessments12 Jan 2026
- 2025A Novel Framework Using Variational Inference with Normalizing Flows to Train Transport Reversible Jump Proposals14 Dec 2025
12 newest of 65 All 65 (JSON)