Energy-Tweedie: Score meets Score, Energy meets Energy
By: Andrej Leban
Denoising and score estimation have long been known to be linked via the classical Tweedie's formula. In this work, we first extend the latter to a wider range of distributions often called "energy models" and denoted elliptical distributions in this work. Next, we examine an alternative view: we consider the denoising posterior $P(X|Y)$ as the optimizer of the energy score (a scoring rule) and derive a fundamental identity that connects the (path-) derivative of a (possibly) non-Euclidean energy score to the score of the noisy marginal. This identity can be seen as an analog of Tweedie's identity for the energy score, and allows for several interesting applications; for example, score estimation, noise distribution parameter estimation, as well as using energy score models in the context of "traditional" diffusion model samplers with a wider array of noising distributions.
Similar Papers
Composition and Control with Distilled Energy Diffusion Models and Sequential Monte Carlo
Machine Learning (Stat)
Teaches computers to create realistic images from noise.
Learning normalized image densities via dual score matching
Machine Learning (CS)
Teaches computers to understand pictures better.
Control Variate Score Matching for Diffusion Models
Machine Learning (CS)
Makes AI create better pictures with less effort.