PixelU: A U-Shaped Transformer for Efficient End-to-End Pixel Diffusion
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
End-to-end pixel-space diffusion models bypass the lossy compression of Latent Diffusion Models (LDMs) but struggle to jointly model low-frequency semantics and high-frequency signals in high-dimensional space. Existing works heavily rely on complex pixel decoders to alleviate this issue. In this paper, we challenge this trend by revealing that these decoders primarily compensate for the optimization difficulties inherent to velocity prediction (v-prediction). Under the clean data paradigm (x-prediction), they are redundant. Motivated by this insight, we advocate for simplicity over complexity and introduce PixelU, a minimalist, single-stage U-shaped Diffusion Transformer tailored for pixel space. PixelU abandons auxiliary decoders in favor of zero-cost skip connections, which provide an "information highway" that directly routes uncorrupted high-frequency spatial details from shallow to deep layers. To further enable the backbone to focus exclusively on modeling low-frequency semantics, we introduce a constant-channel spatial down-sampling mechanism as a natural low-pass filter, which compresses deep features into a compact, low-frequency semantic manifold. Extensive experiments demonstrate that this decoupling of frequencies could outperform the strong baseline (JiT-G) with only about 1/3 of its computation cost. On ImageNet 256×256 and 512×512, PixelU achieves FID of 1.63 and 1.92 respectively, surpassing recent pixel-space methods and establishing a simple yet powerful new paradigm for end-to-end diffusion models.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 50 of its 55 references that are in the map. First posted on arXiv 26 Jun 2026.
Builds on
50 of 55 references.
Closest ancestors · 31
Same direction, newest first
- 2026FrequencyBooster: Full-Frequency Modeling for High-Fidelity Pixel Diffusion18 May
- 2026HyperDiT: Hyper-Connected Transformers for High-Fidelity Pixel-Space Diffusion15 May
- 2025PixelDiT: Pixel Diffusion Transformers for Image Generation25 Nov
- 2025DeCo: Frequency-Decoupled Pixel Diffusion for End-to-End Image Generation24 Nov
- 2025DiP: Taming Diffusion Models in Pixel Space24 Nov
- 2025Latent Diffusion Model without Variational Autoencoder17 Oct
- 2025Diffusion Transformers with Representation Autoencoders13 Oct
- 2025PixNerd: Pixel Neural Field Diffusion31 Jul
- 2025DDT: Decoupled Diffusion Transformer8 Apr
- 2025U-REPA: Aligning Diffusion U-Nets to ViTs24 Mar
- 2025Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models2 Jan
- 2024Deep Compression Autoencoder for Efficient High-Resolution Diffusion Models14 Oct
19 more
- 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think9 Oct
- 2024U-DiTs: Downsample Tokens in U-Shaped Diffusion Transformers4 May
- 2024Applying Guidance in a Limited Interval Improves Sample and Distribution Quality in Diffusion Models11 Apr
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesis5 Mar
- 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers16 Jan
- 2023Relay Diffusion: Unifying diffusion process across resolutions for image synthesis4 Sep
- 2023Fast Training of Diffusion Models with Masked Transformers15 Jun
- 2023A Unified Framework for U-Net Design and Analysis31 May
- 2023Understanding Diffusion Objectives as the ELBO with Simple Data Augmentation1 Mar
- 2023Simple diffusion: End-to-end diffusion for high resolution images26 Jan
- 2022Scalable Diffusion Models with Transformers19 Dec
- 2022All are Worth Words: A ViT Backbone for Diffusion Models25 Sep
- 2022Classifier-Free Diffusion Guidance26 Jul
- 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
- 2021Cascaded Diffusion Models for High Fidelity Image Generation30 May
- 2021Diffusion Models Beat GANs on Image Synthesis11 May
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Implicit Models6 Oct
- 2020Denoising Diffusion Probabilistic Models19 Jun
Other references · 19
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,479
- 2023Vision Transformers Need Registerscited by 844
- 2024Autoregressive Image Generation without Vector Quantizationcited by 644
- 2022Vision Transformer Adapter for Dense Predictionscited by 564
- 2022StyleGAN-XL: Scaling StyleGAN to Large Diverse Datasetscited by 489
- 2022How Do Vision Transformers Work?cited by 305
- 2022Scalable Adaptive Computation for Iterative Generationcited by 190
7 more
- 2022Inception Transformercited by 99
- 2025PixelFlow: Pixel-Space Generative Models with Flowcited by 91
- 2024Normalizing Flows are Capable Generative Modelscited by 89
- 2025What matters for Representation Alignment: Global Information or Spatial Structure?cited by 76
- 2024JetFormer: An Autoregressive Generative Model of Raw Images and Textcited by 44
- 2025Fractal Generative Modelscited by 44
- 2025FARMER: Flow AutoRegressive Transformer over Pixelscited by 17
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026A Frozen Pixel-Space Diffusion Model Can Guide Itself with Its Own Samplescited by 2
- 2026Pixel-Space Diffusion via Observation Operatorscited by 1
- 2026Energy-Guided Flow Matchingcited by 0
- 2026Advanced Pixel Diffusion Model with Guided Sparse Global Refinementcited by 0
- 2026Residual-Stream Burden Shapes Representation Learning in Diffusion Transformerscited by 0
- 2026Persistence Forcing: Exploiting Feature Specialization in Pixel-Space Diffusioncited by 0
- 2026Persistence Forcing: Exploiting Feature Specialization in Pixel-Space Diffusion28 Sep 2026
- 2026Residual-Stream Burden Shapes Representation Learning in Diffusion Transformers27 Sep 2026
- 2026Advanced Pixel Diffusion Model with Guided Sparse Global Refinement1 Sep 2026
- 2026Pixel-Space Diffusion via Observation Operators22 Aug 2026
- 2026Energy-Guided Flow Matching6 Aug 2026
- 2026A Frozen Pixel-Space Diffusion Model Can Guide Itself with Its Own Samples31 Jul 2026
Similar, not cited
Close in wording, no citation either way
- 2026Pixel-Space Diffusion Transformers
- 2026PixelDense: Dense Prediction as Representation Alignment for Pixel Diffusion
- 2026PixelDiT2: Representation-Grounded Pixel Diffusion Transformers
- 2024Simpler Diffusion (SiD2): 1.5 FID on ImageNet512 with pixel-space diffusion
- 2026JLT: Clean-Latent Prediction in Latent Diffusion Transformers
- 2026Structured Residual Connectivity Matters for Diffusion Transformers
- 2026Should We Skip Diffusion?
- 2026PiD: Fast and High-Resolution Latent Decoding with Pixel Diffusion