Flow4R: Unifying 4D Reconstruction and Tracking with Scene Flow¶
会议: ECCV2026
论文: ECCV 官方页面
项目: https://shenhanqian.github.io/flow4r
领域: 几何匹配与定位
关键词: 场景流、动态场景重建、三维点跟踪、相机自运动、位姿权重
一句话总结¶
Flow4R 用共享双视图 Transformer 预测局部几何与跨帧点位置,再通过可学习权重的几何求解分离相机运动和物体运动,在无需全局后优化的条件下统一 4D 重建与跟踪,并在 Point Odyssey 上取得 71.1 的全点跟踪 APD 和 81.00 的重建 APD。
研究背景与动机¶
从视频恢复动态三维世界,难点不只是给每帧估计深度,而是同时回答“这个点在哪里”“相机怎样移动”“物体自身怎样移动”。DUSt3R 等前馈重建方法擅长恢复点图,但其共享参考系的设计源于静态场景;MonST3R 将这一路线扩展到动态视频后,还需要后优化组织多帧几何。另一类跟踪方法通过增加不同时间或坐标系的预测头表达运动,容易让模型结构与特定输出任务绑定。
更根本的问题是,“静止”取决于选定的参考物。站在地面看扶梯和站在扶梯上看地面,对相机自运动与物体运动的解释并不相同。因此,先指定静态背景、再分别回归相机和对象运动,并不是最灵活的建模顺序。与此同时,真实动态场景的三维流标注远少于静态深度数据,单独训练一个依赖完整动态监督的系统也很困难。
本文先学习两帧间能直接观察到的相对几何与运动,再用可靠像素决定参考框架,从同一组预测中导出相机位姿、世界坐标轨迹和光流。核心 idea:以场景流连接局部几何与跨帧对应,用可学习的位姿权重决定怎样解释其中的刚性运动,并联合利用静态与动态数据训练这个统一表示。
方法详解¶
整体框架¶
输入是视角、时间都可以不同的两张 RGB 图像;两条共享参数的路径分别输出与各自像素对齐的三维属性。流程依次是“对称属性预测”“位姿权重求解”,处理视频时再使用“锚帧尺度对齐”,把独立图像对的结果放入一致参考系。
概念上的属性是局部点位置 \(P\)、场景流 \(F\)、位姿权重 \(W\) 和置信度 \(C\)。实际网络用跨视角、跨时刻点位置 \(P_{vt}\) 代替直接输出 \(F\);这不是更换任务,而是为同一个运动关系选择更易学习的回归变量。两幅图各有一套属性,因此天然支持双向跟踪。
%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
A["图像对或锚帧图像对"] --> B["对称属性预测"]
B --> C["位姿权重求解"]
C --> D["锚帧尺度对齐"]
D --> E["一致参考系重建<br/>与三维轨迹"]
B --> F["投影得到光流"]
G["静态与动态标注"] -.->|仅训练:几何与运动监督| B
G -.->|仅训练:位姿监督| C
图中的虚线只表示训练监督,并不是推理时需要额外输入深度、真实位姿或运动标注。单个图像对不需要执行序列尺度对齐;光流也可直接从属性投影得到,不必先消除相机运动。
关键设计¶
1. 对称属性预测:把运动端点作为实际回归目标
两幅图分别通过共享的 ViT 编码器、带跨视图信息交换的解码器和预测头。与为不同时间单独设置预测头的做法不同,两条路径的编码器、解码器、头全部共享参数;交换输入次序仍沿用同一套语义,也不必像 DUSt3R 那样手工构造对称图像对。高分辨率训练阶段采用 DPT 头,将特征恢复为稠密像素对齐输出。
其中 \(P\) 是源图像像素在源相机局部空间中的位置,\(P_{vt}\) 是同一物理点在目标时间、目标相机空间中的位置。“换视角”和“换时间”同时发生,因此它保留了观察者运动与对象运动的总效果,而不是预先把所有前景排除。原文清楚给出的关系为:
直接回归 \(F\) 再加回 \(P\),与直接回归 \(P_{vt}\) 在表示能力上等价,但不保证学习难度相同。论文表 3 发现后者在六个评测列上都更好;作者的解释是,最终评测关注绝对点位置,直接监督运动端点减少了从中间量构造位置时的误差累积。这是经验解释,不是等价参数化必然具有某种优化优势的证明。
这里“参考系无关”的含义应限制为:预测相对运动时不必预先选定哪个场景物体作为静态参考,并不是三维坐标数值对任意坐标变换都保持不变。\(P\) 与 \(P_{vt}\) 仍有明确的相机坐标约定。
2. 位姿权重求解:让可靠对应决定相机运动
总相对运动里混有物体运动,不能对所有对应点等权拟合相机位姿。Flow4R 为每个像素预测归一化权重 \(W\),让几何求解器以加权最小二乘拟合把 \(P\) 对齐到 \(P_{vt}\) 的刚性变换。该变换提供相机运动解释;只改变视角、不改变时间的点位置记为 \(P_v\),据此得到刚性流,剩余部分则描述非刚性运动。要得到源参考系下的运动轨迹,再用估计位姿的逆变换把 \(P_{vt}\) 带回源坐标系。
因此 \(W\) 不是普通的前景分割图。静止的前景也可以帮助定位,而动态、遥远、反光、缺纹理或被遮挡的像素都可能不适合位姿估计。另一个输出 \(C\) 用于表示三维预测的置信度并加权几何与运动回归,不应把它与“是否能支持相机位姿”的 \(W\) 混为一谈。原文规定 \(W\) 的像素权重和为 1,\(C\) 大于 1。
关键是如何训练这个没有理想像素标签的权重图。模型经可微位姿求解器得到变换,将点图变换后与真实位姿、真实几何所给出的结果比较;这条损失对 \(P\) 和 \(P_{vt}\) 停止梯度,只优化 \(W\)。模型于是学习“选哪些点才能把相机运动算对”,而不是照搬语义前景掩码。训练仍依赖真实深度与位姿,这里的自监督指无需直接提供 \(W\) 的标签,不等于全流程无标注。
推理时可以调整甚至覆盖 \(W\),改变用于解释运动的参考区域,而不必重新训练另一组位姿回归头。相机内参也有几何入口:用局部点图与像素位置的重投影关系求焦距,假设两个轴焦距相同;将源点与运动端点投影到图像后作差即可得到光流。论文没有要求所有下游任务都附加独立神经网络头。
3. 锚帧尺度对齐:用同一帧校准独立图像对
处理视频时,将首帧固定为锚帧,分别组成 \((I_0,I_1)\)、\((I_0,I_2)\)、\((I_0,I_3)\) 等图像对,避免仅依赖相邻帧逐步传播轨迹。由于每次预测都包含锚帧本身的局部点图,可以比较同一个对象在不同配对中的预测尺度,而不是在两个无共同参照的点云间猜测尺度。
论文取第 \(n\) 对中锚帧点图的平均欧氏范数作为 \(s_n\),以第一对为尺度参照,把后续点图和场景流乘以 \(s_1/s_n\)。这提供跨配对的尺度一致性,但不意味着恢复了无需外部对齐的绝对米制尺度。它也不是全局束调整:没有在整段视频上联合优化所有位姿与点,序列质量仍受单对预测和锚帧可见范围影响。
一个完整示例¶
设相机绕过一台正在跳跃的机器人,取 4 帧视频,并以第 0 帧为锚帧。模型处理 3 个锚帧图像对;对第 0 帧的某个机器人表面像素,既输出它的当前位置,也输出其在每个目标时间、目标相机空间中的位置,因此此时的总运动包含相机绕行与机器人跳跃。
位姿权重使墙面、地面等可靠背景主导刚性求解,而压低跳跃区域的贡献。把机器人运动端点通过位姿逆变换带回锚帧坐标系后,才得到剔除相机自运动影响的机器人轨迹;背景点应相对稳定。最后利用 3 次预测中共同的锚帧点图统一尺度,才能将各时刻轨迹一起显示。这个例子用于解释数据流,不是论文额外报告的定量实验。
损失函数 / 训练策略¶
训练首先分别用真实与预测点图各自的平均欧氏范数做尺度归一化,两条对称路径使用相同监督。点位置损失使用真实深度与内参反投影得到的局部点图,并以置信度加权回归和对数置信度项约束预测;三维运动损失在有场景流或三维轨迹标签的位置监督运动端点,两者共享 \(C\)。
只有光流或二维轨迹标签时,将运动端点投影到图像平面后进行二维监督,这一项不使用三维置信度加权。这样可把稠密但常限于相邻帧的光流,以及稀疏但跨越更长时间的轨迹纳入同一训练体系;它们提供的监督维度并不完全相同,不能都当作完整的三维运动真值。
静态数据则提供廉价且丰富的刚性运动监督:真实深度加相对相机位姿就能确定目标视角下的点位置。对动态数据,这个刚性目标只适合静态区域,因此使用停止梯度后的 \(W\) 乘以像素总数来加权,压低运动物体贡献;对静态数据,权重直接取 1。停止这一路到 \(W\) 的梯度,避免权重仅为逃避刚性回归误差而退化,与前述只更新 \(W\) 的位姿损失形成明确分工。
原文缓存中位姿最小二乘、部分损失和总损失公式的运算符有缺失,因此这里不猜补精确公式与系数对应关系。上述训练机制来自紧邻公式的完整文字说明;可确认训练同时包含点位置、三维运动、二维运动、位姿权重与刚性运动监督。
模型从 CroCo 初始化,而非直接微调 DUSt3R 或 MonST3R。先以分辨率 224、线性头训练 100 epochs,每轮采样 900K 图像对;再以分辨率 512、随机宽高比和 DPT 头训练 100 epochs,每轮采样 84K 对。视频配对在同场景的 50 帧邻域内抽取,因此不能把训练描述成任意长度视频的联合处理。
优化器为 Adam,两个阶段分别 warmup 10 和 20 epochs,学习率升至 1e-4 后余弦衰减到 1e-6;梯度范数上限为 10。总 batch size 分别为 256 和 64,使用 8 张 NVIDIA A100/H100,完整训练约 4 天。训练混合真实与合成、静态与动态数据,包括提供场景流的 Virtual KITTI 2、提供光流的 Spring,以及提供三维轨迹的 PointOdyssey 等。
实验关键数据¶
主实验¶
WorldTrack 在世界坐标中评测轨迹,因此不仅考查二维对应,也共同检验几何、位姿和运动。APD3D 对齐预测与真值后,计算误差小于 0.1、0.3、0.5、1.0 m 的点比例,并对阈值与前 64 帧汇总,数值越高越好。重建还报告端点误差 EPE,越低越好;表 1 使用全局中位数对齐,表 2 使用全局中位数尺度校正,不能解读为无对齐的绝对尺度评测。
下表选取原文表 1 中的三种方法;ADT 为 Aria Digital Twin,DR 为 Dynamic Replica,PO 为 Point Odyssey,PS 为 Panoptic Studio。APD 按原文百分数尺度列出,参数量单位为十亿。
| 方法 | 全点 ADT | 全点 DR | 全点 PO | 全点 PS | 动态 ADT | 动态 DR | 动态 PO | 参数量 |
|---|---|---|---|---|---|---|---|---|
| MonST3R | 74.4 | 58.1 | 33.5 | 51.3 | 67.9 | 51.9 | 39.4 | 0.7 |
| St4RTrack | 76.0 | 73.7 | 68.0 | 69.7 | 75.3 | 68.1 | 68.7 | 0.7 |
| Flow4R | 78.6 | 78.5 | 71.1 | 64.3 | 70.9 | 77.2 | 72.9 | 0.4 |
原文表 2 的重建结果如下;GA 表示全局对齐后优化,而其余两行是前馈方法。
| 方法 | PO APD ↑ | PO EPE ↓ | TUM-Dynamics APD ↑ | TUM-Dynamics EPE ↓ |
|---|---|---|---|---|
| MonST3R+GA | 72.31 | 0.263 | 63.87 | 0.343 |
| St4RTrack | 78.73 | 0.205 | 83.42 | 0.185 |
| Flow4R | 81.00 | 0.182 | 79.87 | 0.202 |
相对 St4RTrack,Flow4R 的 PO 重建 APD 增加 2.27 个百分点,EPE 降低 0.023;但 TUM-Dynamics APD 低 3.55 个百分点。因此结论应是多个数据集上有竞争力,而不是动态场景重建全面领先。
消融实验¶
原文表 3 只训练完整轮数的一半,以比较运动表示;全部指标都是全点 APD3D,不能把这些值与完整训练主表的差异当作单一模块收益。第一列是直接预测变量,第二列是监督目标;第二行先从预测流构造端点,再监督端点。
| 直接预测 | 监督目标 | 跟踪 ADT | 跟踪 DR | 跟踪 PO | 跟踪 PS | 重建 PO | 重建 TUM |
|---|---|---|---|---|---|---|---|
| \(F\) | \(\bar F\) | 78.03 | 73.26 | 60.23 | 55.80 | 69.36 | 79.78 |
| \(F\) | \(\bar P_{vt}\) | 77.72 | 76.41 | 61.21 | 63.69 | 66.29 | 80.05 |
| \(P_{vt}\) | \(\bar P_{vt}\) | 78.50 | 78.48 | 67.93 | 67.17 | 77.20 | 80.34 |
保持端点监督不变,从预测 \(F\) 改为预测 \(P_{vt}\) 后,PO 跟踪由 61.21 提升到 67.93,PO 重建由 66.29 提升到 77.20。这比只强调“统一场景流”更具体地说明了实际实现为何选择端点回归。
关键发现¶
- 完整模型在 DR 动态点跟踪上为 77.2,St4RTrack 为 68.1,相差 9.1 个百分点;但 ADT 动态点上是 70.9 对 75.3,说明收益随场景而变。
- 表 4 在 RTX PRO 6000 上测得 Flow4R 为 26.8 pairs/s、3152 MB,St4RTrack 为 27.9 pairs/s、6711 MB。显存减少超过 50%,吞吐量则略低,不能称为速度全面提升。
- 权重图可视化显示模型压低动态人物、机器人及部分无纹理区域的贡献,支持其“定位可靠性”解释;正文消融主要检验运动表示,尚未分别量化所有损失项与权重学习策略的贡献。
亮点与洞察¶
- 统一的是几何关系,而不是不断堆叠任务头。同一组点位置与运动端点既可通过投影产生光流,也可通过刚性求解产生位姿,让输出之间存在可解释的联系。
- 让可微求解器反过来训练像素可靠性,比把语义前景当成运动掩码更贴合任务。可迁移的思路是根据下游几何解是否正确学习权重,而不是要求人为穷举所有不可靠像素类型。
- 保留两幅图各自的局部几何不仅带来双向输出,还留下了跨配对尺度校准的共同参照。预测哪些中间量,会直接影响序列推理是否需要额外优化。
局限与展望¶
- 作者指出,高质量真实三维场景流数据仍稀缺;静态数据与合成运动监督不能消除所有真实动态场景的域差异。
- 当前仍是双视图模型,锚帧配对和尺度校准不等于多视图联合建模。作者建议引入类似 VGGT 的多视图注意力,以改善时间一致性与重建质量。
- 严格资源约束下的实时在线跟踪仍是作者承认的开放问题。这里的 pairs/s 是指定 GPU 的图像对吞吐量,不是完整在线系统帧率。
- 从评测范围看,前 64 帧与全局对齐协议不能充分证明任意长序列、绝对尺度或长期遮挡恢复能力;可进一步测试锚帧失去重叠时的失败率与自适应换锚策略,这是笔记提出的后续方向。
相关工作与启发¶
- 对比 DUSt3R / MASt3R:保留双视图 Transformer 的几何学习路线,但将局部几何和跨时间对应显式组织起来,并共享两路解码器及预测头;目标不再局限于静态多视图点图。
- 对比 MonST3R:后者通过局部时间连接与后优化组织动态场景,Flow4R 使用锚帧连接、位姿求解和尺度对齐。少了全局后优化,不代表自动获得所有全局一致性约束。
- 对比 St4RTrack / POMATO:Flow4R 强调对称双向属性和两个视图的局部点图;其锚帧序列处理沿用 St4RTrack 的思路。论文对 POMATO 使用成对模型,因为其序列模型的跟踪方向与其他方法的标准锚帧向外跟踪不一致。
- 对比 VGGT:多视图联合注意力提供更强的跨帧信息交互;Flow4R 当前并未实现这一扩展,而是给出了适合与之结合的紧凑几何表示。
评分¶
- 新颖性: 4/5。贡献集中于对称表示、可学习几何求解与静动态联合监督的组织方式,而非提出新的 Transformer 主干。
- 实验充分度: 4/5。覆盖多数据集的跟踪、重建和效率,并检验运动参数化,但各训练部件的独立消融仍有限。
- 写作质量: 4/5。几何解释与可视化清楚;阅读时需特别区分概念上的流预测与实际端点输出,并注意“参考系无关”的适用含义。
- 价值: 4/5。为低显存的统一 4D 感知提供了可复用路线,但真实长序列与在线系统仍需进一步验证。