← Search

Wendi Zheng

10 accepted papers

2026

Vector Quantization using Gaussian Variational Autoencoder

ICML 2026poster

Vector-quantized variational autoencoders (VQ-VAEs) are discrete autoencoders that compress images into discrete tokens. However, they are difficult to train due to discretization. In this paper, we propose a simple yet effective technique dubbed __Gaussian Quant (GQ)__, which first trains a Gaussia…

Cited by 0SourceScholar
2026

VisionReward: Fine-Grained Multi-Dimensional Human Preference Learning for Image and Video Generation

AAAI 2026technical

Visual generative models have achieved remarkable progress in synthesizing photorealistic images and videos, yet aligning their outputs with human preferences across critical dimensions remains a persistent challenge. Though reinforcement learning from human feedback offers promise for preference al

Cited by 0SourcePDFScholar
2025

CogVideoX: Text-to-Video Diffusion Models with An Expert Transformer

ICLR 2025poster

We present CogVideoX, a large-scale text-to-video generation model based on diffusion transformer, which can generate 10-second continuous videos that align seamlessly with text prompts, with a frame rate of 16 fps and resolution of 768 x 1360 pixels. Previous video generation models often struggle…

2025

ZeroFlow: Overcoming Catastrophic Forgetting is Easier than You Think

ICML 2025poster

Backpropagation provides a generalized configuration for overcoming catastrophic forgetting. Optimizers such as SGD and Adam are commonly used for weight updates in continual learning and continual pre-training. However, access to gradient information is not always feasible in practice due to black-…

Cited by 1SourcePDFScholar
2024

Inf-DiT: Upsampling any-resolution image with memory-efficient diffusion transformer.

ECCV 2024poster

"Diffusion models have shown remarkable performance in image generation in recent years. However, due to a quadratic increase in memory during generating ultra-high-resolution images (e.g. 4096 × 4096), the resolution of generated images is often limited to 1024×1024. In this work. we propose a unid…

2024

Relay Diffusion: Unifying diffusion process across resolutions for image synthesis

ICLR 2024spotlight

Diffusion models achieved great success in image synthesis, but still face challenges in high-resolution generation. Through the lens of discrete cosine transformation, we find the main reason is that *the same noise level on a higher resolution results in a higher Signal-to-Noise Ratio in the frequ…

2023

CogVideo: Large-scale Pretraining for Text-to-Video Generation via Transformers

ICLR 2023poster

In this work, we present CogVideo, a 9B-parameter transformer for text-to-video generation. The CogVideo model has been trained by inheriting a pretrained text-to-image model, CogView2, which significantly reduces the training cost and alleviates the problem of scarcity and weak relevance. We also p…

2023

GLM-130B: An Open Bilingual Pre-trained Model

ICLR 2023poster

We introduce GLM-130B, a bilingual (English and Chinese) pre-trained language model with 130 billion parameters. It is an attempt to open-source a 100B-scale model as good as GPT-3 (davinci) and unveil how models of such a scale can be successfully pre-trained. Over the course of this effort, we fac…

2022

CogView2: Faster and Better Text-to-Image Generation via Hierarchical Transformers

NeurIPS 2022accept

Development of transformer-based text-to-image models is impeded by its slow generation and complexity, for high-resolution images. In this work, we put forward a solution based on hierarchical transformers and local parallel autoregressive generation. We pretrain a 6B-parameter transformer with a…

2021

CogView: Mastering Text-to-Image Generation via Transformers

NeurIPS 2021poster

Text-to-Image generation in the general domain has long been an open problem, which requires both a powerful generative model and cross-modal understanding. We propose CogView, a 4-billion-parameter Transformer with VQ-VAE tokenizer to advance this problem. We also demonstrate the finetuning strateg…