適應多形態多任務，最強開源機器人學習系統「八爪魚」誕生-人工智慧-PHP中文網

在機器人學習方面，常用方法是收集針對特定機器人和任務的資料集，然後用其來訓練策略。但是，如果使用這種方法來從頭開始學習，每個任務都需要收集足夠數據，並且所得策略的泛化能力通常也不佳。

「原理上講，從其他機器人和任務收集的經驗能提供可能的解決方案，能讓模型看到多種多樣的機器人控制問題，而這些問題也許能提升機器人在下游任務上的泛化能力和性能。

#要讓機器人訓練一個統一的控制策略非常困難，其中包括涉及諸多難點，包括操作不同的機器人機體、感測器配置、動作空間、任務規範、環境和計算預算。

為了實現這一目標，已經出現了一些「機器人基礎模型」相關研究成果；它們的做法是直接將機器人觀察映射成動作，然後透過零樣本本方案泛化至新領域或新機器人。這些模型通常被稱為「通才機器人策略（generalist robot policy）」，簡稱 GRP，這強調了機器人跨多種任務、環境和機器人系統執行低階視覺運動控制的能力。

GNM（General Navigation Model，通用導航模型）適用於多種不同的機器人導航場景，RoboCat 可針對任務目標操作不同的機器人機體，RT-X 能透過語言操作五種不同的機器人機體。儘管這些模型確實是重要進展，但它們也存在多方面的限制：它們的輸入觀察通常是預先定義的且通常很有限（例如單相機輸入視訊串流）；它們難以有效微調至新領域；這些模型中最大型的版本都沒有提供人們使用（這一點很重要）。

近日，加州大學柏克萊分校、史丹佛大學、卡內基美隆大學和GoogleDeepMind的18位研究者組成的Octo Model Team發布了他們的開創性研究成果： Octo模型。該項目有效地克服了上述限制。

適應多形態多任務，最強開源機器人學習系統「八爪魚」誕生

論文標題： Octo: An Open-Source Generalist Robot Policy
論文網址：https://arxiv.org/pdf/2405.12213
開源專案：https://octo-models. github.io/

他們設計了一個系統，能讓GRP 更輕鬆地應對下游機器人應用的介面多樣化問題。

此模型的核心是Transformer 架構，其可將任意輸入token（根據觀察值和任務建立）對應成輸出token（然後編碼成動作），而且該架構可使用多樣化的機器人和任務資料集進行訓練。此策略無需額外訓練就能接受不同的相機配置，也能控制不同的機器人，還能透過語言指令或目標影像進行引導 ——— 所有這些只需透過改變輸入模型的 token 即可實現。

最重要的是，該模型還能適應感測器輸入、操作空間或機器人形態不同的新機器人配置，所需的只是採用適當的適配器（adapter）並使用一個小的目標領域資料集和少量計算預算進行微調。

不僅如此，Octo 還已經在迄今為止最大的機器人操控資料集上完成了預訓練—— 該資料集包含來自Open X-Embodiment 資料集的80 萬個機器人演示。 Octo 不僅是第一個可有效微調至新觀察和動作空間的 GRP，也是第一個完全開源（訓練工作流程、模型檢查點和資料）的通才機器人操控策略。該團隊也在論文中強調了其組合 Octo 各組件的獨特創新性。

適應多形態多任務，最強開源機器人學習系統「八爪魚」誕生

Octo 模型

#下面我們來看看Octo 這個開源的通才機器人策略是如何構建的。整體而言，Octo 的設計目標是讓其成為一個靈活且廣泛適用的通才機器人策略，可被大量不同的下游機器人應用和研究計畫使用。

架構

Octo 的核心是基於 Transformer 的策略 π。其包含三個關鍵部分：輸入 token 化器、Transformer 骨幹網路和讀出頭。

