本文研究扩散Transformer(DiT)中的异常标记问题,发现其在编码器和去噪器中均存在,且简单掩蔽无效。提出双阶段寄存器(DSR)干预方法,通过训练或测试时寄存器及扩散寄存器,在ImageNet和文本到图像生成中减少异常伪影,提升生成质量,强调控制异常标记对构建更强DiT的重要性。
完成下面两步后,将自动完成登录并继续当前操作。