用压缩+精修策略,让扩散模型更快生成高质量图像
RePack then Refine: Efficient Diffusion Transformer with Vision Foundation Model
- 先将视觉大模型特征压缩到低维空间,去除冗余
- 在压缩空间训练扩散模型,64轮即达FID 1.82
- 最后用细节修复模块提升画质,FID降至1.65
语义丰富的视觉基础模型(VFMs)特征被用于增强潜在扩散模型(LDMs)。然而,原始的VFM特征通常维度高且冗余,增加了扩散变换器(DiTs)的学习难度并降低训练效率。本文提出RePack then Refine三阶段框架,在引入语义丰富特征的同时进一步提升学习效率。首先,RePack模块将高维特征投影至紧凑的低维流形,过滤冗余并保留关键结构信息;其次,在高度压缩的潜在空间上训练标准DiT进行生成建模;最后,为恢复压缩过程中损失的高频细节,提出一个潜在引导精修模块(Latent-Guided Refiner),该模块最后训练以增强图像细节。在ImageNet-1K上,RePack-DiT-XL/1仅需64个训练周期即达到FID 1.82,加入精修模块后性能进一步提升至FID 1.65,显著优于最新LDMs的收敛效率。结果表明,先打包再精修是平衡生成保真度与训练效率的有效策略。源代码已公开于https://github.com/guanfangdong/RePack-then-Refine。
原文摘要 · Abstract (English)
Semantic-rich features from Vision Foundation Models (VFMs) have been leveraged to enhance Latent Diffusion Models (LDMs). However, raw VFM features are typically high-dimensional and redundant, increasing the difficulty of learning and reducing training efficiency for Diffusion Transformers (DiTs). In this paper, we propose Repack then Refine, a three-stage framework that brings the semantic-rich VFM features to DiT while further accelerating learning efficiency. Specifically, the RePack module projects the high-dimensional features onto a compact, low-dimensional manifold. This filters out the redundancy while preserving essential structural information. A standard DiT is then trained for generative modeling on this highly compressed latent space. Finally, to restore the high-frequency details lost due to the compression in RePack, we propose a Latent-Guided Refiner, which is trained lastly for enhancing the image details. On ImageNet-1K, RePack-DiT-XL/1 achieves an FID of 1.82 in only 64 training epochs. With the Refiner module, performance further improves to an FID of 1.65, significantly surpassing latest LDMs in terms of convergence efficiency. Our results demonstrate that packing VFM features, followed by targeted refinement, is a highly effective strategy for balancing generative fidelity with training efficiency. Source code is publicly available at https://github.com/guanfangdong/RePack-then-Refine.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。