What Matters for Diffusion-Friendly Latent Manifold? Prior-Aligned Autoencoders for Latent Diffusion
Summary
This article introduces Prior-Aligned Autoencoders (PAE), a new method for creating diffusion-friendly latent manifolds that achieves state-of-the-art image generation quality while enabling 13x faster training convergence.
View Cached Full Text
Cached at: 05/11/26, 07:20 AM
Paper page - What Matters for Diffusion-Friendly Latent Manifold? Prior-Aligned Autoencoders for Latent Diffusion
Source: https://huggingface.co/papers/2605.07915 🚀 PAE: State-of-the-Art Latent Diffusion with 13x Faster Training!
We are excited to present Prior-Aligned AutoEncoder (PAE), a new paradigm for constructing diffusion-friendly latent manifolds! By explicitly shaping the latent space geometry, PAE breaks the trade-off between reconstruction fidelity and generation learnability.
🔥 Highlights:
State-of-the-Art Quality: Achieves a new SOTA gFID of 1.03 on ImageNet 256×256, surpassing strong baselines like RAE and FAE.
Unprecedented Efficiency: Enables 13x faster convergence for downstream DiT training. Reaches competitive performance in just 80 epochs compared to 800+ epochs for previous methods.
Diffusion-Friendly Manifold: Explicitly optimizes three key geometric properties: Spatial Structure Coherence, Local Manifold Continuity, and Global Semantic Organization, ensuring smooth and semantically consistent latent spaces.
Robust Few-Step Sampling: Maintains high-quality generation (gFID 1.05) with only 45 denoising steps, thanks to improved local continuity.
Available Models: Pre-trained tokenizers based on DINOv2, SigLIP2, DINOv3, and MAE backbones! Code & Models are fully open-sourced on HuggingFace and ModelScope.
Similar Articles
@xuanchi13: The latent-vs-pixel debate misses the point. GPT Image 2 shows what users notice: pixel-level fidelity. Latent models s…
NVIDIA introduces PiD, a Pixel Diffusion Decoder that replaces traditional VAE/RAE decoders in latent diffusion models, enabling fast, high-resolution decoding with up to 6× speedup and improved visual fidelity.
L2P: Unlocking Latent Potential for Pixel Generation
The L2P paper introduces a Latent-to-Pixel transfer paradigm that leverages pre-trained latent diffusion models to create efficient pixel-space models capable of 4K generation with minimal training overhead.
Smoothing Dark Areas in Molecular Latent Diffusion
This paper introduces TopVAE, a topology-optimized VAE that reduces 'dark areas' in molecular latent diffusion by making the decoder internalize structural and chemical constraints, achieving significant improvements in molecular generation quality.
Don't Retrain, Align: Adapting Autoregressive LMs to Diffusion LMs via Representation Alignment
This paper introduces Repr-Align, a method to adapt autoregressive language models into diffusion language models via representation alignment, achieving up to 4x training acceleration without retraining representations from scratch.
PreDiff-LM: Pretrained Discrete Masked Diffusion Language Modeling with Hybrid Attention
PreDiff-LM proposes a hybrid attention mechanism that preserves causal attention for prompt tokens and bidirectional attention for masked target tokens, enabling adaptation of pretrained autoregressive models for discrete masked diffusion language modeling, achieving improvements in perplexity and downstream tasks over prior diffusion baselines.
