在機器人學習方面,常用方法是收集針對特定機器人和任務的資料集,然後用其來訓練策略。但是,如果使用這種方法來從頭開始學習,每個任務都需要收集足夠數據,並且所得策略的泛化能力通常也不佳。
「原理上講,從其他機器人和任務收集的經驗能提供可能的解決方案,能讓模型看到多種多樣的機器人控制問題,而這些問題也許能提升機器人在下游任務上的泛化能力和性能。
#要讓機器人訓練一個統一的控制策略非常困難,其中包括涉及諸多難點,包括操作不同的機器人機體、感測器配置、動作空間、任務規範、環境和計算預算。為了實現這一目標,已經出現了一些「機器人基礎模型」相關研究成果;它們的做法是直接將機器人觀察映射成動作,然後透過零樣本本方案泛化至新領域或新機器人。這些模型通常被稱為「通才機器人策略(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/
此模型的核心是Transformer 架構,其可將任意輸入token(根據觀察值和任務建立)對應成輸出token(然後編碼成動作),而且該架構可使用多樣化的機器人和任務資料集進行訓練。此策略無需額外訓練就能接受不同的相機配置,也能控制不同的機器人,還能透過語言指令或目標影像進行引導 ——— 所有這些只需透過改變輸入模型的 token 即可實現。
最重要的是,該模型還能適應感測器輸入、操作空間或機器人形態不同的新機器人配置,所需的只是採用適當的適配器(adapter)並使用一個小的目標領域資料集和少量計算預算進行微調。
不僅如此,Octo 還已經在迄今為止最大的機器人操控資料集上完成了預訓練—— 該資料集包含來自Open X-Embodiment 資料集的80 萬個機器人演示。 Octo 不僅是第一個可有效微調至新觀察和動作空間的 GRP,也是第一個完全開源(訓練工作流程、模型檢查點和資料)的通才機器人操控策略。該團隊也在論文中強調了其組合 Octo 各組件的獨特創新性。
#下面我們來看看Octo 這個開源的通才機器人策略是如何構建的。整體而言,Octo 的設計目標是讓其成為一個靈活且廣泛適用的通才機器人策略,可被大量不同的下游機器人應用和研究計畫使用。
架構 Octo 的核心是基於 Transformer 的策略 π。其包含三個關鍵部分:輸入 token 化器、Transformer 骨幹網路和讀出頭。 如圖2 所示,其中輸入token 化器的作用是將語言指令、目標和觀察序列轉換成token,Transformer 骨幹會把這些token 處理成嵌入,讀出頭則是得出所需的輸出,即動作。 為了將任務定義(例如語言指令和目標圖像)與觀察(例如相機視訊串流)轉換成常用的已token 化的格式,該團隊針對不同模態使用了不同的token 化器: 對於語言輸入,先token 化,然後透過一個預先訓練的Transformer 將其處理成一個語言嵌入token 序列。具體而言,他們使用的模型是 t5-base (111M)。 對於影像觀察和目標,則是透過較淺的捲積堆疊來處理,然後再拆分成平展後圖塊構成的序列。 最後,透過在任務和觀察 token 中添加可學習的位置嵌入並按一定順序排列它們來建立 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 進行了實證分析,在多個維度上評估了其作為機器人基礎模型的效能: 圖 4 展示了評估 Octo 的 9 種任務。 直接使用Octo 控制多台機器人 該團隊比較了Octo、RT-1-X、RT-2-X 的零樣本操控能力,結果如圖5。 可以看到,Octo 的成功率比 RT-1-X(3500 萬參數)高 29%。而在 WidowX 和 RT-1 Robot 評估上,Octo 與 550 億參數的 RT-2-X 效能相當。 此外,RT-1-X 和 RT-2-X 僅支援語言指令,而 Octo 也支援以目標影像為條件。團隊還發現,在 WidowX 任務上,如果使用目標影像為條件,成功率比使用語言為條件高 25%。這可能是因為目標影像能提供更多有關任務完成的資訊。 Octo 能有效地使用資料來適應新領域 表1 給出了資料高效能微調的實驗結果。 可以看到,比起從頭開始訓練或使用預訓練的VC-1 權重進行預訓練,微調Octo 得到的結果更好。在 6 種評估設定上,Octo 相較於第二名基準的平均優勢為 52%! 並且不得不提的是:針對所有這些評估任務,微調Octo 時使用的配方和超參數全都一樣,由此可見該團隊找到了一個非常好的預設配置。 通才機器人策略訓練的設計決策 #上面的結果顯示Octo 確實能作為零樣本多機器人控制器,也能作為策略微調的初始化基礎。接下來,團隊分析了不同設計決策對 Octo 策略表現的影響。具體而言,他們關注的重點是以下幾個面向:模型架構、訓練資料、訓練目標、模型規模。為此,他們進行了消融研究。 表 2 給出了在模型架構、訓練資料和訓練目標的消融研究結果。 圖6 則展現了模型規模對零樣本成功率的影響,可以看出來更大的模型有更好的視覺場景感知能力。 整體而言,Octo 各元件的有效性都得到了證明。 任務和觀察token 化器
Transformer 骨幹和讀出頭
實驗
以上是適應多形態多任務,最強開源機器人學習系統「八爪魚」誕生的詳細內容。更多資訊請關注PHP中文網其他相關文章!

