PiD: Fast and High-Resolution Latent Decoding with Pixel Diffusion

2026-05-22Computer Vision and Pattern Recognition

Computer Vision and Pattern Recognition
AI summary

The authors present PiD, a new way to turn compressed image representations (latents) into full high-resolution pictures. Unlike traditional methods that focus on reversing the compression, PiD uses a pixel-level diffusion process to directly generate detailed large images efficiently. It combines decoding and upscaling in one step, making it faster and producing clearer images, especially for very big pictures. PiD works with different types of latent inputs and runs quickly on standard GPUs, significantly cutting down the time compared to previous upscaling methods.

latent spacelatent diffusionpixel diffusionimage decodingimage upscalingVAE (Variational Autoencoder)RAE (Reconstruction Autoencoder)denoising diffusionsuper-resolutionmodel distillation
Authors
Yifan Lu, Qi Wu, Jay Zhangjie Wu, Zian Wang, Huan Ling, Sanja Fidler, Xuanchi Ren
Abstract
Most practical high-resolution text-to-image systems, including latent diffusion and autoregressive models, perform generation in a compact latent space, and a decoder maps the generated latents back to pixels. Yet the latent-to-pixel decoder is reconstruction-oriented, optimized to invert the encoder rather than synthesize more details, and becomes increasingly costly at megapixel scale. This drawback calls for a more expressive and efficient decoding paradigm. Motivated by recent progress in scalable pixel-space diffusion, we introduce PiD, a Pixel diffusion Decoder that reformulates latent decoding as conditional pixel diffusion, unifying decoding and upsampling into one generative module. By denoising directly in high-resolution pixel space, PiD synthesizes $4\times$ and even $8\times$ upscaled images with low latency. For latent conditioning, a lightweight sigma-aware adapter injects noise-corrupted latents into the pixel diffusion backbone, enabling PiD to decode partially denoised latents and terminate the latent diffusion process early. To further improve efficiency, we distill the model using DMD2, reducing inference to just 4 steps. PiD applies to both conventional VAE latents and semantic latents (e.g., SigLIP, DINOv2) used in recent RAE-based models. PiD decodes latents of $512 \times 512$ images into $2048 \times 2048$ pixels in under 1 second with 13 GB peak memory on a consumer RTX 5090, and as fast as 210 ms on a GB200 GPU, about $6\times$ faster than cascaded diffusion-based super-resolution pipelines with better visual fidelity.