HyperAIHyperAI

Command Palette

Search for a command to run...

PhysisForcing: 로봇 조작을 위한 물리 강화 세계 시뮬레이터

초록

비디오 생성 모델은 체화된 세계 시뮬레이션을 위한 유망한 패러다임으로 부상했다. 그러나 일반 도메인 비디오 생성기와 로봇 특화 데이터로 미세 조정된 모델 모두 여전히 불연속적인 운동 궤적과 일관되지 않은 로봇-객체 상호작용을 포함하여 물리적으로 타당하지 않은 조작을 생성할 수 있으며, 이는 세계 시뮬레이터로서의 신뢰성을 제한한다. 광범위한 실험을 통해, 우리는 이러한 물리적 불안정성이 주로 두 가지 요인, 즉 움직이는 객체의 변형과 상호작용하는 개체들 간의, 특히 접촉 시의 비현실적인 시공간 상관관계에서 기인함을 발견한다. 이 관찰을 바탕으로, 우리는 픽셀 수준 및 의미 수준 특징의 공동 최적화를 통해 물리 정보 영역에 감독을 집중함으로써 물리적 일관성을 강화하는 확장 가능한 훈련 프레임워크인 PhysisForcing을 제안한다. 이 프레임워크는 기준점 궤적을 이용해 DiT 특징을 감독하는 픽셀 수준 궤적 정렬 손실과, 동결된 비디오 이해 인코더에서 추출한 지역 간 관계와 DiT 특징을 정렬하는 의미 수준 관계 정렬 손실로 구성된다. R-Bench, PAI-Bench, EZS-Bench에서의 광범위한 실험은 PhysisForcing이 강력한 기준 모델들에 비해 체화된 비디오 생성을 일관되게 개선함을 보여주며, Wan2.2-I2V-A14B 및 Cosmos3-Nano 기본 모델의 R-Bench 성능을 각각 22.3%, 9.2% 향상시켰고(기본 미세 조정 대비 7.1%, 3.7% 향상), Cosmos3Nano 변형이 최고 종합 점수를 달성했다. 생성뿐 아니라, WorldArena 행동 계획 프로토콜 하에서 세계 모델로서 폐루프 성공률을 16.0%에서 24.0%로 높이고, 하위 정책 성공을 더욱 개선하여, 물리적으로 정렬된 비디오 모델이 로봇 조작을 위한 더 강력한 표현을 산출함을 시사한다. 코드 및 추가 결과는 https://dagroup-pku.github.io/PhysisForcing.github.io/에서 확인할 수 있다.

One-sentence Summary

Researchers from Peking University and NVIDIA propose PhysisForcing, a training framework that reinforces physical consistency in video world simulators for robotic manipulation by jointly optimizing pixel-level trajectory alignment and semantic-level relational alignment losses, improving generation benchmarks including a +22.3%+22.3\%+22.3% gain on R-Bench and raising the closed-loop manipulation success rate from 16.0%16.0\%16.0% to 24.0%24.0\%24.0%.

Key Contributions

  • PhysisForcing is a training framework that enforces physical consistency in video generation by jointly optimizing a pixel-level trajectory alignment loss and a semantic-level relational alignment loss on interaction-critical regions.
  • Experiments on R-Bench, PAI-Bench, and EZS-Bench show that PhysisForcing consistently improves embodied video generation, with relative gains of 22.3% on Wan2.2-I2V-A14B and 9.2% on Cosmos3-Nano over base models on R-Bench, and the Cosmos3Nano variant achieving the best overall score.
  • When used as a world model in the WorldArena protocol, PhysisForcing raises closed-loop manipulation success rate from 16.0% to 24.0% and improves downstream policy success, demonstrating that physically aligned video generation yields stronger representations for robotic manipulation.

Introduction

Video generation models hold promise as world simulators for embodied intelligence, enabling scalable data generation and policy learning in robotic manipulation. However, they often produce physically implausible dynamics like object penetration or discontinuous motion, especially in contact-rich tasks. Prior approaches either lack sufficient exposure to manipulation data, treat all pixels uniformly during training, or apply single-level and post-hoc physical constraints that miss the hierarchical nature of physical plausibility. The authors introduce PhysisForcing, a region-focused, hierarchical alignment framework that, during training, identifies interaction-critical regions and applies two complementary losses: a pixel-level point-trajectory loss for local motion consistency, and a semantic-level token-similarity alignment loss for correct global object interactions.

Method

