cs.LGJul 28, 2026

Guiding Posterior Exploration with Optimizer-Derived Geometry

Authors: Moritz SchlagerEmanuel SommerThomas MöllenhoffDavid Rügamer

Organizations: TUM, MCML · LMU Munich, MCML · RIKEN AIP

Abstract

Sampling-based methods offer a principled approach to uncertainty quantification in Bayesian neural networks. Their practical use, however, is often challenged by the computational cost of exploring high-dimensional and multimodal posterior distributions. To overcome these difficulties, Bayesian Deep Ensembles, i.e., warmstarting the sampling from several optimized solutions, have proven to be an effective strategy. In this paper, we demonstrate that curvature estimates computed during the warmstart as a byproduct in adaptive optimizers such as AdamW can inform the sampling phase at negligible additional cost. Specifically, our proposed preconditioned sampling strategy based on optimizer-derived geometries can substantially reduce or even eliminate the need for a lengthy sampling burn-in phase and leads to greater numerical stability. This approach consistently maintains or improves predictive performance and uncertainty quantification without any additional computational costs. We confirm the consistency of our findings across various datasets and network architectures.

Explore similar work

CardsList
  1. Don't Stop Me Yet: Sampling Loss Minima via Dissipative Riemannian Mechanics

    May 14, 2026Albert Kjøller Jacobsen, Leo Uhre Jakobsen, Johanna Marie Gegenfurtner +1SamplersReparameterization

  2. Calibrated Sampling-Free Uncertainty Estimation in Bayesian Deep Learning

    Jun 15, 2026Tobias Jan Wieczorek, Leon de Andrade, Thomas Möllenhoff +1Bayesian Neural NetworksDeep Learning