Загрузка видео...

Не удалось загрузить видео

На главную

Does off-policy value-based RL scale? In LLMs, larger scale predictably improves performance. Value-based RL learns from arbitrary data and is sample-efficient, but folk wisdom says it doesn't scale 🧵⬇️We show predictability for scaling value-based RL!

23,994 просмотров • 1 год назад •via X (Twitter)

Комментарии: 0

Нет доступных комментариев

Здесь появятся комментарии из оригинального поста

Похожие видео

Model-Free Reinforcement Learning (MFRL) has been alluring, especially with supercharged compute with physics on GPU. However, the methods use 0-th order gradients, and are often not the best optimizers. Can we do better than PPO in continuous control for robotics? Turns out yes! 🥳 tl;dr: Faster, better RL than PPO in continuous control 💪 The answer lies in using more information from the simulation. We are juicing the simulation on GPU as it is, why not use it for gradients as well? This has been a driving question in a series of our works. We first studied this problem in ICLR 2022 paper on Short Horizon Actor Critic Naive gradient based methods are stuck in local minima and have exploding/vanishing gradients. SHAC solved this problem truncated rollouts and model based value estimation, where the model is Differentiable Sim. This boosted sample efficiency and wall-clock time immensely especially in high dimensional systems such as humanoids Yet, given enough compute PPO often caught up. Our follow up paper on on Adaptive Horizon Actor Critic at ICML 2024 discovers the cause and provides a fix. However, we find that even when given ground-truth dynamics, not all gradients are useful due to sample error. 1st-Order Model-Based Reinforcement Learning methods employing differentiable simulation provide gradients with reduced variance but are susceptible to bias in scenarios involving stiff dynamics, such as physical contact. We find that back-propagating through contact and long trajectories drastically reduces gradient accuracy. Using this insight, we propose AHAC to dynamically adapt its roll-out horizon to avoid differentiating through stiff contact. AHAC is a first-order model-based RL algorithm that learns high-dimensional tasks in minutes (wall clock) and outperforms PPO by 40%, even in the limit of data provided to PPO. This work is led by Ignat Georgiev alongside Krishnan Srinivasan, Jie Xu, Eric Heiden and ample assistance from warp team at NVIDIA Robotics (Miles Macklin)

Animesh Garg

52,308 просмотров • 2 лет назад

AI has had exactly two scaling axes that worked so far, and the second one is starting to look finite too the first one was pretraining: with scaling parameters and data, we got world knowledge (i.e. ChatGPT had read enough to know things), but it started saturating a while ago the second one was RL, and people had been doing RL the whole time before that: RLHF is RL but it never scaled far because it was trying to control the exact output, which tokens come out, how the text reads, but you can only push that so far before you’re just polishing RLVR dropped that constraint: giving the model a task, then checking whether the final answer is right, and ignoring everything in between -- so the model does whatever it wants in the middle and only the endpoint gets graded, and that’s much closer to actual RL and it’s what bought us planning and reasoning (arguably, tool use sits around 2.5 on this list -- while useful, it's not a different kind of thing) so one axis gave knowledge, the other gave reasoning, and both of them are one model working alone the next axis is how many models you can get working on the same problem, which is a different kind of axis than the previous two we know that multi-agent RL has always been the harder problem: I spent years in that literature and the gap between single-agent and multi-agent is definitely not incremental -- it’s a whole different class of difficulty! which is also why the derivatives are steep at the start, nobody has picked the easy wins yet... and the thing that gates this multi-agent coordination is communication: models can only coordinate as well as they can exchange information, and right now they do that by writing sentences to each other imagine what could we possibly achieve if we properly open that third axis development by letting models to exchange information in their native "language" without loosing any computational data that they produce during inference

Sasha Malysheva

12,177 просмотров • 1 месяц назад

LongWriter Unleashing 10,000+ Word Generation from Long Context LLMs discuss: Current long context large language models (LLMs) can process inputs up to 100,000 tokens, yet struggle to generate outputs exceeding even a modest length of 2,000 words. Through controlled experiments, we find that the model's effective generation length is inherently bounded by the sample it has seen during supervised fine-tuning (SFT). In other words, their output limitation is due to the scarcity of long-output examples in existing SFT datasets. To address this, we introduce AgentWrite, an agent-based pipeline that decomposes ultra-long generation tasks into subtasks, enabling off-the-shelf LLMs to generate coherent outputs exceeding 20,000 words. Leveraging AgentWrite, we construct LongWriter-6k, a dataset containing 6,000 SFT data with output lengths ranging from 2k to 32k words. By incorporating this dataset into model training, we successfully scale the output length of existing models to over 10,000 words while maintaining output quality. We also develop LongBench-Write, a comprehensive benchmark for evaluating ultra-long generation capabilities. Our 9B parameter model, further improved through DPO, achieves state-of-the-art performance on this benchmark, surpassing even much larger proprietary models. In general, our work demonstrates that existing long context LLM already possesses the potential for a larger output window--all you need is data with extended output during model alignment to unlock this capability.

AK

50,995 просмотров • 2 лет назад

We are at NeurIPS Conference for our 4th #MyoChallenge and 7th #MyoSymposium! What started as a discussion with Vittorio Caggiano is a global community now MyoSuite 💪 In 2022, we started with two key hypotheses - 1⃣𝑺𝒄𝒂𝒍𝒊𝒏𝒈 𝑯𝒚𝒑𝒐𝒕𝒉𝒆𝒔𝒊𝒔: can we scale data driven learnings to achieve human level motor control? 2⃣𝑬𝒎𝒃𝒐𝒅𝒊𝒎𝒆𝒏𝒕 𝑯𝒚𝒑𝒐𝒕𝒉𝒆𝒔𝒊𝒔: Akin to Neurons's inspiration behind NN, are there embodied priors that will form the critical substrate to get to human performance? After 3 years, both these hypotheses are running strong. But in different ways than we anticipated -scaling hypothesis predicted vanilla RL algorithms (developed over OpenAI Gym and robotics tasks) will scale & realize human level motor control. RL did scale with better simulation & compute infrastructure but the curse of dimensionality became the limiter for high dimensional MSK systems. This is where our 2nd Embodiment Hypothesis kicked in. Spatial as well as morphological embodied priors facilitated development of next generation of algorithms at the intersection of representation and reinforcement learning - (DepRL from Pierre Schumacher et al, MyoDex & SAR from Cameron Vittorio Caggiano et al, Kinesis from Alberto Chiappa, muscleVAE from Yusen et al, etc) Current #MyoChallenge result evaluations phase was humbling realization - what started with a team of two evolved as a global community, the entry barriers has been lowered enough for even high school students, and underrepresented groups with limited resources to participate. It's incredible to realize the progress we have seen in 3 years. All the same time idiosyncrasies of the behaviors leave us quite unsatisfied and wanting more. Ahead of us there are exciting challenges on all frontiers -- embodiment, proprioceptive+exteroceptive sensing and control, validation -- presenting large real world potentials in health, wellness, sports, robotics. While there is a lot for us to be proud of, open challenges in understanding, as well as emulating human level motor intelligence remains. Join MyoSuite team for the awaited #MyoSymposium in discussing these frontiers on Saturday, Dec. 6th from 8-11 am: Ballroom 6D.

Vikash Kumar

11,145 просмотров • 9 месяцев назад