如圖2 所示，其中輸入token 化器的作用是將語言指令、目標和觀察序列轉換成token，Transformer 骨幹會把這些token 處理成嵌入，讀出頭則是得出所需的輸出，即動作。

適應多形態多任務，最強開源機器人學習系統「八爪魚」誕生

任務和觀察token 化器

為了將任務定義（例如語言指令和目標圖像）與觀察（例如相機視訊串流）轉換成常用的已token 化的格式，該團隊針對不同模態使用了不同的token 化器：

對於語言輸入，先token 化，然後透過一個預先訓練的Transformer 將其處理成一個語言嵌入token 序列。具體而言，他們使用的模型是 t5-base (111M)。

對於影像觀察和目標，則是透過較淺的捲積堆疊來處理，然後再拆分成平展後圖塊構成的序列。

最後，透過在任務和觀察 token 中添加可學習的位置嵌入並按一定順序排列它們來建立 Transformer 的輸入序列。

Transformer 骨幹和讀出頭

#將輸入處理成一種統一化的 token 序列之後，就能交給 Transformer 處理了。這與先前的研究工作類似：基於觀察和動作系列來訓練基於 Transformer 的策略。

Octo 的注意力模式是逐塊遮罩式：觀察 token 只能依照因果關係關注來自同一或先前時間步驟的 token 以及任務 token。對應於不存在觀察的 token 會被完全遮蔽掉（例如沒有語言指令的資料集）。這種模組化設計很方便，可在微調階段中新增或移除觀察或任務。

除了這些輸入 token 模組，團隊還插入了已學習完成的讀出 token。讀出 token 會專注於其先前的觀察和任務 token，但不會被任何觀察或任務 token 關注。因此，讀出 token 只能讀取和處理內部嵌入，而無法影響內部嵌入。讀出 token 的作用類似於 BERT 中的 [CLS] token，充當截至目前的觀察序列的緊湊向量嵌入。針對讀出 token 的嵌入，會使用一個輕量的實現擴散過程的「動作頭」。這個動作頭會預測多個連續動作構成的一個「區塊（chunk）」。

這樣的設計可讓使用者在下游微調時向模型靈活地新增新的任務和觀察輸入或動作輸出頭。當在下游添加新的任務、觀察或損失函數時，可以在整體上保留Transformer 的預訓練權重，僅添加新的位置嵌入、一個新的輕量編碼器、或由於規範變化而必需的新頭的參數。這不同於先前的架構 —— 對於先前的架構，如果新增或移除影像輸入或改變任務規範，就需要重新初始化或重新訓練預訓練模型的大量組件。

要讓Octo 成為真正的「通才」模型，這種靈活性至關重要：由於我們不可能在預訓練階段覆蓋所有可能的機器人感測器和動作配置，因此，如果能在微調階段調整Octo 的輸入和輸出，讓其成為機器人社群的多功能工具。另外，先前使用標準 Transformer 骨幹或融合使用視覺編碼器與 MLP 輸出頭的模型設計固定了模型輸入的類型和順序。相較之下，切換 Octo 的觀察或任務並不需要對大部分模型進行重新初始化。

訓練資料

#該團隊從Open X-Embodiment 取用了包含25 個資料集的混合數據集。圖 3 給出了資料集的組成。

適應多形態多任務，最強開源機器人學習系統「八爪魚」誕生

有關訓練目標和訓練硬體配置等更多細節請參閱原始論文。

模型檢查點和程式碼

#重點來了！團隊不僅發了 Octo 的論文，還完全開源了所有資源，其中包括：

預先訓練完成的 Octo 檢查點，包括 2700 萬參數的 Octo-Small 和 9300 萬參數的 Octo-Base。
用於 Octo 模型的微調腳本，基於 JAX。
用於在 Open X-Embodiment 資料集上預先訓練 Octo 的模型預訓練工作流程，基於 JAX。用於 Open X-Embodiment 資料的資料載入器，相容於 JAX 和 PyTorch。