Video wird geladen...

Video konnte nicht geladen werden

Zur Startseite

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 Aufrufe • vor 6 Monaten •via X (Twitter)

11 Kommentare

Profilbild von David McAllister
David McAllistervor 6 Monaten

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

Profilbild von Kush Hari
Kush Harivor 6 Monaten

You just got… Mcallister’d!!!

Profilbild von Samarth Sinha
Samarth Sinhavor 6 Monaten

My hero 😍😍😍😍😍

Profilbild von atharva ☆
atharva ☆vor 6 Monaten

wow

Profilbild von Siddharth Ancha
Siddharth Anchavor 6 Monaten

Really cool work @davidrmcall!!!

Profilbild von James Thewlis
James Thewlisvor 6 Monaten

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

Profilbild von Ritvik Singh
Ritvik Singhvor 6 Monaten

Congrats, David!

Profilbild von Phúc
Phúcvor 6 Monaten

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

Profilbild von Ronald Skorobogat
Ronald Skorobogatvor 6 Monaten

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?

Profilbild von David McAllister
David McAllistervor 6 Monaten

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?

Profilbild von AqibAi
AqibAivor 6 Monaten

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.

Ähnliche Videos

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 Aufrufe • vor 3 Jahren