arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.01516cs.CV

用于人物图像合成的跨视图掩码扩散 Transformer

Cross-view Masked Diffusion Transformers for Person Image Synthesis

KAIST(韩国科学技术院)

Trung X. Pham, Zhang Kang, Chang D. Yoo

AI总结:

提出 X-MDPT,一种用于姿态引导人物图像生成的新型掩码扩散 Transformer 模型,通过三个关键模块在 DeepFashion 数据集上超越现有方法,以更少参数实现更优性能和更快推理。

摘要:

我们提出了 X-MDPT(跨视图掩码扩散预测 Transformer),一种用于姿态引导人物图像生成的新型扩散模型。X-MDPT 通过采用在潜在补丁上操作的掩码扩散 Transformer 脱颖而出,这有别于现有工作中常用的 Unet 结构。该模型包含三个关键模块:1) 去噪扩散 Transformer,2) 将条件整合为单一向量供扩散过程使用的聚合网络,以及 3) 利用参考图像语义信息增强表示学习的掩码交叉预测模块。X-MDPT 展现出可扩展性,随着模型增大,FID、SSIM 和 LPIPS 指标均有改善。尽管设计简单,我们的模型在 DeepFashion 数据集上超越了现有先进方法,同时在训练参数、训练时间和推理速度方面表现出高效性。我们紧凑的 33MB 模型实现了 7.42 的 FID,仅用少 $11\times$ 的参数便超越了先前的 Unet 潜在扩散方法(FID 为 8.07)。我们的最佳模型以 $\ rac{2}{3}$ 的参数超越了基于像素的扩散方法,并实现了 $5.43 \ imes$ 的推理速度提升。代码可在 https://github.com/trungpx/xmdpt 获取。

英文摘要:

We present X-MDPT ($\underline{Cross}$-view $\underline{M}$asked $\underline{D}$iffusion $\underline{P}$rediction $\underline{T}$ransformers), a novel diffusion model designed for pose-guided human image generation. X-MDPT distinguishes itself by employing masked diffusion transformers that operate on latent patches, a departure from the commonly-used Unet structures in existing works. The model comprises three key modules: 1) a denoising diffusion Transformer, 2) an aggregation network that consolidates conditions into a single vector for the diffusion process, and 3) a mask cross-prediction module that enhances representation learning with semantic information from the reference image. X-MDPT demonstrates scalability, improving FID, SSIM, and LPIPS with larger models. Despite its simple design, our model outperforms state-of-the-art approaches on the DeepFashion dataset while exhibiting efficiency in terms of training parameters, training time, and inference speed. Our compact 33MB model achieves an FID of 7.42, surpassing a prior Unet latent diffusion approach (FID 8.07) using only $11\times$ fewer parameters. Our best model surpasses the pixel-based diffusion with $\frac{2}{3}$ of the parameters and achieves $5.43 \times$ faster inference. The code is available at https://github.com/trungpx/xmdpt.

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

↑