A Continuous Time Framework for Discrete Denoising ModelsDownload PDF

Published: 31 Oct 2022, 18:00, Last Modified: 11 Jan 2023, 17:10NeurIPS 2022 AcceptReaders: Everyone
Keywords: diffusion, score-based, score, markov chain, markov, continuous time
TL;DR: Denoising diffusion models for discrete data using the time reversal of continuous time markov chains.
Abstract: We provide the first complete continuous time framework for denoising diffusion models of discrete data. This is achieved by formulating the forward noising process and corresponding reverse time generative process as Continuous Time Markov Chains (CTMCs). The model can be efficiently trained using a continuous time version of the ELBO. We simulate the high dimensional CTMC using techniques developed in chemical physics and exploit our continuous time framework to derive high performance samplers that we show can outperform discrete time methods for discrete data. The continuous time treatment also enables us to derive a novel theoretical result bounding the error between the generated sample distribution and the true data distribution.
Supplementary Material: pdf
19 Replies

Loading