English

E2ED^2:Direct Mapping from Noise to Data for Enhanced Diffusion Models

Computer Vision and Pattern Recognition 2025-03-11 v2

Abstract

Diffusion models have established themselves as the de facto primary paradigm in visual generative modeling, revolutionizing the field through remarkable success across various diverse applications ranging from high-quality image synthesis to temporal aware video generation. Despite these advancements, three fundamental limitations persist, including 1) discrepancy between training and inference processes, 2) progressive information leakage throughout the noise corruption procedures, and 3) inherent constraints preventing effective integration of modern optimization criteria like perceptual and adversarial loss. To mitigate these critical challenges, we in this paper present a novel end-to-end learning paradigm that establishes direct optimization from the final generated samples to initial noises. Our proposed End-to-End Differentiable Diffusion, dubbed E2ED^2, introduces several key improvements: it eliminates the sequential training-sampling mismatch and intermediate information leakage via conceptualizing training as a direct transformation from isotropic Gaussian noise to the target data distribution. Additionally, such training framework enables seamless incorporation of adversarial and perceptual losses into the core optimization objective. Comprehensive evaluation across standard benchmarks including COCO30K and HW30K reveals that our method achieves substantial performance gains in terms of Fr\'echet Inception Distance (FID) and CLIP score, even with fewer sampling steps (less than 4). Our findings highlight that the end-to-end mechanism might pave the way for more robust and efficient solutions, \emph{i.e.,} combining diffusion stability with GAN-like discriminative optimization in an end-to-end manner.

Keywords

Cite

@article{arxiv.2412.21044,
  title  = {E2ED^2:Direct Mapping from Noise to Data for Enhanced Diffusion Models},
  author = {Zhiyu Tan and WenXu Qian and Hesen Chen and Mengping Yang and Lei Chen and Hao Li},
  journal= {arXiv preprint arXiv:2412.21044},
  year   = {2025}
}

Comments

Technical report

R2 v1 2026-06-28T20:52:16.297Z