Video yükleniyor...

Video Yüklenemedi

Ana Sayfaya Dön

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 görüntüleme • 6 ay önce •via X (Twitter)

11 Yorum

David McAllister profil fotoğrafı
David McAllister6 ay önce

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 profil fotoğrafı
Kush Hari6 ay önce

You just got… Mcallister’d!!!

Samarth Sinha profil fotoğrafı
Samarth Sinha6 ay önce

My hero 😍😍😍😍😍

atharva ☆ profil fotoğrafı
atharva ☆6 ay önce

wow

Siddharth Ancha profil fotoğrafı
Siddharth Ancha6 ay önce

Really cool work @davidrmcall!!!

James Thewlis profil fotoğrafı
James Thewlis6 ay önce

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

Ritvik Singh profil fotoğrafı
Ritvik Singh6 ay önce

Congrats, David!

Phúc profil fotoğrafı
Phúc6 ay önce

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 profil fotoğrafı
Ronald Skorobogat6 ay önce

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 profil fotoğrafı
David McAllister6 ay önce

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 profil fotoğrafı
AqibAi6 ay önce

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.

Benzer Videolar

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 görüntüleme • 3 yıl önce