DreamX-Phi 1.0:讓機器人「預知未來畫面」的動作條件影片世界模型
DreamX-Phi 1.0: Action-Conditioned Video World Model for Robotic Manipulation
背景
機器人操作任務的一大瓶頸,是缺乏能準確模擬「執行某個動作後會發生什麼」的環境模型。傳統作法要嘛依賴昂貴的真實機器人試錯,要嘛依賴物理模擬器,但模擬器與真實世界之間存在明顯的 sim-to-real 落差。近年興起的「影片世界模型」(video world model)嘗試用生成式影片模型直接預測未來畫面,把動作規劃、策略訓練甚至評測都建立在「想像出來的」影片 rollout 上。但這類模型常見的問題是:畫面看起來很真實、很流暢,卻可能出現「手臂動錯方向」或「被抓取的物體憑空消失/變形」等不忠實(unfaithful)於實際物理與動作指令的情況——也就是「看起來對,實際上錯」。DreamX Team 提出的 DreamX-Phi 1.0,正是針對這個「真實感 vs 忠實度」落差而設計的動作條件(action-conditioned)影片世界模型,鎖定雙臂機器人操作場景,輸入一張觀測影格、一句語言指令,以及一段包含末端執行器(end-effector)姿態與夾爪開合狀態的動作序列,輸出對應的未來畫面預測。
方法
DreamX-Phi 1.0 以 Wan2.2-TI2V-5B 這個文字/影像轉影片的 5B 參數 transformer 為基底進行微調與擴充,核心設計有三個層次。第一,為了讓模型準確跟隨每隻手臂各自下達的路徑指令,團隊採用 PRoPE 風格的幾何編碼,把每隻手臂的 SE(3)(三維剛體旋轉+平移)變換直接注入到 attention 機制中,而不是像過去做法把動作壓縮成一般性的 token 向量——這樣能保留「哪隻手臂對應哪段運動」的身份資訊與剛體運動結構,避免雙臂任務中「認錯手臂」的問題。第二,單靠動作條件並不足以約束場景整體幾何與細小被操作物體的演變,因此加入一個輕量的深度(depth)分支來監督場景級三維幾何一致性,並利用 SAM3 分割遮罩搭配一個凍結參數的 V-JEPA 教師模型,在抓取過程中維持物體外觀與存在的一致性,防止物體在生成影片中被生成模型「幻覺」掉。第三,考量到多步驟擴散式生成器在部署時延遲過高,團隊透過 distribution-matching distillation,把原本需要多步取樣的生成器蒸餾成一個只需幾步推論的學生模型,兼顧生成品質與推論效率。訓練資料上,團隊混合了無動作標註的第一人稱(egocentric)影片、真實機器人示範資料,以及模擬軌跡,其中動作條件資料池包含約 25,000 段經過 DreamX-Refiner 精煉處理的雙臂 RoboTwin 模擬片段。
實驗結果
在 WorldArena 2.0 Challenge 這個針對機器人世界模型的公開評測賽事中,DreamX-Phi 1.0 在 Track 1(影片預測賽道)以 EWMScore-P 60.65 分,在 31 支參賽隊伍中拿下第一名;在 Track 2(以世界模型輔助策略訓練的賽道)於 Adjust Bottle 任務上達到 67.19% 的成功率,並列第二名。此外,團隊也在前一代的 WorldArena 1.0 上做了離線(offline)評測,取得 76.88 分,較當時官方排行榜第一名高出 3.24 分,顯示其方法在不同版本評測基準上都具備一致的競爭力,而非單純針對單一比賽版本調參。這些數字說明,PRoPE 幾何編碼、深度分支與 SAM3+V-JEPA 一致性監督這三項設計,對於同時提升「動作跟隨準確度」與「下游策略訓練有效性」都有實質貢獻,而非只是讓影片視覺上更好看。
意義
DreamX-Phi 1.0 的價值不只在拿下比賽名次,而在於它示範了一條「用生成式影片模型取代或輔助物理模擬器」的可行路徑:當世界模型足夠忠實地反映動作與物理約束時,它既可以作為機器人策略的低成本訓練環境(Track 2 場景),也可以作為評估、除錯乃至規劃階段的「沙盒」。文中特別強調「真實感不等於忠實度」這個觀察,對整個具身智慧(embodied AI)與世界模型社群具有指導意義——未來評測影片世界模型時,除了畫質與流暢度等生成指標,更需要針對動作一致性、物體恆常性等維度做專門檢驗。同時,將幾何結構(SE(3)、深度)顯式注入 attention、並用蒸餾大幅壓縮推論步數的組合拳,也為後續想把大型影片生成模型落地到真實機器人系統的團隊,提供了一套兼顧精度與效率的參考範式。隨著模型與程式碼公開,預期會加速雙臂機器人操作、世界模型驅動的策略學習等方向的後續研究。