Transformer世界模型。Transformer式動態模型(例如TransDreamer)用自注意力機制代替循環,做到平行訓練同直接存取長歷史。但標準嘅二次方注意力開銷(同序列長度成平方關係)令到長時間想像同長上下文視覺建模嘅運算成本好高。
結構化狀態空間模型(S4)。S4係專為長距離依賴而設計嘅結構化狀態空間序列模型,佢結合咗循環狀態空間嘅解釋同卷積形式,可以做到平行序列訓練。論文認為S4係一個好有潛力嘅中間方案:推理時係線性時間(快過二次方),訓練又可以平行。
S4WM嘅獨特之處。唔似其他人將S4用喺傳統序列基準任務,S4WM將S4嵌入到潛在視覺世界模型入面,包含圖像編碼/解碼同隨機潛在動態,然後喺同一世界模型框架下同RNN同Transformer競爭。佢哋聲稱呢個係第一個用S4做世界模型、可以透過潛在想像生成高維度圖像序列嘅方法。
文獻回顧入面可以咁樣一句總結佢嘅定位:
之前嘅世界模型主要用循環潛在動態或者Transformer。RNN模型推理好快但訓練要順序進行,Transformer世界模型訓練可平行但時間成本係二次方。S4WM嘗試用結構化狀態空間層做替代骨幹,目標係保留平行訓練嘅同時,支援高效嘅長時域動態建模。