Shorna Alam

CountsDiff: A diffusion model on the natural numbers for generation and imputation of count-based data

Diffusion models have excelled at generative tasks for both continuous and token-based domains, but their application to discrete ordinal data remains underdeveloped. We present \emph{CountsDiff}, a diffusion framework designed to natively model distributions on the natural numbers. CountsDiff exten…