Video yükleniyor...

Video Yüklenemedi

Ana Sayfaya Dön

We can teach LLMs to write better robot code through natural language feedback. But can LLMs remember what they were taught and improve their teachability over time? Introducing our latest work, Learning to Learn Faster from Human Feedback with Language Model Predictive Control

86,652 görüntüleme • 2 yıl önce •via X (Twitter)

13 Yorum

Jacky Liang profil fotoğrafı
Jacky Liang2 yıl önce

TL;DR We improve LLMs’ in-context teachability by fine-tuning them to model human-robot interactions. Here’s an example of robot teaching sessions before and after finetuning. Before, the LLM requires many corrections to do a high-five; after, the model gets there much faster

Jacky Liang profil fotoğrafı
Jacky Liang2 yıl önce

Given a dataset of these chat-based teaching sessions, how do we improve LLMs’ teachability (number of chat turns it takes to reach task success)? We want to improve teachability not just on train tasks, but also on test tasks and test embodiments.

Jacky Liang profil fotoğrafı
Jacky Liang2 yıl önce

Instead of finetuning to directly output final answers, we propose Language Model Predictive Control: 1) Finetune the LLM to predict the entire chat session 2) During inference: sample rollouts of future chat sessions, return the first response of the shortest successful chat

Jacky Liang profil fotoğrafı
Jacky Liang2 yıl önce

Compared to retrieval (RAG) and directly predict the answer (LMPC-Skip), our method (LMPC-Rollouts) achieves the highest improvement in teachability over the base model (PaLM 2-S) With 1 chat turn, LMPC-Skip does the best, but LMPC-Rollouts improves more with corrective feedback

Jacky Liang profil fotoğrafı
Jacky Liang2 yıl önce

On test tasks, LMPC-Rollouts have the highest success rate when the chat has 2+ turns; it also has the highest good rating rate (if human teachers rated each chat response positively).

Jacky Liang profil fotoğrafı
Jacky Liang2 yıl önce

While we only fine-tune on 3 embodiments (robot dog, mobile manipulator, aloha), we also see improvements on test embodiments (bimanual kuka, kuka+hand), both have different robot APIs and tasks

Jacky Liang profil fotoğrafı
Jacky Liang2 yıl önce

Our method enables better teaching on real robots too! Here’s an example with the robot dog…

Jacky Liang profil fotoğrafı
Jacky Liang2 yıl önce

…and here’s another example for our mobile manipulation robot

Jacky Liang profil fotoğrafı
Jacky Liang2 yıl önce

See more details in our paper, and videos and demo on our website:

Jacky Liang profil fotoğrafı
Jacky Liang2 yıl önce

This was a huge collaboration from many folks at Google DeepMind Robotics. I learned a ton from everyone and am super grateful for the amazing teamwork! Special thanks to @xf1280 @Stacormed @andyzeng_! Big shoutouts to our incredible collaborators: @montseglz @JMarakiii …

Jacky Liang profil fotoğrafı
Jacky Liang2 yıl önce

… @bauzavillalonga Matthew Bennice @AlexBewleyAI @AdilDostmohamed @ChuyuanFu @NimTheCoder Marissa Giustina @keerthanpg @lqh20 Jan Humplik, Jasmine Hsu, Nikhil Joshi, Ben Jyenis, Chase Kew, @SeanKirmani Edward Lee, @kuanghueilee, Assaf Hurwitz Michaely, Joss Moore, Ken Oslund, …

Jacky Liang profil fotoğrafı
Jacky Liang2 yıl önce

…, Dushyant Rao, @allenzren Baruch Tabanpour @QuanVng @ayzwah @xiao_ted Ying Xu, Vincent Zhuang, as well as our incredible advising leads: Peng Xu, Erik Frey, Ken Caluwaerts, Tingnan Zhang, @brian_ichter @JonathanTompson @leilatakayama Vincent Vanhoucke @IzhakShafran …

Jacky Liang profil fotoğrafı
Jacky Liang2 yıl önce

