Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks
Luwei Sun, Dongrui Shen, Jianfe Li, Yulong Zhao, Han Feng
TL;DR
The paper addresses the challenge of approximating ratio-type functionals $\frac{f_1}{f_2}$ arising in conditional diffusion-model training. It introduces SignReLU neural networks to directly approximate such ratios, with explicit approximation bounds and stability control near small denominators. Specializing to diffusion models (DDPMs), the authors construct a neural estimator for the reverse process and decompose the excess KL risk into approximation and estimation components, providing finite-sample guarantees and a training-set augmentation strategy that reduces statistical error. Under Hölder-smooth assumptions, the work establishes near-minimax convergence rates and demonstrates how ratio-based analysis yields end-to-end guarantees for diffusion-based conditional generation with practical implications for sample efficiency and stability.
Abstract
Motivated by challenges in conditional generative modeling, where the target conditional density takes the form of a ratio f1 over f2, this paper develops a theoretical framework for approximating such ratio-type functionals. Here, f1 and f2 are kernel-based marginal densities that capture structured interactions, a setting central to diffusion-based generative models. We provide a concise proof for approximating these ratio-type functionals using deep neural networks with the SignReLU activation function, leveraging the activation's piecewise structure. Under standard regularity assumptions, we establish L^p(Omega) approximation bounds and convergence rates. Specializing to Denoising Diffusion Probabilistic Models (DDPMs), we construct a SignReLU-based neural estimator for the reverse process and derive bounds on the excess Kullback-Leibler (KL) risk between the generated and true data distributions. Our analysis decomposes this excess risk into approximation and estimation error components. These results provide generalization guarantees for finite-sample training of diffusion-based generative models.
