DistriFusion:高分辨率扩散模型的分布式并行推理

💡 原文中文,约400字,阅读约需1分钟。
📝

内容提要

我们提出了一种有效的去噪扩散模型,用于生成高分辨率图像。该模型在小尺寸图像块上训练,通过特征拼贴预测移位图像块的特征,实现整个图像的无缝生成。与经典的扩散模型相比,Patch-DM还减少了内存复杂度。

🎯

关键要点

  • 提出了一种有效的去噪扩散模型,用于生成高分辨率图像。
  • 模型在小尺寸图像块上训练,例如64x64。
  • 算法名为Patch-DM,设计了一种新的特征拼贴策略。
  • 特征拼贴避免了合成大尺寸图像时的边界伪影。
  • 通过特征拼贴实现整个图像的无缝生成。
  • 在新收集的自然图像数据集和标准基准上产生高质量图像合成结果。
  • 与以前的基于图像块的生成方法比较,获得了最先进的FID分数。
  • Patch-DM减少了内存复杂度。
➡️

继续阅读