cs.LGFeb 20, 2026

The Geometry of Noise: Why Diffusion Models Don't Need Noise Conditioning

Authors: Mojtaba Sahraee-Ardakan, Mauricio Delbracio, Peyman Milanfar

Organizations: Google

Abstract

Autonomous (noise-agnostic) generative models, such as Equilibrium Matching and blind diffusion, challenge the standard paradigm by learning a single, time-invariant vector field that operates without explicit noise-level conditioning. While recent work suggests that high-dimensional concentration allows these models to implicitly estimate noise levels from corrupted observations, a fundamental paradox remains: what is the underlying landscape being optimized when the noise level is treated as a random variable, and how can a bounded, noise-agnostic network remain stable near the data manifold where gradients typically diverge? We resolve this paradox by formalizing Marginal Energy, Emarg(u)=−log⁡p(u)E_{\text{marg}}(\mathbf{u}) = -\log p(\mathbf{u}), where p(u)=∫p(u∣t)p(t)dtp(\mathbf{u}) = \int p(\mathbf{u}|t)p(t)dt is the marginal density of the noisy data integrated over a prior distribution of unknown noise levels. We prove that generation using autonomous models is not merely blind denoising, but a specific form of Riemannian gradient flow on this Marginal Energy. Through a novel relative energy decomposition, we demonstrate that while the raw Marginal Energy landscape possesses a 1/tp1/t^p singularity normal to the data manifold, the learned time-invariant field implicitly incorporates a local conformal metric that perfectly counteracts the geometric singularity, converting an infinitely deep potential well into a stable attractor. We also establish the structural stability conditions for sampling with autonomous models. We identify a ``Jensen Gap'' in noise-prediction parameterizations that acts as a high-gain amplifier for estimation errors, explaining the catastrophic failure observed in deterministic blind models. Conversely, we prove that velocity-based parameterizations are inherently stable because they satisfy a bounded-gain condition that absorbs posterior uncertainty into a smooth geometric drift.

Figures & tables

Explore similar work

CardsList
  1. Noise Schedule Design for Diffusion Models: An Optimal Control Perspective

    May 21, 2026Seo Taek Kong, Weina Wang, R. SrikantDiffusion SamplingDiffusion-Based Image Generation

  2. Do Heavy Tails Help Diffusion? On the Subtle Trade-off Between Initialization and Training

    May 13, 2026Hamza Cherkaoui, Hélène Halconruy, Antonio OcelloDiffusion Model SamplingHeavy-Tailed Noise