…Maja Mataric @DorsaSadigh Nicolas Heess @Kanishka_Rao Nik Stewart, Jie Tan, Carolina Parada Thanks everyone!!

Benzer Videolar

3D-LLM: Injecting the 3D World into Large Language Models paper page: Large language models (LLMs) and Vision-Language Models (VLMs) have been proven to excel at multiple tasks, such as commonsense reasoning. Powerful as these models can be, they are not grounded in the 3D physical world, which involves richer concepts such as spatial relationships, affordances, physics, layout, and so on. In this work, we propose to inject the 3D world into large language models and introduce a whole new family of 3D-LLMs. Specifically, 3D-LLMs can take 3D point clouds and their features as input and perform a diverse set of 3D-related tasks, including captioning, dense captioning, 3D question answering, task decomposition, 3D grounding, 3D-assisted dialog, navigation, and so on. Using three types of prompting mechanisms that we design, we are able to collect over 300k 3D-language data covering these tasks. To efficiently train 3D-LLMs, we first utilize a 3D feature extractor that obtains 3D features from rendered multi- view images. Then, we use 2D VLMs as our backbones to train our 3D-LLMs. By introducing a 3D localization mechanism, 3D-LLMs can better capture 3D spatial information. Experiments on ScanQA show that our model outperforms state-of-the-art baselines by a large margin (e.g., the BLEU-1 score surpasses state-of-the-art score by 9%). Furthermore, experiments on our held-in datasets for 3D captioning, task composition, and 3D-assisted dialogue show that our model outperforms 2D VLMs. Qualitative examples also show that our model could perform more tasks beyond the scope of existing LLMs and VLMs.

AK

249,708 görüntüleme • 3 yıl önce

New Course: Reinforcement Fine-Tuning LLMs with GRPO! Learn to use reinforcement learning to improve your LLM performance in this short course, built in collaboration with Predibase by Rubrik, and taught by Travis Addair, its Co-Founder and CTO, and Arnav Garg, its Senior Engineer and Machine Learning Lead. Reasoning models have been one of the most important developments in LLMs. Reinforcement Fine-Tuning (RFT) uses rewards to encourage LLMs to find solutions to multi-step reasoning tasks such as solving math problems and debugging code - without needing pre-existing training examples like in traditional supervised fine-tuning. Group Relative Policy Optimization (GRPO) is a reinforcement fine-tuning algorithm gaining rapid adoption. Developed by the DeepSeek team and used to train the R1 reasoning model, GRPO uses reward functions that you can write in Python to assign rewards to model responses. It’s beneficial for tasks with verifiable outcomes and can work well even with fewer than 100 training examples. It can also significantly improve the reasoning ability of smaller LLMs, making applications faster and more cost effective. In this course, you’ll take a technical deep dive into RFT with GRPO. You’ll learn to build reward functions that you can use in the GRPO training process to guide an LLM toward better performance on multi-step reasoning tasks. In detail, you’ll: - Learn when reinforcement fine-tuning is a better fit than supervised fine-tuning, especially for tasks involving multi-step reasoning or limited labeled data. - Understand how GRPO uses programmable reward functions as a more scalable alternative to the human feedback required for other reinforcement learning algorithms, such as RLHF and DPO. - Frame the Wordle game as a reinforcement fine-tuning problem and see how an LLM can learn to plan, analyze feedback, and improve its strategy over time. - Design reward functions that power the reinforcement fine-tuning process. - Learn techniques for evaluating more subjective tasks, such as rating the quality of a text summary, using an LLM as a judge. - Understand why reward hacking happens and how to avoid it by adding penalty functions to discourage undesirable behaviors. - Learn the four key components of the loss calculation in the GRPO algorithm: token probability distribution ratios, advantages, clipping, and KL-divergence. - Launch reinforcement fine-tuning jobs using Predibase’s hosted training services. By the end of this course, you’ll be able to build and fine-tune LLMs using reinforcement learning to improve reasoning without relying on large labeled datasets or subjective human feedback. Please sign up here:

Andrew Ng

86,457 görüntüleme • 1 yıl önce