arXiv影像 AI
Looped-DiT:藉由循環 Transformer 區塊實現影像生成的高效運算擴展
Looped-DiT: Scaling Image Generation Efficiency via Recurrent Transformer Blocks
本研究提出 Looped-DiT,在每個去噪步驟中重複運行共享的 Transformer 區塊,使 260M 參數的小模型在多項指標上超越 6.5 倍大的模型,推論運算量降低 4.9 倍。
2 分鐘閱讀
#looped-dit
1 篇文章
Looped-DiT: Scaling Image Generation Efficiency via Recurrent Transformer Blocks
本研究提出 Looped-DiT,在每個去噪步驟中重複運行共享的 Transformer 區塊,使 260M 參數的小模型在多項指標上超越 6.5 倍大的模型,推論運算量降低 4.9 倍。