The authors propose PhysisForcing, a method that injects physics supervision into video generation through a region-focused hierarchical alignment framework. As shown in the figure below:

The pipeline begins by identifying physics-informative regions where robot-object interactions occur, followed by the application of two complementary training signals: pixel-level trajectory alignment for local motion consistency and semantic-level relational alignment for interaction outcome consistency.

To localize these interaction-rich regions, the authors first extract dense temporal trajectories from the input video VRT×C×H×WV \in \mathbb{R}^{T \times C \times H \times W}VRT×C×H×W using an off-the-shelf point tracker. They compute the motion magnitude aia_iai for each trajectory, noting that large local motion often indicates contact-rich areas. However, to filter out background jitter, they introduce a depth-aware foreground weight rir_iri derived from the estimated depth map D0D_0D0 of the first frame. By combining the motion magnitude with this foreground weight, they calculate a physics-informative score qi=airiq_i = a_i \cdot r_iqi=airi for each trajectory. An adaptive threshold based on the mean score is then applied to generate a trajectory-level motion mask, which is subsequently projected onto each frame to form a spatiotemporal physics mask Mphy\mathbf{M}^{\text{phy}}Mphy. This mask provides spatial guidance for the subsequent physics supervision steps.

For pixel-level physics alignment, the authors enforce per-point trajectory continuity on the manipulator and the manipulated object. They extract the hidden feature Hl\mathbf{H}^lHl from an intermediate block of the Denoising Transformer (DiT) and refine it using a lightweight MLP ϕ()\phi(\cdot)ϕ(). The refined feature is reshaped into frame-wise feature maps F^\hat{\mathbf{F}}F^. Using the first-frame feature as the query Q=F^0\mathbf{Q} = \hat{\mathbf{F}}_0Q=F^0 and the remaining frames as keys Kt=F^t\mathbf{K}_t = \hat{\mathbf{F}}_tKt=F^t, they compute a similarity map sit\mathbf{s}_i^tsit for each query point. By normalizing this map and computing the coordinate expectation, they obtain the predicted point location p^it\hat{\mathbf{p}}_i^tp^it at each frame. These predicted trajectories are then supervised by the reference trajectories extracted from the ground truth video using a masked mean squared error loss Lpixphy\mathcal{L}_{\text{pix}}^{\text{phy}}Lpixphy, restricted to the physics-informative regions identified earlier.

While pixel-level alignment constrains point-wise motion, manipulation plausibility also relies on the relational dynamics between different regions. To capture this, the authors introduce semantic-level physics alignment. They utilize a frozen self-supervised video understanding encoder to extract a target representation Fu\mathbf{F}^uFu from the input video. Simultaneously, they project the hidden features from the same intermediate DiT block into the encoder representation space using another lightweight MLP ψ()\psi(\cdot)ψ() and resize them to match the encoder spatio-temporal token layout F^u\hat{\mathbf{F}}^uF^u. The physics-informative mask is resized to select corresponding spatio-temporal tokens from both representations. They then compute pairwise relational matrices R^\hat{\mathbf{R}}R^ and R\mathbf{R}R for both the DiT-side and encoder-side tokens, capturing the spatio-temporal relations among the selected physics-informative tokens. The semantic-level alignment loss is defined as the L1 distance between these two relational matrices:

Lsemphy=1K2i=1Kj=1KR^(i,j)R(i,j)\mathcal{L}_{\text{sem}}^{\text{phy}} = \frac{1}{K^2} \sum_{i=1}^{K} \sum_{j=1}^{K} \left| \hat{\mathbf{R}}(i, j) - \mathbf{R}(i, j) \right|Lsemphy=K21i=1Kj=1KR^(i,j)R(i,j)

This effectively transfers the relational structure of the encoder into the DiT.

PhysisForcing is applied during the fine-tuning of a pre-trained DiT-based video generation backbone. The overall training objective combines the standard flow matching loss with the pixel-level and semantic-level physics losses:

L=LFM+λpixLpixphy+λsemLsemphy\mathcal{L} = \mathcal{L}_{\text{FM}} + \lambda_{\text{pix}} \mathcal{L}_{\text{pix}}^{\text{phy}} + \lambda_{\text{sem}} \mathcal{L}_{\text{sem}}^{\text{phy}}L=LFM+λpixLpixphy+λsemLsemphy

All auxiliary models, including the point tracker, depth estimator, and video understanding encoder, are utilized exclusively during training and are discarded at inference, ensuring that the method introduces no additional computational cost during video generation.

Experiment

