Local Fokker--Planck Geometry for Score Estimation: Heat-Ball Mean-Value Representations and Exact High-Dimensional Sampling

Jiayao Bai, Lang Deng, Yi Du +1

Score-based generative models and Langevin samplers rely on estimating the score function ${\nabla }_x\log p_t(x)$ of a forward diffusion. Classically this is tractable when the drift…