mlsanityData to 7 Oct 2026

Mixture-of-Gaussians-Guided Schedule Design for Brownian Bridge Diffusion Models

arXiv 2607.03517 ↗First posted 3 Jul 2026cs.LG cs.CVcited by 1

On the map: Diffusion models → Inverse problems and image restoration

Abstract (arXiv)

Brownian Bridge Diffusion Models (BBDM) offer an appealing framework for image restoration and inverse problems by constructing a stochastic bridge from the clean signal directly to the degraded observation, rather than to pure noise. Despite their promise, the choice of bridge schedule is typically inherited from heuristics, and a principled analytical framework for schedule design has been lacking. In this work, we develop such a framework by offering a novel analysis of BBDM reverse dynamics under a Mixture-of-Gaussians (MoG) prior. This setting yields a closed-form ideal posterior and a corresponding MMSE denoiser, while the BBDM-induced reconstruction law is captured analytically through a tractable surrogate. Building on these expressions, we formulate two complementary schedule-design objectives: a Wasserstein criterion targeting perceptual quality and an MSE criterion targeting reconstruction fidelity. Our work exposes an inherent tradeoff between the two and proves the existence of universal schedules for both that are independent of the degradation and prior. Extensive experiments on controlled MoG settings confirm full alignment between theory and practice, and experiments on the FFHQ dataset across inpainting, deblurring, and super-resolution tasks validate the practical value of our schedule-design criteria.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Inverse problems and image restoration, one of 14 directions in Diffusion models. Builds on 4 of its 27 references that are in the map. First posted on arXiv 3 Jul 2026.

Builds on

4 of 27 references.

Closest ancestors · 1

Same direction, newest first

  1. 2026Analyzing and Guiding Zero-Shot Posterior Sampling in Diffusion Models7 Feb

Other references · 3

  1. 2021Image Super-Resolution via Iterative Refinementcited by 1,506
  2. 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
  3. 2025Spectral Analysis of Diffusion Models with Application to Schedule Designcited by 12

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026Energy-oriented Diffusion Bridge for Image Restoration with Foundational Diffusion Models
  2. 2026SDDBMs: Soft Denoising Diffusion Bridge Models
  3. 2026A Posterior-Dynamics Framework for Imaging Inverse Problems with Pretrained Diffusion Priors
  4. 2025DAPS++: Rethinking Diffusion Inverse Problems with Decoupled Posterior Annealing
  5. 2023A Variational Perspective on Solving Inverse Problems with Diffusion Models
  6. 2026Timestep Rescheduling in Diffusion Inversion
  7. 2023Image Restoration Through Generalized Ornstein-Uhlenbeck Bridge
  8. 2024Frequency-Guided Posterior Sampling for Diffusion-Based Image Restoration