Experiments across three embodied video generation benchmarks show PhysisForcing improves multiple backbones, yielding more physically plausible videos and outperforming strong baselines. The enhanced world model also boosts success rates in policy learning on contact-rich tasks. Ablations verify that the pixel and semantic losses are complementary, interaction-focused supervision drives gains, and a mid-block alignment layer is optimal.

PhysisForcing substantially lifts video generation performance on R-Bench, with PF-Cosmos achieving the highest overall average (63.8) and PF-Wan following closely (62.0), both outperforming the best commercial and open-source baselines. Among backbone models without PhysisForcing, open-source counterparts reach only moderate scores, led by HunyuanVideo 1.5 at 46.0 average. PF-Cosmos attains the top overall average of 63.8, surpassing the strongest commercial model Wan2.6 (60.7) and all open-source models. PhysisForcing boosts the Wan backbone by 22.3% (to 62.0) while the highest-scoring open-source model without it, HunyuanVideo 1.5, reaches just 46.0.

Replacing the video backbone with PhysisForcing in Fast-WAM raises the average success rate from 68.2% to 72.8% across six RoboTwin 2.0 tasks. Gains are concentrated on contact-rich placing and pressing, while tasks that already had high success show flat or slightly degraded performance. PhysisForcing improves place_empty_cup by 21.5 points and press_stapler by 11.0 points, the largest gains among all tasks. Tasks with high baseline success, such as shake_bottle and adjust_bottle, see no improvement or a small drop, and stack_bowls_two decreases by 6.5 points.

Incorporating PhysisForcing into the Wan2.2-TI2V-5B video backbone raises the average closed-loop action planning success rate from 16.0% to 24.0%, surpassing all prior world-model planners including the previous best WoW at 20.5%. Gains are observed on both evaluation tasks, indicating robust planning improvements. PhysisForcing provides an 8-point absolute improvement in average success rate over the base video model. It outperforms the strongest prior world-model planner, WoW, by a clear margin. The success rate on Task 1 nearly doubles, while Task 2 also sees a solid increase.

Ablating the two physical losses on R-Bench shows they are complementary: the pixel-level loss alone delivers a larger individual improvement, especially on the embodiment metric, while adding the semantic loss further lifts task-oriented performance. Their combination, PhysisForcing, achieves the highest overall average and task scores on the 5B model, and consistent gains on the larger 14B model confirm the benefit scales across backbones. The pixel-level physical loss alone raises the average R-Bench score by 2.4 points over the finetuned baseline, exceeding the semantic loss's 1.4-point gain. Combining both losses into PhysisForcing yields the highest task score (38.9 vs. 35.4) and highest overall average (47.5 vs. 44.8) on the 5B model, even though the embodiment score is slightly below the pixel-loss-only setting. On the larger 14B model, the pixel loss boosts average score from 57.9 to 60.7 and the semantic loss to 60.0, demonstrating the improvements are not limited to a single model scale.

Applying physics losses uniformly over all video tokens improves average performance over the baseline, but restricting them to interaction-critical regions yields a further boost, especially on task-oriented metrics. This demonstrates that irrelevant background areas dilute the physical signal and focusing supervision on where robot-object interactions occur is what drives task-level correctness. Uniformly applying physics losses lifts the average score from 44.8 to 46.0, while focusing on physics-informative regions pushes it to 47.5. The largest gain is on task-oriented performance, which rises from 35.4 to 38.9 when supervision is concentrated on interaction-critical regions. The embedding metric also improves with region focus, from 56.5 to 58.2, showing better physical consistency. Background and near-static areas dilute the learning signal; restricting loss computation to robot-object interaction zones sharply improves task-level results.

PhysisForcing substantially improves performance across video generation, robotic manipulation, and world-model planning tasks. The method boosts video generation quality beyond both commercial and open-source baselines, enhances contact-rich manipulation policies, and raises closed-loop planning success rates. Complementary pixel-level and semantic physics losses, combined with supervision focused on robot-object interaction regions, drive these gains, and the benefits scale to larger video backbones.


AI로 AI 구축

아이디어에서 출시까지 — 무료 AI 코코딩, 즉시 사용 가능한 환경, 최적의 GPU 가격으로 AI 개발을 가속화하세요.

AI 협업 코딩
바로 사용 가능한 GPU
최적의 가격

HyperAI Newsletters

최신 정보 구독하기
한국 시간 매주 월요일 오전 9시 에 이번 주의 최신 업데이트를 메일로 발송합니다
이메일 서비스 제공: MailChimp