大规模DiT模型部署轻松拿下！港中文等提出TerDiT：性能不输全精度模型

文章预览

↑ 点击 蓝字  关注极市平台 作者丨科技猛兽 编辑丨极市平台 极市导读   受低比特 LLM 量化感知训练方案的启发，本文研究了三值 DiT 模型的 QAT 量化方法，并引入了 DiT 特异性改进来更好地训练。   >> 加入极市CV技术交流群，走在计算机视觉的最前沿 本文目录 1 TerDiT：三值 Diffusion Transformer (来自港中文，上海交大，上海 AI Lab) 1 TerDiT 论文解读 1.1 Diffusion Transformer 变大之后的部署难题 1.2 Diffusion Transformer 1.3 模型量化 1.4 QAT 特定的模型架构改进 1.5 部署策略 1.6 实验设置 1.7 实验结果 太长不看版 大规模的预训练文生图扩散模型促进了高保真度图像的生成，尤其是随着基于 Transformer 的 Diffusion 架构的出现。在这些扩散模型中，Diffusion Transformer 表现出了优越的图像生成能力，做到了较低的 FID 分数和更高的可扩展性。然而，由于参数量大，部署大规模 DiT ………………………………