AIxiv專欄是機器之心發(fā)布學術、技術內容的欄目。過去數(shù)年,機器之心AIxiv專欄接收報道了2000多篇內容,覆蓋全球各大高校與企業(yè)的頂級實驗室,有效促進了學術交流與傳播。如果您有優(yōu)秀的工作想要分享,歡迎投稿或者聯(lián)系報道。投稿郵箱:liyazhou@jiqizhixin.com;zhaoyunfeng@jiqizhixin.com
本文作者來自清華大學、字節(jié)跳動、中科院自動化所、上海交通大學和新加坡國立大學。作者列表:李興航、李沛言、劉明桓、王棟、劉濟榕、康炳易、馬驍、孔濤、張翰博和劉華平。第一作者李興航是清華大學計算機系博士生。通訊作者是字節(jié)跳動機器人研究員孔濤,新加坡國立大學博士后張翰博和清華大學計算機系教授劉華平。近年來,視覺語言基礎模型(Vision Language Models, VLMs)大放異彩,在多模態(tài)理解和推理上展現(xiàn)出了超強能力,F(xiàn)在,更加酷炫的視覺語言動作模型(Vision-Language-Action Models, VLAs)來了!通過為 VLMs 加上動作預測模塊,VLAs 不僅能 “看” 懂和 “說” 清,還能 “動” 起來,為機器人領域開啟了新玩法!
雖然 VLAs 在各種任務和場景中表現(xiàn)搶眼,但大家在模型設計上卻走了很多不同的路,比如用什么架構、怎么選數(shù)據(jù)、怎么調訓練策略等等,這導致領域內對 “怎么做好一個 VLA” 還沒有統(tǒng)一的答案。為了理清這些問題,我們通過一系列的實驗,提出了一個全新模型 RoboVLMs。
論文標題:Towards Generalist Robot Policies: What Matters inBuilding Vision-Language-Action Models
論文地址:https://arxiv.org/pdf/2412.14058
這個模型超級簡單,但性能卻相當硬核!它不僅在三個模擬任務中取得了高分,還在真實機器人實驗中交出了滿分答卷。這篇文章就是要帶你一起看看,我們是如何用 RoboVLMs 解鎖 VLA 的無限可能!四大靈魂拷問:RoboVLMs 是怎么煉成的?我們圍繞四個關鍵問題,對 VLA 的設計展開了深度探索,下面就帶你看看答案!1. 為什么要用 VLA 模型?簡單說,通過實驗,我們發(fā)現(xiàn)設計合理的 VLA 不僅能輕松搞定常見的操作任務,還能在陌生場景中穩(wěn)穩(wěn)發(fā)揮。仿真任務中拿下頂尖成績在 CALVIN 和 SimplerEnv 環(huán)境里,RoboVLMs 取得了壓倒性的勝利:
任務成功率:表現(xiàn)穩(wěn)定且超越主流模型。
泛化能力:即使在陌生場景中,表現(xiàn)依然抗打!
圖 1 SimplerEnv 仿真環(huán)境中的評測結果
圖 2 針對視覺語言預訓練的消融實驗結果
真實機器人實驗也不輸在真實環(huán)境中,RoboVLMs 面對更復雜的挑戰(zhàn),仍然比其他模型表現(xiàn)更好。比如,在果蔬分類任務中,它不僅能精準識別,還能應對干擾環(huán)境,穩(wěn)穩(wěn)完成分類操作。無論是已知場景還是新任務,它都能輕松拿下。
圖 3 真實環(huán)境下的評測結果對于未見過的技能描述、背景、干擾物體和目標物體,RoboVLMs 均能很好的完成任務。
2. 怎么設計一個靠譜的 VLA 架構?這里面講究可不少!比如:
動作空間:用連續(xù)動作空間比離散的好很多。
歷史信息:加多步歷史信息后,模型的操作更穩(wěn)準狠。
歷史信息組織模塊:一個專門的模塊可以讓模型更懂 “上下文”。
經過一系列實驗,我們確認了這些設計選擇是提升模型性能和泛化能力的關鍵。進一步的實驗也表明,最優(yōu)的設計來自于基于 KosMos 基座模型的架構,并且結合了專門的歷史信息組織模塊。這樣的設計在 CALVIN 中實現(xiàn)了出色的泛化能力,在 zero-shot 設置下僅有輕微的性能下降,而其他設計形式的模型則出現(xiàn)了顯著掉分。這一結論直接說明,架構設計的好壞對模型的泛化能力和效率至關重要。
3. 選什么基座模型最合適?我們對比了當前主流的 8 種視覺語言模型(VLM),結果發(fā)現(xiàn) KosMos 和 Paligemma 的表現(xiàn)遙遙領先,輕松碾壓其他模型。無論是任務完成的精確度還是泛化能力,它們都展現(xiàn)出了壓倒性的優(yōu)勢。究其原因,主要得益于它們經過了扎實且全面的視覺語言預訓練,從而為模型提供了強大的先驗知識和理解能力。這一發(fā)現(xiàn)讓我們更加確信:選對基座模型,就是讓 VLA 模型起飛的關鍵一步!想要讓模型在多模態(tài)任務中表現(xiàn)驚艷,一個經過深度預訓練、具備強大視覺語言表征能力的 VLM 基座顯然能提供無與倫比的助力。而一旦打好了這個基礎,后續(xù)的設計和訓練才能真正發(fā)揮最大潛力。
4. 跨本體數(shù)據(jù)什么時候加入最合適?實驗告訴我們一個黃金法則:在預訓練階段引入跨本體數(shù)據(jù)(如 Open-X Embodiment 數(shù)據(jù)集)可以顯著提升模型的魯棒性和少樣本場景下的表現(xiàn)。反之,直接將跨本體數(shù)據(jù)和微調數(shù)據(jù)混合訓練,效果就沒那么顯著了。這些結論為未來 VLA 模型的訓練策略指明了方向。具體實驗中,我們在 WidowX+Bridge 和 Google Robot 兩大環(huán)境下分別進行了不同訓練策略的測試:WidowX+Bridge 環(huán)境:
Bridge Finetune:直接在完整的 Bridge 數(shù)據(jù)集上微調(測試任務不包括在內)。
OXE Pre-Train:先用 OXE 數(shù)據(jù)集預訓練模型。
Post-Train:用經過 OXE 預訓練的模型再在 Bridge 數(shù)據(jù)集上微調。
Google Robot 環(huán)境:
RT-Partial Finetune:僅在特定的 RT 任務上微調。
RT Finetune:在完整的 RT 數(shù)據(jù)集上微調(包括測試任務)。
OXE Pre-Train:先用 OXE 數(shù)據(jù)集預訓練模型。
Post-Train:在 OXE 預訓練基礎上用 RT 數(shù)據(jù)集進一步訓練。
實驗結果進一步驗證了:在預訓練階段引入跨本體數(shù)據(jù)不僅能提升泛化能力,還能讓模型在少樣本和高復雜任務下表現(xiàn)更佳。
展望未來:VLA 的進階之路雖然 RoboVLMs 已經很能打了,但接下來的發(fā)展空間更讓人期待!未來可以探索:
更細化的設計優(yōu)化:比如再打磨 VLM 內部結構、信息融合模塊和訓練目標,讓它更高效。
挑戰(zhàn)復雜任務:像 “做早餐” 這種長鏈條任務,也許是下一個突破點!
多模態(tài)協(xié)作能力:進一步讓機器人 “看懂”、“聽清”、“動得更聰明”。
RoboVLMs 的出現(xiàn),驗證了視覺語言動作模型的可能性,也讓機器人更接近成為我們的全能助手。未來,它們或許不僅能理解語言和視覺,還能真正幫我們完成那些繁瑣又復雜的任務。接下來會有更多驚喜等著我們!