跳到正文
arXiv cs.AI· Jiongfan Yang, Weichen Fan, Haiwen Diao, Zhongang Cai, Ziwei Liu·· 13 小时前AI 评分31

SpecSlot:面向像素空间生成 Transformer 的读写分配路由式角色耦合 Tokenization

Read, Allocate, Route: Role-Coupled Tokenization for Pixel-Space Generative Transformers

AI 导读

SpecSlot 是一种像素空间 rectified-flow Transformer,通过将每个空间锚点扩展为多个 slot 并各分配不相交的 DCT 系数频带,使输出细节与全局注意力序列长度解耦,仅最低频 slot 参与全局注意力。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:像素空间生成式 Transformer 通常为每个图像块使用一个 token,因此要捕捉更精细的细节就需要更多 token,并使全局注意力变得越来越昂贵。我们观察到,更精细的细节并不一定需要更多的全局通信:低频结构受益于全局交换,而高频细节可以在局部建模。基于这一洞见,我们提出了 SpecSlot,一种像素空间整流流 Transformer,它将输出细节与全局注意力序列的长度解耦。SpecSlot 将每个空间锚点扩展为多个槽位,每个槽位分配一组不相交的 DCT 系数;这将锚点密度与频谱细分分离开来。每个锚点只有最低频的槽位参与全局注意力,而所有槽位在局部交互。这使得模型能够在不扩大全局序列的情况下增加频谱细节。空间头进一步为频谱预测添加局部校正,而生成状态、目标和损失仍保持在像素空间中。在 ImageNet-256 上,SpecSlot 在收敛更快的同时,达到了与 FLOPs 匹配的 JiT 对照模型相当的最终 FID。在 ImageNet-512 上,它以少 3.9% 的 FLOPs 将 FID 较对照模型提升了 2.84,并以低 38.2% 的成本将 FID 与 JiT-B/16 的差距缩小到仅 0.69。在文本到图像生成中,SpecSlot 以低 40.6% 的前向成本在 DPG-Bench 上超过 JiT-B/8 达 1.06。
评论: 22 页,10 张图;正在 ICLR 2027 审稿中
主题: 计算机视觉与模式识别(cs.CV);人工智能(cs.AI);机器学习(cs.LG);神经与进化计算(cs.NE)
引用为: arXiv:2610.40647 [cs.CV]
  (或 arXiv:2610.40647v1 [cs.CV] 用于此版本)
  https://doi.org/10.48550/arXiv.2610.40647

由 DataCite 颁发的 arXiv DOI

提交历史

来自:Jiongfan Yang [查看电子邮件]
[v1] 2026 年 10 月 29 日星期四 17:59:21 UTC(9,853 KB)

来源:arXiv cs.AI · arxiv.org