← Search

Zhisheng Xiao

6 accepted papers

2024

EM Distillation for One-step Diffusion Models

NeurIPS 2024poster

While diffusion models can learn complex distributions, sampling requires a computationally expensive iterative process. Existing distillation methods enable efficient sampling, but have notable limitations, such as performance degradation with very few sampling steps, reliance on training data ac…

Cited by 20SourcePDFScholar
2024

UFOGen: You Forward Once Large Scale Text-to-Image Generation via Diffusion GANs

CVPR 2024highlight

Text-to-image diffusion models have demonstrated remarkable capabilities in transforming text prompts into coherent images yet the computational cost of the multi-step inference remains a persistent challenge. To address this issue we present UFOGen a novel generative model designed for ultra-fast o…

2022

Adaptive Multi-stage Density Ratio Estimation for Learning Latent Space Energy-based Model

NeurIPS 2022accept

This paper studies the fundamental problem of learning energy-based model (EBM) in the latent space of the generator model. Learning such prior model typically requires running costly Markov Chain Monte Carlo (MCMC). Instead, we propose to use noise contrastive estimation (NCE) to discriminatively l…

Cited by 14SourcePDFScholar
2022

Tackling the Generative Learning Trilemma with Denoising Diffusion GANs

ICLR 2022spotlight

A wide variety of deep generative models has been developed in the past decade. Yet, these models often struggle with simultaneously addressing three key requirements including: high sample quality, mode coverage, and fast sampling. We call the challenge imposed by these requirements the generative…

2021

VAEBM: A Symbiosis between Variational Autoencoders and Energy-based Models

ICLR 2021spotlight

Energy-based models (EBMs) have recently been successful in representing complex distributions of small images. However, sampling from them requires expensive Markov chain Monte Carlo (MCMC) iterations that mix slowly in high dimensional pixel space. Unlike EBMs, variational autoencoders (VAEs) gene…

2020

Likelihood Regret: An Out-of-Distribution Detection Score For Variational Auto-encoder

NeurIPS 2020poster

Deep probabilistic generative models enable modeling the likelihoods of very high dimensional data. An important application of generative modeling should be the ability to detect out-of-distribution (OOD) samples by setting a threshold on the likelihood. However, a recent study shows that probabili…