Hugging Face Trending Papers

SpeeDiff: Scalable Pixel-Anchored End-to-End Latent Diffusion Model

arXiv Computer Vision
Oct 28

SpeeDiff: Scalable Pixel-Anchored End-to-End Latent Diffusion Model

SpeeDiff is a latent diffusion model that jointly trains a VAE and diffusion network from scratch, using a Tweedie Pixel Reconstruction loss to prevent latent collapse. The approach allows scaling a transformer‑based architecture and improves representation alignment. Experiments show the SpeeDiff‑XL model trains 140× faster than Vanilla SiT and 61× faster than REPA, achieving an FID of 1.50 on ImageNet 256×256 and 1.53 on 512×512 without guidance.

By Bingliang Zhang, Wenda Chu, Yizhuo Li, Linjie Yang, Yisong Yue, Katherine L. Bouman, Yang Song, Qiushan Guo
arXiv AI
Oct 19

Rethinking the design space of Vanilla Pixel Diffusion Transformers

The paper revisits pixel-space diffusion models, showing that a vanilla diffusion transformer can generate directly in pixel space by swapping velocity prediction for clean image prediction. It introduces MFlow, an adaptive parameterization that interpolates between these two prediction modes based on data representation and model capacity, and demonstrates competitive performance on ImageNet, UCF‑101, and text‑to‑image tasks. Additionally, the authors add ConvRefiner, a lightweight convolutional module trained post‑hoc to restore high‑frequency details while preserving FID.

By Soumik Mukhopadhyay, Long Mai, Abhinav Shrivastava, Aniruddha Mahapatra
arXiv Computer Vision
5d ago

PixelMoE: Scaling Pixel-Space Diffusion with LLM-Style Mixture-of-Experts

PixelMoE introduces a pixel‑space generative framework that applies LLM‑style Mixture‑of‑Experts (MoE) to diffusion models, enabling sparse scaling of model capacity without diffusion‑specific routing. It combines Token‑Choice routing, fine‑grained routed experts, a shared expert, and bias‑based load balancing, and adds Pixel‑Space Hierarchical Flow Matching to vary spatial resolution during denoising. Across model scales, PixelMoE outperforms dense counterparts and achieves competitive FID scores on class‑conditional ImageNet‑1K, with code and pretrained models publicly released.

By Ruixiao Dong, Jianmin Bao, Bozhi Luan, Zijie Meng, Xin Tao, Pengfei Wan, Kun Gai, Li Li, Houqiang Li
arXiv Computer Vision
Oct 12

Learning Representations to Guide Pixel Diffusion Transformers

The paper investigates how visual representations contribute to pixel diffusion transformers and finds that higher representation quality does not always improve generative performance, and that separating architecture into specialized decoders mainly redistributes modeling rather than enhancing generative suitability. Based on these findings, the authors propose Representation Guidance (RG), a method that compacts generation-relevant information into a low-dimensional bottleneck and explicitly uses it to steer decoder generation. RG achieves state‑of‑the‑art results on ImageNet 256×256 with fewer parameters, scales efficiently, and extends to text‑to‑image tasks with strong benchmark performance.

By Zihao Pan, Mengting Chen, Cheng Shi, Taihang Hu, Zhengrong Yue, Zikang Wang, Yuan Wang, Wei Li, Sibei Yang
arXiv Computer Vision
Oct 16

UltraDiff: Transferring High-Fidelity Priors to Compressed Latent Spaces for High-Resolution Image Generation

UltraDiff is a new framework for high‑resolution image generation with Latent Diffusion Models that tackles the trade‑off between computational cost and fidelity caused by latent compression. It jointly learns latent diffusion and then distills a marginal score using a dual‑branch DiT called MixTeacher, which models both f8 and f32 distributions and resolves compressed‑space ambiguity with high‑fidelity priors. Vector‑field distillation in the native f32 space forces the student model to learn the teacher’s conditional expectation, filtering out compression artifacts and achieving f8‑level quality while retaining f32 efficiency—generating a 2K image in 6.2 seconds on a single A100 GPU, over 40× faster than the f8 baseline.

By Jingjing Ren, Haitian Zheng, Haoyu Chen, Connelly Barnes, Zhifei Zhang, Yuqian Zhou, Zhe Lin, Lei Zhu