正在加载视频...

视频加载失败

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 年前