寄存器对像素空间扩散Transformer至关重要

这项工作研究寄存器token在扩散Transformer（DiT）中的作用。作者发现，DiT不像ViT那样出现高范数patch-token异常，却仍能从寄存器中获益；而且寄存器在像素空间DiT中比潜空间DiT更有效。中间表示分析显示，高噪声阶段寄存器可改善特征图，并据此提出放大寄存器贡献的Register Guidance。