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

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

На главную

We developed a simple, sample-efficient online RL technique for post-training image generation models. We see it as a possible steerable alternative to CFG, driven by any scalar reward, including human preference.

70,991 просмотров • 6 месяцев назад •via X (Twitter)

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

Фото профиля David McAllister
David McAllister6 месяцев назад

Check out our blog post at for a walkthrough of our design decisions. w/ fantastic collaborators: Miika Aittala, Tero Karras, Janne Hellsten, @akanazawa Timo Aila and Samuli Laine

Фото профиля Kush Hari
Kush Hari6 месяцев назад

You just got… Mcallister’d!!!

Фото профиля Samarth Sinha
Samarth Sinha6 месяцев назад

My hero 😍😍😍😍😍

Фото профиля atharva ☆
atharva ☆6 месяцев назад

wow

Фото профиля Siddharth Ancha
Siddharth Ancha6 месяцев назад

Really cool work @davidrmcall!!!

Фото профиля James Thewlis
James Thewlis6 месяцев назад

Epoch 0 is not a happy kitty 😂 very cool work!

Фото профиля Ritvik Singh
Ritvik Singh6 месяцев назад

Congrats, David!

Фото профиля Phúc
Phúc6 месяцев назад

Hi McAllister. Thanks for your interesting work. I found that the idea of “copying gradient at x0” to any intermediate state x_t is quite similar to DRTune techniques - also a method to finetune text-to-images model with reward

Фото профиля Ronald Skorobogat
Ronald Skorobogat6 месяцев назад

really cool work. the uniform copy beating backprop-through-chain (rows I vs J) leans on flow jacobians being roughly psd via ot straightness. did you try it on distilled few-step students or late-training checkpoints where flows get more rotational?

Фото профиля David McAllister
David McAllister6 месяцев назад

Thanks! We didn’t try with distilled students though that sounds promising. Our experiments initialized from pretrained SD3 checkpoints, would you consider that late training?

Фото профиля AqibAi
AqibAi6 месяцев назад

Interesting direction—especially the “any scalar reward” framing makes it feel much more general than CFG. Curious how stable it is in practice across different reward signals, and how you’re handling reward hacking / drift during online updates.

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

Let's reverse engineer Disney's adorable, lifelike robot! I couldn't find a whitepaper, but this is how I think it's trained: 1. The emotional behaviors are curated by Disney animation artists, keyframe by keyframe. But it cannot be "rendered" directly on the robot because it doesn't take into account the complex real-world physics. 2. Reinforcement learning (RL) is a great tool for training low-level robot controllers. RL needs a reward function to optimize, and it's typically a task reward (e.g. walk in a straight line as fast as possible). The problem is that RL doesn't know what counts as "natural behavior", and often produces weird-looking body postures that somehow still maximize the reward. This is a human alignment problem just like ChatGPT. 3. Enters Adversarial Motion Prior (AMP): a technique that learns the human preference by training a classifier on what we consider "emotional & cute". In GAN literature, this is called a discriminator. Disney artists are good at creating such a dataset. You can then add AMP as an auxiliary reward in simulation to nudge the robot towards desired behaviors. AMP was developed by Peng et al. 2021 and Escontrela et al. 2022. 4. Add lots of data augmentation to make the controller robust to physical disturbances. In RL, it's called "domain randomization". This is a very powerful technique that bridges the gap between simulator and reality. Previously, OpenAI used domain randomization to train a 5-finger robot hand to manipulate a Rubik's Cube: IEEE news article gave hints about the pipeline: Finally, praying for world peace 🙏. I hope robotics like this will bring more joy to the world.

Jim Fan

314,807 просмотров • 3 лет назад