PhysiFormer: Learning to Simulate Mechanics in World Space

AK's Picks Embodied AI and Robotics World Models Sim-Platform GenAI Diffusion
2026年06月25日
我们提出了PhysiFormer——一种面向物理合理三维物体运动的扩散型Transformer模型。与在依赖视角的像素空间中运行的视频世界模型不同,PhysiFormer将物体表示为以世界坐标系表达的三维网格(3D mesh)。给定初始顶点位置、顶点速度以及物体材质类型(刚性或弹性),该模型可采样生成未来各顶点的运动轨迹。尽管此前相关神经物理方法往往构建于特定设计的隐空间之上,或显式地施加刚性约束与因果性约束,PhysiFormer却表明:仅需将顶点轨迹预测建模为一个直接在世界坐标系中进行的单一去噪扩散过程,无须任何此类归纳偏置(inductive bias),即可取得优异性能。其概率化建模方式能够刻画所学动力学中的不确定性,从而在给定初始条件下生成多种合理且多样的未来演化结果,使该框架有望应用于存在未观测不确定性的实际场景。为提升计算效率,模型采用时间、空间与物体三重维度解耦的注意力机制,实现了无需显式物体编码的排列不变(permutation-invariant)多物体联合推理。PhysiFormer在逾10万条仿真轨迹上完成训练,不仅能准确生成刚体与弹性体的动力学行为,还可泛化至混合材质场景、未曾见过的真实世界几何结构以及更大规模的物体数量。在轨迹预测精度、刚性保持能力及基于动量的物理一致性等关键指标上,PhysiFormer显著优于自回归基线模型。我们的研究结果表明,坐标空间中的扩散建模是迈向“视角无关、几何感知”世界建模的重要一步,有望服务于机器人学、计算机图形学及物理驱动的设计任务。可视化示例、源代码与预训练模型均已开源,详见:https://yimingc9.github.io/physiformer。
We present PhysiFormer, a diffusion transformer for physically-plausible 3D object motion. Unlike video world models that operate in view-dependent pixel space, PhysiFormer represents objects as 3D meshes expressed in world coordinates. Given the initial vertex positions and velocities, as well as object material type, rigid or elastic, the model samples future vertex trajectories. While related neural physics approaches build on ad-hoc latent spaces or explicitly enforce rigidity and causality, PhysiFormer shows that excellent results can be obtained without any such inductive biases, by casting vertex trajectory prediction as a single denoising diffusion process directly in world coordinates. The probabilistic formulation captures uncertainty in the learned dynamics, enabling diverse plausible futures from initial conditions, making this framework potentially useful for applications with unobserved uncertainty. The model features attention factorised over time, space, and objects for efficiency, enabling permutation-invariant multi-object reasoning without needing explicit object encoding. Trained on over 100k simulated trajectories, PhysiFormer generates rigid and elastic mechanics, and generalises to mixed-material settings, unseen real-world geometries, and larger object counts. It substantially outperforms autoregressive baselines in trajectory accuracy, rigidity preservation, and momentum-based physical consistency. Our results position coordinate-space diffusion as a promising step toward view-invariant, geometry-aware world modelling for robotics, graphics, and physical design. Visualisations, code, and models are available at https://yimingc9.github.io/physiformer.
许愿