一维有序令牌:让自回归图像生成的测试时搜索更高效 (1D) Ordered Tokens Enable Efficient Test-Time Search
证明1D有序令牌的coarse-to-fine结构让AR图像生成更适合测试时搜索,甚至无需训练AR模型即可生成图像。
前置知识
自回归图像生成 (Autoregressive Image Generation)
将图像token化为一串离散token序列,然后像语言模型那样按顺序逐个预测下一个token,最终通过detokenizer解码回像素空间。代表方法有Parti、LlamaGen、Janus等。训练目标是经典的最大似然 $p_\theta(x | c) = \prod_{t=1}^{T} p_\theta(x_t | x_{<t}, c)$,其中 $c$ 是文本条件。
本文研究的核心对象就是AR图像模型,1D vs 2D的token结构直接影响这种next-token预测范式下搜索算法能拿到多少有用信号。
1D有序令牌 (1D Ordered Tokens)
一类特殊的图像tokenizer,输出的token序列被显式训练成coarse-to-fine或语义优先级的顺序。代表是FlexTok:通过nested dropout训练,使前缀也能被解码为有意义的图像,且早期token携带全局语义(颜色、主体类别),后期token才补充纹理细节。其vocabulary大小通常为64K,支持变长解码。
这是论文研究的核心token结构。它的'语义有序性'是让verifier在搜索早期就能可靠打分的前提,也是与传统2D网格tokenizer的本质区别。
2D网格令牌 (2D Grid Tokens)
VQGAN类标准方案,将图像切成固定的空间网格,每个token对应一个局部像素块。自回归生成按raster-scan顺序(左上到右下)依次预测。第一个token只覆盖图像左上角的小patch,语义信息极少。这是LlamaGen、Janus等模型的默认选择。
作为对照基线,2D网格token是当前主流AR图像生成的默认方案。论文通过控制实验表明它在前缀阶段提供弱语义信号,导致搜索效率远低于1D有序token。
测试时搜索 (Test-Time Search)
在推理阶段,由一个'verifier'对多个候选生成进行打分并指导搜索的范式。形式化为 $\hat{x} = \arg\max_x g(x, c)$ s.t. $x_t \in K(p_\theta(\cdot|x_{<t}, c))$,其中 $g$ 是verifier打分函数,$K$ 限制搜索空间。常见算法包括Best-of-N、Beam Search、Lookahead Search。最初在AlphaGo等强化学习中证明有效,在LLM推理中成为关键扩展手段。
本文验证verifier-guided search从LLM推理迁移到AR图像生成时的关键瓶颈:中间状态是否可被verifier可靠评估——而这又取决于token结构。
Verifier (验证器)
为搜索提供目标信号的函数,输入图像(或部分token序列)输出标量分数。常见类型:图像-文本对齐(CLIPScore、ImageReward、HPSv2、PickScore、CycleReward)、图像-图像对齐(DreamSim)、图像质量(LAION美学分数)、规则式(GroundedSAM检测物体位置/数量)。这些verifier在本文中作为'评测目标'被组合使用。
本文系统比较了8种verifier,发现不同verifier在不同维度有专长,ensemble融合能取得最稳健的总体表现。这对实际部署搜索系统至关重要。
研究动机
近年来测试时搜索在LLM推理(math/reasoning)和扩散模型(Ma et al. 2025)中被证明能显著提升生成质量,但AR图像生成领域的TTS研究相对滞后。一个关键瓶颈在于:传统2D网格tokenizer按raster-scan顺序生成,首个token只覆盖图像左上角的小patch,所携带的语义信息极其有限——对一个像'一盆植物和一个甜甜圈'这样的prompt,搜索早期阶段看到的只有一小块墙或背景,verifier几乎无法据此判断候选的好坏,导致beam search等需要中间信号的方法严重失效。已有的TTS-VAR、ScalingAR、GridAR等工作主要针对特定token结构设计搜索策略,但忽视了'什么样的token结构才能从搜索中受益'这一根本问题。换言之,token结构定义了整个搜索空间和中间状态的可评估性,却很少被作为一等变量来研究。
本文的目标是本文的核心目标是系统研究token结构(特别是1D有序token vs 2D网格token)对AR图像生成测试时扩展行为的影响,并回答三个具体问题:(1) 哪种token结构更受益于搜索,能带来更好的TTS scaling?(2) 如果去掉AR先验,纯靠verifier在token空间搜索,能不能直接生成图像?(3) 不同搜索算法(Best-of-N、Beam、Lookahead)、不同verifier、不同AR先验强度如何与token结构交互?论文最终给出的答案是:1D有序令牌的coarse-to-fine结构使中间状态携带可评估的全局语义,从而让beam search等结构化搜索方法异常高效,甚至在uniform prior下也能进行text-to-image生成。
与已有工作不同的是,本文的独特切入角度是把token结构从'表征设计'重新定位为'搜索的接口设计'。与TTS-VAR、ScalingAR、GridAR等'设计更好的搜索算法'路线不同,本文反过来问:什么性质的token能让搜索算法work?作者通过控制实验(同数据、同架构、同训练量的FlexTok vs 2D grid tok)分离出'结构'这一个变量,避免混淆模型容量和数据差异。同时,他们不只验证一个已训练好的AR模型,而是构造了一个完整的SoTo(Search-over-Tokens)框架,横跨搜索算法×verifier×AR先验的三维设计空间,并通过理论分析证明1D有序token因nested dropout训练最小化中间重建误差,从而为搜索gap提供更紧的边界。
核心方法
方法的核心思路可以分两层来看。直觉上,1D有序token(如FlexTok)按'先全局语义、后细节纹理'的顺序生成图像——第一个token就要重建整张图的粗略轮廓,到第4个token已经能看出'盆栽植物和甜甜圈'的整体语义,而第32个token才开始描绘纹理。这意味着在搜索的每一步,verifier都能基于已有前缀看到一个'语义完整的中间图像',从而给出有意义的打分。相比之下,2D网格token的前几个patch只是图像角落的像素,没有可读语义。基于这一观察,论文提出SoTo(Search-over-Tokens)框架,把测试时搜索分解为三个可独立设计的组件:(A) 搜索算法(Best-of-N、Beam、Lookahead);(B) Verifier(图像-文本/图像-图像/图像质量三大类共8种);(C) AR先验(条件→无条件→uniform,从强到弱)。技术路线是:先用理论分析证明1D有序token的搜索gap $\Delta \leq 2L\epsilon_{t_0} + \eta_{t_0}$ 由中间重建误差 $\epsilon_t$ 控制(L是verifier的Lipschitz常数),再用大量控制实验(FlexTok-3.4B配相同训练的2D grid tok基线;Janus-1.3B;Semanticist;Infinity-2B)在COCO、GenEval、DreamBench++三个benchmark上验证这一结论,并横向比较搜索算法、verifier、先验强度的影响。
核心创新点是把'中间状态的可验证性'作为token结构设计的评价标准,并实证验证1D有序结构在此维度上完胜2D网格结构。具体来说,与已有方法相比有三个本质区别:(1) 视角差异——之前工作(如TTS-VAR、ScalingAR)都在研究'如何为已有token结构设计更好的搜索算法',本文反过来研究'什么样的token结构能让搜索有效';(2) 极端化验证——首次展示了在完全不训练AR模型的情况下,仅凭CLIP/ImageReward verifier和uniform token prior,就能通过beam search在FlexTok的token空间生成语义连贯的图像(GenEval上single-object达到79%、two-object达到32%);(3) 零样本多模态控制——使用DreamSim verifier对仅在文本-图像对上训练的FlexTok模型进行搜索,使其能在推理时根据参考图像控制生成内容(DINO-I提升+18.4),而无需额外的微调。这些都是2D grid模型(如Janus仅+5.9 DINO-I)做不到的。
方法步骤详情
SoTo框架的实施分四步。第一步:定义token序列 $x=(x_1, ..., x_T)$,$x_t \in \mathcal{V}$,其中 $\mathcal{V}$ 是FlexTok的64K token词表。detokenizer Dec(·) 将token前缀映射到像素空间,verifier $g(x, c) = g_{\text{img}}(\text{Dec}(x), c)$。第二步:选择搜索算法。Best-of-N从AR先验采样N个完整序列,挑verifier最高者;Beam Search在token序列上进行稀疏验证,每隔 $s$ 步(FlexTok采用exponential schedule $t \in \{2^0, 2^1, ..., 2^8\}$ 共9步)解码当前前缀并打分,保留top-k;Lookahead Search在Beam基础上对每个候选额外rollout $L$ 步再验证(适合2D grid)。第三步:选择verifier。按需在CLIPScore、ImageReward、HPSv2、PickScore、CycleReward、Likelihood、Aesthetic Score、DreamSim、GroundedSAM以及它们的rank-based ensemble之间挑选。第四步:选择AR先验强度。条件先验最强(标准text-conditioned FlexTok);无条件先验使用空CFG token;uniform先验完全不依赖AR模型,每步从vocabulary中均匀采样10%。每条beam的候选在每步从 $p(x_t|x_{<t})$ 采样 $M=10$ 个token,加上beam width $k=5$ 形成 $k \cdot M = 50$ 个候选,再用verifier选top- $k$。NFE(Number of Function Evaluations)被用作推理算力的统一度量。
技术新颖性
技术新颖性主要体现在四个层面。第一,理论层面:在Pearl的启发式误差框架下,证明搜索gap $\Delta$ 与tokenizer的中间重建误差 $\epsilon_t$ 通过Lipschitz常数 $L$ 直接挂钩——$\Delta \leq 2L\epsilon_{t_0} + \eta_{t_0}$。由于FlexTok的nested dropout训练 $L_{\text{nested}} = \mathbb{E}_t[\|x_{1:t} - x\|^2_2]$ 显式最小化中间重建误差,其$\epsilon_t$随$t$迅速衰减(接近PCA-like分解 $\epsilon_t = \sqrt{\sum_{s=t+1}^{P}\lambda_s^2}$),而2D grid只在 $t=T$ 才有重建约束,中间步骤无结构保证。第二,方法层面:把'coarse-to-fine latent space + verifier-guided search'组合成一个不依赖AR模型的训练-free生成范式,这是首次系统地在图像生成领域验证这一可能性(并发工作Beyer et al. 2025也观察到类似现象,但本文框架更完整)。第三,实验层面:构造了FlexTok的1D vs 2D控制实验(同数据、同架构、同训练量),并在Janus-1.3B、Semanticist、Infinity-2B等多个独立tokenizer/generation paradigm上交叉验证,发现'结构决定可搜索性'这一结论具有跨架构泛化性。第四,应用层面:展示了用image-image verifier做zero-shot multimodal control(FlexTok在DreamBench++上DINO-I从32.5跃升至50.9,+18.4),而无需任何finetuning。
实验结果
论文的核心实验分为五大块,每块都揭示了1D有序token相比2D grid的结构优势。第一,TTS扩展行为(图6):在COCO 300-image子集上,FlexTok(beam search)从baseline的CLIPScore约0.80扩展到约0.94,呈现陡峭上升的scaling曲线;而2D grid tok在beam search下基本不增长(仍约0.82)。两种token在Best-of-N和Lookahead搜索下表现接近,但在beam搜索下分化最严重——这说明只有当中间状态携带可验证的全局语义时,beam search才有效。第二,与Janus对比(图7):无搜索时Janus略胜(高约0.01 CLIPScore),但FlexTok在beam搜索下扩展迅速,在约10^3-10^4 NFE预算内即反超。第三,模型尺寸Pareto前沿(图8):530M FlexTok + 充足搜索能击败3.4B FlexTok(无搜索)。最优模型尺寸与算力呈幂律关系 $y = 4.5 \times 10^3 \cdot x^{0.44}$——即推理算力每增加一个数量级,最优参数量约提升1.7倍。第四,DreamBench++零样本控制(表1):FlexTok + Beam Search使DINO-I从32.5提升至50.9(+18.4),CLIP-I从68.1提升至76.5(+8.4),而Janus即使配合Lookahead Search仅提升+5.9 DINO-I、+2.9 CLIP-I。第五,纯搜索生成(表2):使用uniform prior的beam search就能达到GenEval single-object 79%、two-object 32%;unconditional AR prior进一步提升到85%/33%;conditional AR prior + 搜索达到100%/81%(无搜索条件AR仅97%/48%)。第六,verifier横评(图11):8种verifier在GenEval 6个类别上各有专长,ImageReward在Overall上得分67%最高,单类别排名上HPSv2在Counting上74%最佳、GroundedSAM在Position上31%最佳;rank-based ensemble取得最佳平均排名(2.12),在Single Obj、Colors、Color Attr、Counting上均逼近oracle。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 文本到图像生成(COCO Karpathy 300-image子集) | CLIPScore (%) | FlexTok-3.4B + Beam Search: 90.04(基线80.39,提升+9.65);2D grid + Beam: 81.59(基线79.06,提升+2.53);方差:±0.50(1K-image测试) | FlexTok-3.4B无搜索:80.39;2D grid无搜索:79.06 | FlexTok相比2D grid在beam搜索下多提升+7.12个百分点,差异显著 |
| 组合式文本到图像(GenEval) | Overall Accuracy (%) | FlexTok + Beam + Ensemble verifier: 67%(与oracle 76%差距9个百分点);ImageReward verifier: 67%;Conditional AR + Beam Search (single/two object): 100/81 | FlexTok Base(无搜索): 57%;Conditional AR无搜索 (single/two object): 97/48 | 整体准确率提升+10个百分点;two-object准确率从48%跃升至81% |
| 概念保持生成(DreamBench++) | DINO-I / CLIP-I / CLIP-T | FlexTok + Beam Search: 50.9 (+18.4) / 76.5 (+8.4) / 33.1 (-1.0) | FlexTok Base: 32.5 / 68.1 / 34.1 | FlexTok在概念保持上大幅提升+18.4 DINO-I,远超Janus的+5.9 |
| 概念保持生成(DreamBench++) | DINO-I / CLIP-I / CLIP-T | Janus + Lookahead Search: 40.7 (+5.9) / 71.9 (+2.9) / 35.6 (+0.1) | Janus Base: 34.8 / 69.0 / 35.5 | Janus即使配Lookahead Search也仅+5.9 DINO-I,远低于FlexTok |
| 纯搜索生成(GenEval subset, 180 prompts) | Single Object / Two Object Accuracy (%) | Uniform prior + Beam Search: 79/32;Unconditional AR + Beam: 85/33;Conditional AR + Beam: 100/81 | Conditional AR无搜索: 97/48 | 无需AR模型,uniform prior即可达79% single-object,验证training-free生成 |
| 跨范式比较(COCO, 1K images) | CLIPScore (%) | FlexTok-1.3B + Beam Search: 90.0 (+9.6);Infinity-2B + Beam Search: 88.1 (+6.2) | FlexTok-1.3B无搜索: 80.4;Infinity-2B无搜索: 81.9;Janus-1.3B无搜索: 82.3 | FlexTok的beam搜索增益(+9.6)高于Infinity的scale-wise结构(+6.2)高于Janus的2D grid (+5.3) |
局限与改进
作者明确承认的局限性:(1) 搜索效率受detokenizer瓶颈——FlexTok使用flow-based detokenizer需要多次去噪,在搜索中需反复detokenize,导致wall-clock时间主要由detokenization而非AR generation或verification主导(如图14:beam搜索9步时detokenizer占绝大部分);(2) 早期token前缀的重建质量可能退化,影响verifier判断可靠性;(3) 实验只在相对小规模模型(最大3.4B)上验证,更大模型是否仍保留此scaling优势未知;(4) 分析局限于少数tokenizer(FlexTok、Semanticist、Janus、Infinity),未覆盖所有架构(如纯扩散、MaskGIT)。我自己观察到的额外局限:(5) verifier hacking风险——附录G明确指出,当搜索算力增大时,CLIP/GroundedSAM优化会让美学分数下降,Optimizing Aesthetic Score会让GenEval Accuracy下降,说明任何verifier都可能过拟合;(6) prior bottleneck——uniform prior搜索下模型无法生成prompt中提到的'wine'这类细节概念(图10),证明搜索不能弥补AR先验的盲区;(7) 论文的controlled comparison在300张COCO图上完成,虽然做了1K-image方差分析(差异±0.50),但GenEval 553 prompts的全量实验结果未充分报告方差;(8) FlexTok 2D变体的训练量虽与FlexTok 1D匹配,但实际使用中FlexTok本身的detokenizer设计可能就比VQGAN的detokenizer更适合部分解码,这是潜在混淆变量。
独立分析的弱点
独立分析可以指出三个关键弱点及改进方向。第一个弱点:detokenizer计算瓶颈严重。Wall-clock分析(图14)显示beam搜索的detokenization成本远高于AR generation本身,这意味着NFE虽然公平但实际部署时延迟很高。改进方向:(a) 训练one-step detokenizer(替代flow-based的多次去噪);(b) 设计adaptive verification schedule——根据verifier饱和度提前停止,避免无效搜索;(c) 缓存detokenized中间表示的相似结果。第二个弱点:verifier hacking无法彻底消除。实验显示优化单一verifier会产生明显的'盲区利用'(如优化Aesthetic让GenEval Accuracy下降)。改进方向:(a) 训练更鲁棒的reward model,特别是能区分'高质量图像'与'符合prompt的图像';(b) 在ensemble中引入正则化约束,避免单个verifier分数过度主导;(c) 使用process-based verifier对每步token序列提供细粒度反馈,而非最终scalar。第三个弱点:prior bottleneck——当AR先验完全缺失(uniform prior)时,即使有verifier也无法生成prompt中的精细语义元素(图10的'wine'案例)。改进方向:(a) 设计hybrid prior——uniform + 轻量条件化(如类别标签);(b) 在训练free设置中引入预训练的vision-language model的粗粒度先验;(c) 设计multi-stage搜索:第一阶段用弱先验建立global layout,第二阶段切换到强prior精修。
未来方向
作者在结论部分明确提出的未来方向:(1) 设计专门利用coarse-to-fine结构的搜索算法(如adaptive token position selection、learned search policies、verifier-aware branching),而非通用beam search;(2) 开发自适应算力分配策略(early stopping based on verifier saturation);(3) 更高效的detokenizer(one-step decoding)或adaptive decoding schedule;(4) 研究tokenizer设计如何进一步提升中间重建保真度。基于实验结果,我还看到几个有前景的延伸方向:(a) 把ordered token结构扩展到视频或多模态生成——视频token天然有时间维度,可能更适合coarse-to-fine ordering;(b) 用self-play或RL在ordered token空间训练专门的search policy,进一步拉开与vanilla beam search的差距;(c) 把训练free生成范式推到text-to-3D(NeRF/Gaussian Splatting)或text-to-motion,看1D ordering是否在其他模态同样有效;(d) 在RL post-training(如Self-Tok 2025已观察到ordered tokens的RL增益更大)中结合search,研究RL与search的互补性;(e) 开发更细粒度的process-based verifier(如dense per-token reward)替代当前的global scalar feedback。
复现评估
复现可行性整体较高但有挑战。代码、模型权重和交互式可视化均已在 https://soto.epfl.ch 开源。核心实验依赖FlexTok预训练模型(来自Bachmann et al. 2025)和Janus-1.3B(Wu et al. 2024a),Semanticist和Infinity也有公开checkpoint。数据方面,COCO Karpathy和GenEval都是标准benchmark,Dreambench++公开可下载。主要复现难点:(1) 算力要求——paper中提到使用Lambda academic grant、ETH Swiss AI Initiative和CSCS Alps超算,单次beam搜索在300-image COCO子集上需数小时GH200时间,lookahead search(图14显示200秒/张图像)尤其昂贵;(2) verifier依赖多——CLIP、ImageReward、HPSv2、PickScore、CycleReward、DreamSim、Aesthetic Score、GroundedSAM每个都需要单独加载,加起来约占20GB显存;(3) FlexTok本身的flow-based detokenizer需要复现其multi-step denoising;(4) 1D vs 2D controlled comparison需要从FlexTok repo获取匹配的2D grid变体checkpoint,这部分训练在FlexTok原论文已完成;(5) variance analysis(表6)显示5次随机子集标准差仅±0.50,结果较稳定。总体而言,对有GPU集群的研究组来说中等难度,核心创新点(1D ordered token的search优势)可通过小规模实验验证;但要复现全部图(如所有verifier对比)则需要较多计算资源。
论文图表
(a) 展示给定prompt 'a potted plant and a donut',1D ordered token (FlexTok) 逐token生成:token 2026 (位置1) → 'potted plant' (约8) → 'A potted plant' (约16) → 'A potted plant and an object' → 'a round object' → 'and' → 'a donut' (到256)。中间的detokenized图像展示从粗到细的渐进过程。2D grid token则生成Top-left区域的局部patch,无明显语义。(b) FlexTok在beam search下的test-time scaling曲线显著优于2D grid token的best-of-N基线。
这是论文的核心motivation图,直观展示了1D vs 2D token结构在'中间状态可解释性'上的本质差异,是理解全文的入口。