APEX: Asynchronous Prefix Denoising for Extreme Compression
cs.CV, cs.LG
Submitted: 2026-10-09
Updated: 2026-10-09
Comments: 24 pages
Project page: https://daviddinkevich.github.io/APEX
License: http://creativecommons.org/licenses/by/4.0/
The gist: Fewer latent tokens promise cheaper diffusion.
Abstract
Fewer latent tokens promise cheaper diffusion. Yet aggressive latent compression creates a reconstruction-generation dilemma: adding channels improves reconstruction but makes generation harder. We introduce APEX, short for Asynchronous Prefix denoising for EXtreme compression, which pairs a hierarchical latent space with asynchronous denoising over channel groups. Our autoencoder orders channel prefixes from structure to detail, while one shared denoiser generates them at staggered local times without adding tokens. On ImageNet-512, APEX outperforms existing methods at every evaluated channel count under 64x and 128x compression. At matched training and guided sampling compute, it also outperforms a 256-token model at every evaluated budget, with larger gains at lower compute. Code is available at https://daviddinkevich.github.io/APEX.
Sources
- DA-VAE: Plug-in Latent Compression for Diffusion via Detail Alignment
- HieraTok: Multi-Scale Visual Tokenizer Improves Image Reconstruction and Generation
- DC-AE 1.5: Accelerating Diffusion Model Convergence with Structured Latent Space
- Diffusion Models Beat GANs on Image Synthesis
- Latent-Compressed Variational Autoencoder for Video Diffusion Models
- Unified Latents (UL): How to train your latents
- Auto-Encoding Variational Bayes
- Boosting Generative Image Modeling via Joint Image-Feature Synthesis
- Matryoshka Representation Learning
- Toward Diffusible High-Dimensional Latent Spaces: A Frequency Perspective
- REPA-E: Unlocking VAE for End-to-End Tuning with Latent Diffusion Transformers
- WF-VAE: Enhancing Video VAE by Wavelet-Driven Energy Flow for Latent Video Diffusion Model
- Flow Matching for Generative Modeling
- Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow
- Decoupled Weight Decay Regularization
- SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers
- Spectrum Matching: a Unified Perspective for Superior Diffusability in Latent Diffusion
- DINOv2: Learning Robust Visual Features without Supervision
- Semantics Lead the Way: Harmonizing Semantic and Texture Modeling with Asynchronous Latent Diffusion
- ChannelTok: Efficient Flexible-Length Vision Tokenization
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models