arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.02493cs.CVcs.AI

PixelGen: 通过感知监督改进像素扩散

PixelGen: Improving Pixel Diffusion with Perceptual Supervision

State Key Laboratory of Multimedia Information Processing, School of Computer Science, Peking University(多媒体信息处理国家重点实验室,计算机学院,北京大学)

Zehong Ma, Ruihan Xu, Shiliang Zhang

AI总结:

PixelGen通过引入LPIPS和P-DINO损失增强x预测,采用噪声门控策略提升图像质量,在ImageNet-256上取得5.11的FID分数,且在文本到图像生成中表现优异。

摘要:

Pixel diffusion直接在像素空间生成图像,避免了两阶段潜在扩散的VAE伪影和表征瓶颈。最近的JiT进一步简化了像素扩散,通过x预测生成干净图像而非速度。然而,标准像素级扩散损失对所有像素平等对待,将模型容量用于感知上不重要的信号,常导致模糊样本。我们提出PixelGen,一种端到端的像素扩散框架,通过感知监督增强x预测。具体而言,PixelGen在x预测上引入两种互补的感知损失:LPIPS用于局部纹理,P-DINO用于全局语义。为保持样本覆盖,PixelGen进一步提出噪声门控策略,仅在低噪声时间步应用这些损失。在无分类器自由引导的ImageNet-256上,PixelGen在80训练周期内取得5.11的FID分数,超越潜在扩散基线。此外,PixelGen能高效扩展到文本到图像生成,在仅6天训练8xH800 GPU上达到0.79的GenEval分数。这些结果表明,感知监督显著缩小了像素与潜在扩散之间的差距,同时保持了简单的单阶段流程。代码可在https://github.com/Zehong-Ma/PixelGen获取。

英文摘要:

Pixel diffusion generates images directly in pixel space, avoiding the VAE artifacts and representational bottlenecks of two-stage latent diffusion. Recent JiT further simplifies pixel diffusion with x-prediction, where the model predicts clean images rather than velocity. However, the standard pixel-wise diffusion loss treats all pixels equally, spending model capacity to perceptually insignificant signals and often leading to blurry samples. We propose PixelGen, an end-to-end pixel diffusion framework that augments x-prediction with perceptual supervision. Specifically, PixelGen introduces two complementary perceptual losses on top of x-prediction: an LPIPS loss for local textures and a P-DINO loss for global semantics. To preserve sample coverage, PixelGen further proposes a noise-gating strategy that applies these losses only at lower-noise timesteps. On ImageNet-256 without classifier-free guidance, PixelGen achieves an FID of 5.11 in 80 training epochs, surpassing the latent diffusion baselines. Moreover, PixelGen scales efficiently to text-to-image generation, reaching a GenEval score of 0.79 with only 6 days of training on 8xH800 GPUs. These results show that perceptual supervision substantially narrows the gap between pixel and latent diffusion while preserving a simple one-stage pipeline. Codes are available at https://github.com/Zehong-Ma/PixelGen.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