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

跨分辨率分布匹配的扩散蒸馏

Cross-Resolution Distribution Matching for Diffusion Distillation

Huawei(华为);Nanjing University of Science and Technology(南京理工大学);HiThink Research(海康研究院)

Feiyang Chen, Hongpeng Pan, Haonan Xu, Xinyu Duan, Zhefeng Wang, Yang Yang

AI总结:

提出跨分辨率分布匹配蒸馏(RMD)框架,通过logSNR映射和分布匹配弥合分辨率间隙,实现高保真、少步数多分辨率级联推理,在SDXL和Wan2.1-14B上分别加速33.4倍和25.6倍。

摘要:

扩散蒸馏是加速图像和视频生成的核心,但现有方法受限于去噪过程,步数减少已基本饱和。部分时间步的低分辨率生成可进一步加速推理,但由于跨分辨率分布差距,质量明显下降。我们提出跨分辨率分布匹配蒸馏(RMD),一种新颖的蒸馏框架,弥合跨分辨率分布差距,实现高保真、少步数多分辨率级联推理。具体来说,RMD使用对数信噪比(logSNR)曲线为每个分辨率划分时间步区间,并引入基于logSNR的映射来补偿分辨率引起的偏移。沿分辨率轨迹进行分布匹配,以减少低分辨率生成器分布与教师高分辨率分布之间的差距。此外,在上采样过程中引入预测噪声重注入机制,以稳定训练并提高合成质量。定量和定性结果表明,RMD在保持高保真生成的同时,在各种骨干网络上加速推理。值得注意的是,RMD在SDXL上实现高达33.4倍的加速,在Wan2.1-14B上实现25.6倍的加速,同时保持高视觉保真度。

英文摘要:

Diffusion distillation is central to accelerating image and video generation, yet existing methods are fundamentally limited by the denoising process, where step reduction has largely saturated. Partial timestep low-resolution generation can further accelerate inference, but it suffers noticeable quality degradation due to cross-resolution distribution gaps. We propose Cross-Resolution Distribution Matching Distillation (RMD), a novel distillation framework that bridges cross-resolution distribution gaps for high-fidelity, few-step multi-resolution cascaded inference. Specifically, RMD divides the timestep intervals for each resolution using logarithmic signal-to-noise ratio (logSNR) curves, and introduces logSNR-based mapping to compensate for resolution-induced shifts. Distribution matching is conducted along resolution trajectories to reduce the gap between low-resolution generator distributions and the teacher's high-resolution distribution. In addition, a predicted-noise re-injection mechanism is incorporated during upsampling to stabilize training and improve synthesis quality. Quantitative and qualitative results show that RMD preserves high-fidelity generation while accelerating inference across various backbones. Notably, RMD achieves up to 33.4X speedup on SDXL and 25.6X on Wan2.1-14B, while preserving high visual fidelity.

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

↑