Poster
Generative Fractional Diffusion Models
Gabriel Nobis · Maximilian Springenberg · Marco Aversa · Michael Detzel · Rembert Daems · Roderick Murray-Smith · Shinichi Nakajima · Sebastian Lapuschkin · Stefano Ermon · Tolga Birdal · Manfred Opper · Christoph Knochenhauer · Luis Oala · Wojciech Samek
East Exhibit Hall A-C #2700
Abstract:
We introduce the first continuous-time score-based generative model that leverages fractional diffusion processes for its underlying dynamics. Although diffusion models have excelled at capturing data distributions, they still suffer from various limitations such as slow convergence, mode-collapse on imbalanced data, and lack of diversity. These issues are partially linked to the use of light-tailed Brownian motion (BM) with independent increments. In this paper, we replace BM with an approximation of its non-Markovian counterpart, fractional Brownian motion (fBM), characterized by correlated increments and Hurst index $H \in (0,1)$, where $H=1/2$ recovers the classical BM. To ensure tractable inference and learning, we employ a recently popularized Markov approximation of fBM (MA-fBM) and derive its reverse time model, resulting in \emph{generative fractional diffusion models} (GFDM). We characterize the forward dynamics using a continuous reparameterization trick and propose an augmented score matching to efficiently learn the score-function, which is partly known in closed form, at minimal added cost. The ability to drive our diffusion model via fBM provides flexibility and control. $H<1/2$ enters the regime of \emph{rough paths} leading to diverse outcome whereas $H>1/2$ regularizes diffusion paths and invokes long-term memory as well as a heavy-tailed behaviour (super-diffusion). The Markov approximation allows added control by varying the number of Markov processes linearly combined to approximate fBM. Our evaluations on real image datasets demonstrate that GFDM achieves greater pixel-wise diversity and enhanced image quality, as indicated by a lower FID, offering a promising alternative to traditional diffusion models.
Live content is unavailable. Log in and register to view live content