浙大Block3D把文生三维从25.7秒压到5秒,几何指标反升

Block3D: Efficient Text-to-3D Generation via Block-Wise Diffusion

Bowen Cui, Weijie Wang, Zeyu Zhang, Yefei He, Mingda Lin, Haoyu Zhao, Yuanyu He, Donny Y. Chen, Feng Chen, Bohan Zhuang

cs.CV

2026-08-20

浙大ZIP Lab把Cube的逐token自回归改成块内并行去噪,再按置信度改错。100个留出物体上,端到端从25.71秒降到4.99秒,Chamfer-L1从0.094降到0.078。

这篇在解决什么

文生 3D 卡在同一对矛盾上。自回归按形状 token 一个一个吐,序列一长就慢,吐出去的码也不能改。扩散或流匹配可以并行,却要在整段表征上反复扫,分辨率一高,每步都更贵。浙大 ZIP Lab 联合莫纳什和阿德莱德,把因果依赖从「单个码」上移到「连续的一块码」,想同时保住几何、压住延迟。

方法

表征沿用冻结的 Cube VQ:每个网格压成 N=1024 个码,码本大小 16384,解码器也冻住。文本条件来自冻结 CLIP ViT-L/14 的 77 个 token。只微调 Cube 那套 23 层 DualStream RoFormer 作为先验。

1024 个位置切成块。默认块长 B=64,一共 16 块,从左到右生成。当前块一开始全是 mask,块内双向注意,已经提交的前缀只做因果注意,未来块不进序列。块内最多 T=4 步,每一步做两件事:把高置信 mask 填成码(M2T),把已经填了但模型更想改的码换掉(T2T)。置信阈值分别是 0.95 和 0.9。每步还有一个确定性配额,保证 4 步之内 mask 清零。分类器无关引导从 3.0 线性降到 0.75,候选用引导 logits 提出,接不接受看条件分支的 softmax,避免 CFG 把置信度本身抬高。

训练时每个样本随机走 mask 流或替换流,残差率在 0.45 到 0.95 之间。先无梯度 rollout 一步,再只对 rollout 之后仍不等于真值的位置做交叉熵。已提交的块训练时也是教师强制,推理时是模型自己生成的前缀,曝光偏差还在,只是被限制在块与块之间。

结果

评测是 TRELLIS-500K 里随机留出的 100 个物体,微调池剔除这 100 个,在 30 万物体上训 3.5 万步。几何采 8192 个表面点。延迟在一张 A100 80GB 上从条件编码计到网格解码。

方法CD-L1↓NC↑F@1%↑CLIP均值耗时
ShapeLLM-Omni0.2290.4900.08919.0836.89s
TRELLIS-text0.2220.4960.09020.4111.65s
Cube0.0940.6320.21923.8725.71s
Block3D0.0780.6680.30923.244.99s

相对单独微调的 Cube,均值耗时从 25.71 秒到 4.99 秒,5.15 倍;中位数和 P90 一起降,标准差从 0.80 收到 0.43。几何三项全过 Cube,CLIPScore 从 23.87 掉到 23.24,文形对齐略退一步。

块长是质量和速度的主旋钮。B=32 要 12.98 秒,F@1% 0.296;B=64 是选定点;B=96 只要 3.62 秒,F@1% 掉到 0.279;B=256 快到 2.15 秒,F@1% 只剩 0.103。去噪步数从 4 加到 20,时间涨到 13.66 秒,F@1% 还略低于 4 步。T2T 把 F@1% 从仅 M2T 的 0.287 拉到 0.309。

为什么重要

这是把语言模型里的 block diffusion 和可编辑离散去噪,接到固定长度 3D 码上的工程组合,不是新的几何表示。对已经在用 Cube 码的人,改先验就能把端到端从半分钟级打到约 5 秒,几何还更好。CLIP 略降说明加速优先保了形状,提示对齐没有一起涨。

默认最多 80 次 batched 前向(含 CFG 则 160 次逻辑分支),相对 Cube 的 1024 次逐 token 前向,调度优势写在纸面上,最终数字以测到的 4.99 秒为准。

局限与存疑

提交过的块不能重开,块间曝光偏差还在。块是按 Cube 的一维码序切的,不是语义零件,和 PartDiffuser 那种按部件去噪不是一类。评测只有 100 个物体、每条 prompt 一次生成,没有多样性统计。Cube 对照共享了冻结组件和训练预算,TRELLIS-text、ShapeLLM-Omni、AR3D-R1 用的是官方推理设置,不在同一微调预算里。CLIP 低于 Cube,文形对齐不是免费附赠。

术语

原文与代码

相关论文

全部论文解读