1 前言在发布DALL·E的15个月后,OpenAI在今年春天带了续作DALL·E 2,以其更加惊艳的效果和丰富的可玩性迅速占领了各大AI社区的头条。近年来,随着生成对抗网络(GAN)、变分自编码器(VAE)、扩散模型(Diffusion models)的出现,深度学习已向世人展现其强大的图像生成能力;加上GPT-3、BERT等NLP模型的成功,人类正逐步打破文本和图像的信息界限。在DALL·E 2中,只需输入简单的文本(prompt),它就可以生成多张1024*1024的高清图像。这些图像甚至

Wav2vec 2.0 [1],HuBERT [2] 和 WavLM [3] 等语音预训练模型,通过在多达上万小时的无标注语音数据(如 Libri-light )上的自监督学习,显著提升了自动语音识别(Automatic Speech Recognition, ASR),语音合成(Text-to-speech, TTS)和语音转换(Voice Conversation,VC)等语音下游任务的性能。然而这些模型都没有公开的中文版本,不便于应用在中文语音研究场景。 WenetSpeech [4] 是

“Making large models smaller”这是很多语言模型研究人员的学术追求,针对大模型昂贵的环境和训练成本,陈丹琦在智源大会青源学术年会上做了题为“Making large models smaller”的特邀报告。报告中重点提及了基于记忆增强的TRIME算法和基于粗细粒度联合剪枝和逐层蒸馏的CofiPruning算法。前者能够在不改变模型结构的基础上兼顾语言模型困惑度和检索速度方面的优势;而后者可以在保证下游任务准确度的同时实现更快的处理速度,具有更小的模型结构。陈丹琦 普

由于复杂的注意力机制和模型设计,大多数现有的视觉 Transformer(ViT)在现实的工业部署场景中不能像卷积神经网络(CNN)那样高效地执行。这就带来了一个问题:视觉神经网络能否像 CNN 一样快速推断并像 ViT 一样强大?近期一些工作试图设计 CNN-Transformer 混合架构来解决这个问题,但这些工作的整体性能远不能令人满意。基于此,来自字节跳动的研究者提出了一种能在现实工业场景中有效部署的下一代视觉 Transformer——Next-ViT。从延迟 / 准确性权衡的角度看,

3月27号,Stability AI的创始人兼首席执行官Emad Mostaque在一条推文中宣布,Stable Diffusion XL 现已可用于公开测试。以下是一些事项:“XL”不是这个新的AI模型的官方名称。一旦发布稳定性AI公司的官方公告,名称将会更改。与先前版本相比,图像质量有所提高与先前版本相比,图像生成速度大大加快。示例图像让我们看看新旧AI模型在结果上的差异。Prompt: Luxury sports car with aerodynamic curves, shot in a

人工智能就是一个「拼财力」的行业,如果没有高性能计算设备,别说开发基础模型,就连微调模型都做不到。但如果只靠拼硬件,单靠当前计算性能的发展速度,迟早有一天无法满足日益膨胀的需求,所以还需要配套的软件来协调统筹计算能力,这时候就需要用到「智能计算」技术。最近,来自之江实验室、中国工程院、国防科技大学、浙江大学等多达十二个国内外研究机构共同发表了一篇论文,首次对智能计算领域进行了全面的调研,涵盖了理论基础、智能与计算的技术融合、重要应用、挑战和未来前景。论文链接:https://spj.scien

译者 | 李睿审校 | 孙淑娟近年来, Transformer 机器学习模型已经成为深度学习和深度神经网络技术进步的主要亮点之一。它主要用于自然语言处理中的高级应用。谷歌正在使用它来增强其搜索引擎结果。OpenAI 使用 Transformer 创建了著名的 GPT-2和 GPT-3模型。自从2017年首次亮相以来,Transformer 架构不断发展并扩展到多种不同的变体,从语言任务扩展到其他领域。它们已被用于时间序列预测。它们是 DeepMind 的蛋白质结构预测模型 AlphaFold

说起2010年南非世界杯的最大网红,一定非「章鱼保罗」莫属!这只位于德国海洋生物中心的神奇章鱼,不仅成功预测了德国队全部七场比赛的结果,还顺利地选出了最终的总冠军西班牙队。不幸的是,保罗已经永远地离开了我们,但它的「遗产」却在人们预测足球比赛结果的尝试中持续存在。在艾伦图灵研究所(The Alan Turing Institute),随着2022年卡塔尔世界杯的持续进行,三位研究员Nick Barlow、Jack Roberts和Ryan Chan决定用一种AI算法预测今年的冠军归属。预测模型图


熱AI工具

Undresser.AI Undress
人工智慧驅動的應用程序,用於創建逼真的裸體照片

AI Clothes Remover
用於從照片中去除衣服的線上人工智慧工具。

Undress AI Tool
免費脫衣圖片

Clothoff.io
AI脫衣器

AI Hentai Generator
免費產生 AI 無盡。

熱門文章

熱工具

Dreamweaver CS6
視覺化網頁開發工具

禪工作室 13.0.1
強大的PHP整合開發環境

Safe Exam Browser
Safe Exam Browser是一個安全的瀏覽器環境,安全地進行線上考試。該軟體將任何電腦變成一個安全的工作站。它控制對任何實用工具的訪問,並防止學生使用未經授權的資源。

SublimeText3 Mac版
神級程式碼編輯軟體(SublimeText3)

Atom編輯器mac版下載
最受歡迎的的開源編輯器