Helix4D:复杂动态 4D 网格生成 Helix4D: Complex 4D Mesh Generation
将图像生成3D的Trellis2扩展为视频驱动的4D动态网格生成
前置知识
扩散流匹配(Flow Matching)
扩散流匹配是一类生成式建模方法,通过学习从简单噪声分布到目标数据分布的连续常微分方程轨迹来进行采样,比传统 DDPM 更稳定高效。
Trellis2 的三个阶段(稀疏结构、几何、材质)均基于 flow matching,本文将其升级为视频条件 4D 生成仍沿用同一框架。
RoPE 旋转位置编码(Rotary Position Embedding)
RoPE 通过对查询和键向量施加与位置相关的旋转矩阵 $R(\omega_i p)$,使注意力分数只依赖相对位置,从而天然支持长度外推。
Trellis2 的 backbone 采用 3D 空间 RoPE,本文需在不增加维度的情况下把其扩展到 4D 时空编码,是技术核心难点。
O-Voxel 稀疏体素表示
O-Voxel 是一种稀疏近表面体素表示,将 3D 资产编码为活跃体素集合 $F = \{(f_i^{shape}, f_i^{mat}, p_i)\}_{i=1}^{L}$,支持非闭合、薄壁和内表面结构。
O-Voxel 不要求 watertight 网格,因此能表示开口、薄结构和内表面,这是 Helix4D 能处理透明物体和鱼在瓶内等场景的关键。
交叉注意力掩码(Cross-frame Attention Mask)
通过定义二元掩码矩阵 $M \in \{0,1\}^{S \times S}$ 控制每对 token 是否可互相注意,从而在不改变模型架构的前提下实现不同注意力模式(全局/因果/滑动窗口等)。
本文的滑动窗口 + 首帧锚点机制就是通过设计特定的 $M$ 矩阵实现的,使每个帧只能看到邻近 $w$ 帧和首帧。
研究动机
现有视频到 4D 生成方法在处理复杂场景时存在三大明显短板:第一,基于形变的方法(Mesh4D、ActionMesh、Motion 3-to-4 等)以首帧网格为锚点预测 warp 场,根本无法处理后续帧出现新物体、拓扑变化等场景(如图 5 中颜料飞溅到人像、菜篮子从手中脱落);第二,几乎所有现有方法都假设 watertight 网格,因此无法表示透明/半透明物体(如玻璃瓶中的鱼)和内表面结构;第三,端到端的 4D 生成器(SS4D、Sculpt4D、ShapeGen4D)虽然不受拓扑限制,但每帧几何和材质质量有限。优化类方法虽然能生成高质量结果,但需要数小时/资产,无法满足实际应用需求。
本文的目标是论文旨在把当前最强的静态 3D 基础模型 Trellis2 升级为视频驱动的 4D 网格生成器,同时保留其在透明物体、薄壁、内表面等罕见但重要的几何与材质能力。期望在 ActionBench 上超越专门为该场景设计的 ActionMesh,并在作者自建的 52 视频 Helix4DBench(覆盖形态变化、新物体出现、破碎、透明与体积现象)上对 SS4D、ShapeGen4D、Mesh4D、Motion 3-to-4、ActionMesh 五大基线全面领先。
与已有工作不同的是,本文的独特切入角度是:与其从零训练 4D 模型,不如系统性地将 Trellis2 这种已具备强几何/材质先验的 3D 模型"嫁接"到视频条件上。具体而言通过两个关键技术实现:(a) 在不破坏预训练的前提下引入跨帧交互——滑动窗口注意力 + 首帧锚点(由冻结 Trellis2 生成);(b) 提出 4D RoPE 复用预训练中冗余的低频空间频率编码时间,避免引入新的相位干扰预训练权重。这种"嫁接既有基础模型"的范式与 SS4D 等独立训练 4D 模型的方法形成鲜明对比,是把静态 3D 基础模型的能力提升到 4D 的首次系统实践。
核心方法
Helix4D 整体思路是"嫁接而非重建":保留 Trellis2 的三阶段 DiT backbone(稀疏结构 → 几何 → 材质)和 O-Voxel 表示,仅在每个 self-attention 层注入两个时间相关机制。直观上,模型把视频视为帧序列,每帧的 token 序列按帧索引拼接,整体做一次 cross-frame attention:第 $f$ 帧的 token 只能看到局部窗口 $|f-f'| \leq w$ 内的 token,并额外被允许看到首帧的全部 token(首帧由冻结 Trellis2 在零去噪步生成)。同时为保持 RoPE 维度不变,将预训练中冗余的低频空间频率"借走"用于编码时间轴,从而得到 4D RoPE $\mathcal{R}_{\Omega,(p,t)} = \mathcal{R}_{\Omega,p}^{spatial} \oplus \mathcal{R}_{\Omega,t}^{temporal}$。每个 stage 仅微调预训练的 self-attention 层,20K 迭代即可完成训练。
与已有方法相比,核心创新是双重的。第一是"首帧锚点 + 滑动窗口"的注意力机制:全注意力代价太高(单次 4D 重建可达 $10^5$ token),因果注意力丧失未来信息,滑动窗口缺全局锚点会导致首帧几何逐渐漂移,而 SS4D 风格的方案则打破预训练 RoPE 的相对位置性质。本文的 mask 设计为 $M(f,s),(f',s') = 1$ 当 $|f-f'| \leq w$ 或 $f'=0$,既保持 $2.3\times$ 加速(相对全注意力),又通过首帧注入继承 Trellis2 在透明/内表面等罕见 case 的能力。第二是"频率复用"式 4D RoPE:观察到 Trellis2 的空间 RoPE 中 $\alpha \geq 0.4$ 的低频部分贡献很小(替换为 identity 后 CD-3D 仅从 0.0079 升到 0.0181),可重新分配给时间维度。设置 $\alpha = 0.75$(按 $T/N = 1/4$ 的比例分配)实现无参数扩展,保持空间/时间相位分离从而保留相对位置性质。
方法步骤详情
方法分两大模块。模块一:跨帧注意力。输入视频帧 token 集合 $\{(f,s, x_{f,s}, p_{f,s})\}$,$f$ 为帧索引。把首帧 $f=0$ 放序列最前,去噪步设为 0(用真实或冻结 Trellis2 生成的潜变量),其余帧 $f \geq 1$ 用含噪潜变量。self-attention 层用掩码 $M$ 限制可见,$w=5$ 窗口覆盖 $|f-f'| \leq 5$ 局部邻域,首帧全局可见。Flow matching 损失只在 $f \geq 1$ 上计算。模块二:4D RoPE。每个轴 $M=d/6$ 频率对拆为高频与低频块,前者保留空间 RoPE,后者替换为时间 RoPE。4D 旋转为 $\mathcal{R}_{\Omega,(p,t)} = \mathcal{R}_{\Omega,p}^{spatial} \oplus \mathcal{R}_{\Omega,t}^{temporal}$,维度 $d$ 不变。三 stage 一致应用,AdamW 学习率 $2\times 10^{-5}$,batch 32,32 张 A100 训练 20K 迭代。
技术新颖性
技术新颖性主要体现在三方面。其一是首次系统地把 Trellis2 这种 O-Voxel 静态 3D 基础模型升级到 4D 生成,论证了"静态 3D 基础模型可被提升到 4D 而不必从头训练"的可行性,相比 SS4D 等独立训练方法显著节省数据需求。其二是滑动窗口 + 首帧锚点的混合注意力机制设计,通过 frozen Trellis2 生成首帧作为锚点解决了 4D 训练数据中透明/内表面样本稀缺的问题(详见 Tab.3,去掉该组件 ULIP-2 从 0.2881 降到 0.2809)。其三是参数自由的 4D RoPE 频率复用方案:与 SS4D 在 3D RoPE 上叠加 1D 时间 RoPE(破坏相位、引入预训练未见的旋转)相比,本文通过 block-diagonal 分离的空间/时间 RoPE 既保持相对位置性质又不需要新增参数,并通过 $\alpha = 0.75$ 的设计让空间/时间频率按物理尺度比例分配。
实验结果
论文在三个基准全面验证。Helix4DBench(52 个自建视频)本文全面领先:CLIP 0.8723、CLIP-N 0.7569、ULIP-2 0.4600(vs Mesh4D 0.4353,+5.7%)、Uni3D 0.4063(vs 0.3768,+7.8%)、DreamSim 0.2593、FVD 950。用户研究相对 ActionMesh 胜率 89.8%、Motion 3-to-4 80.7%。ActionBench 上 CD-3D 0.051 优于 ActionMesh 0.053(+3.8%)。TexVerse CD-3D 0.045 与 CD-4D 0.097 均最佳,远超 SS4D(0.052/0.103)。消融 Tab.3:去首帧条件 CD-3D 从 0.0464 升到 0.0488,去 4D RoPE 升到 0.0478(视觉出现多余手臂)。Tab.4 表明本文(0.0454)比全注意力(0.0470)更好且仅 $1/2.3$ 时间。Tab.A1 表明 $\alpha \in [0.4, 1.0]$ 区间 CD-3D 稳定在 0.0079。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| Helix4DBench 3D-图像对齐 | ULIP-2 ↑ | 0.4600 | 0.4353 (Mesh4D) | +5.7% |
| Helix4DBench 3D-图像对齐 | Uni3D ↑ | 0.4063 | 0.3768 (Motion 3-to-4) | +7.8% |
| Helix4DBench 视频质量 | FVD ↓ | 950 | 955 (Motion 3-to-4) | 略优 |
| Helix4DBench 感知相似 | DreamSim ↓ | 0.2593 | 0.2697 (Motion 3-to-4) | 略优 |
| ActionBench 几何精度 | CD-3D ↓ | 0.051 | 0.053 (ActionMesh) | +3.8% |
| TexVerse 测试集 | CD-3D ↓ | 0.045 | 0.052 (SS4D) | +13.5% |
| TexVerse 测试集 | CD-4D ↓ | 0.097 | 0.100 (Mesh4D) | 略优 |
| 用户研究(vs ActionMesh) | 胜率 ↑ | 89.8% | 10.2% | 大幅领先 |
局限与改进
作者明确承认三类局限性。第一是网格破洞:Helix4D 继承了 Trellis2 的非 watertight 假设,因此生成网格偶尔出现破洞,无法保证表面闭合。第二是材质色彩偏移:透明物体常被错误地赋予金属质感,这是 Trellis2 材质分支本身的问题。第三是高频细节的时间一致性:输出 mesh 序列时(而非单帧网格),高频几何细节(如飘带、绳子、薄壁)随时间变化会出现闪烁和不连贯。Tab.4 中本文 CD-4D 0.0970 虽优于 SS4D 0.120,但仍弱于 CD-3D 0.0454 的提升幅度,说明 4D 时间一致性仍是相对薄弱环节。从论文整体看,还有一个隐含限制:方法在物体中心视频上训练,对包含相机运动、场景级多物体交互的视频泛化能力未经验证,作者在结论中将此列为未来工作。
独立分析的弱点
独立分析可观察到几个进一步可改进的弱点。其一是训练资源门槛较高:32 张 A100 训练 20K 迭代对学术机构而言门槛偏高,且代码权重未提及开源情况,复现成本较大。其二是对输入视频质量敏感:由于依赖 Wan2.2 生成 16 帧视频作为输入条件,输入视频的相机稳定性、视角变化和帧间一致性会显著影响最终 4D 输出,但论文没有分析视频质量对结果的敏感度。其三是评测偏置:Helix4DBench 由作者用 Wan2.2 从 Trellis2 示例图构造,可能存在评估偏置(与作者 pipeline 更契合),且 ActionBench 主要含简单运动,无法充分展示方法在拓扑变化上的优势(CD-3D 仅领先 ActionMesh 0.002)。其四是首帧锚点的副作用:当首帧本身存在错误时,整段视频都会继承这个错误,缺少 fallback 机制。改进方向可考虑引入相机位姿条件、首帧置信度机制等。
未来方向
作者在结论中明确指出四个未来方向:(1) 扩展到更长视频(当前仅 16 帧),需要解决滑动窗口的复杂度增长和长程一致性问题;(2) 场景级 4D 生成,包含多物体交互和物体间关系;(3) 更强相机运动支持,将视频中的相机轨迹纳入条件;(4) 更准确的物理动力学建模,特别是软体、流体(如火焰、烟雾)等体积现象。基于本文成果还可以延伸的方向包括:(a) 把"嫁接 3D 基础模型到 4D"的范式推广到其他静态 3D 模型(如 Hunyuan3D、Direct3D-S2),探索哪些 3D 架构最容易升级到 4D;(b) 把 4D RoPE 的频率复用思想应用于其他模态扩展,例如 3D → 3D+语义、2D 视频 → 3D+时间的统一位置编码;(c) 把首帧锚点机制扩展为"关键帧锚点",允许用户指定多个参考帧以约束复杂动画序列;(d) 与物理仿真结合做"物理先验 + 几何先验"联合 4D 生成。
复现评估
复现评估方面,论文在数据、训练、评测三个层面都给出了较详细的信息但代码尚未明确开源。数据层:训练集来自 TexVerse-1K 的 ~55K 资产子集,每资产提取 16 帧并转 O-Voxel,分辨率 $1024^3$;TexVerse 32 测试集给出了完整的 GLB 资产路径列表(Tab.A4)。训练层:32 张 A100 GPU 训练 20K 迭代,batch size 32,AdamW 学习率 $2\times 10^{-5}$,仅微调 self-attention 层;超参数 $\alpha=0.75$、窗口 $w=5$、$F=16$ 帧均明确给出。评测层:Helix4DBench 52 视频提供了完整文本 prompt(Tab.A3)和构造流程;用户研究 Tab.A2 给出 21 名参与者、458 次比较统计。主要门槛:(1) Trellis2 预训练权重获取;(2) 32 卡 A100 训练资源;(3) Helix4DBench 视频的端到端复现。难度:中高,需要熟悉 3D 生成、扩散模型和视频处理的交叉背景。
论文图表
图 1 是论文 teaser,分三组展示 Helix4D 的能力。左:帆船场景,精确重建薄壁结构(船帆、绳索、栏杆)和断裂的旗帜等高频细节。中:金属融化场景,几何和纹理保留精细装饰。右:玻璃瓶内游动的鱼,展示透明外壳和内表面重建(剖开网格以显示内部)。最下一行展示生成的动态序列,覆盖拓扑变化、形变、破碎、融化、透明、薄结构六大挑战性场景。
这张图直观展示了论文宣称的核心能力——处理透明、薄壁、内表面、拓扑变化等复杂 4D 场景,是论证动机部分(现有方法难以处理这些 case)最强有力的视觉证据。